From 148712f9a69adacc156469c5ea968e6fb1dd09f4 Mon Sep 17 00:00:00 2001
From: kevincaires <kevincaires@ymail.com>
Date: Sat, 3 Sep 2022 13:52:07 -0300
Subject: [PATCH] ADD options in news command.

---
 utils/commands.py | 49 +++++++++++++++++++++++++++++++++++++----------
 1 file changed, 39 insertions(+), 10 deletions(-)

diff --git a/utils/commands.py b/utils/commands.py
index da7286d..3261b8a 100644
--- a/utils/commands.py
+++ b/utils/commands.py
@@ -314,22 +314,51 @@ async def machine_info(bot: object, *args: str) -> str:
 
 
 @client.command(aliases=['nw'])
-async def news(bot: object) -> None:
-    f"""
+async def news(bot: object, *options: str) -> None:
+    """
     Return some news from Google.
+    options:
+        quantity: int
+        search: str
     """
-    _news = News(quantity=5)
-    news = _news.news()
     embed = Embed(type='rich')
+    filter = {}
+    news = None
+
+    if not options:
+        _news = News(quantity=5)
+        news = _news.news()
+
+    else:
+        # Validate option operation.
+        if not all(['=' in op for op in options]):
+            return await bot.send('Blabla')
+
+        for op in options:
+            key, value = op.split('=')
+            filter[key] = value
+
+        _news = News(quantity=filter.get('quantity', 5))
+        news = _news.filter(phrase=filter.get('search'))
+
+    if not news:
+        return
 
     for new in news:
-        # TODO: Descomentar o código do match case
-        dt = datetime.fromisoformat(
-            new['publishedAt']
-        ).astimezone(pytz.timezone('America/Sao_Paulo'))
+        try:
+            dt = datetime.fromisoformat(
+                new['publishedAt']
+            ).astimezone(pytz.timezone('America/Sao_Paulo'))
+        except ValueError:
+            dt = datetime.strptime(new['publishedAt'], '%Y-%m-%dT%H:%M:%Sz')
+        except Exception as e:
+            logger.error(e) 
+            continue
+
+        embed.add_field(name='Font', value=new['source']['name'], inline=False)
         embed.add_field(name='Published at', value=dt.isoformat(), inline=False)
-        embed.add_field(name='link', value=new['url'], inline=False)
+        embed.add_field(name='Link', value=new['url'], inline=False)
         embed.add_field(name=new['title'], value=new['description'], inline=False)
         embed.add_field(name='---', value='---')
 
-    return await bot.send(f'**`{new["source"]["name"]}`**', embed=embed)
+    return await bot.send(f'**`News`**', embed=embed)