improve corona command

ubot/modules/corona.py

@@ -0,0 +1,27 @@
+# SPDX-License-Identifier: GPL-2.0-or-later
+
+from requests import get
+
+from ubot.micro_bot import micro_bot
+
+ldr = micro_bot.loader
+
+
+@ldr.add(pattern="corona")
+async def corona(event):
+    text_arg = event.pattern_match.group(1)
+
+    if text_arg:
+        with get(f"https://corona.lmao.ninja/countries/{text_arg}") as response:
+            if response.status_code == 200:
+                response = response.json()
+            else:
+                await event.reply(f"`An error occurred, response code: `**{response.status}**")
+                return
+
+        response_text = f"`Stats for `**{response['country']}**\n\n`Cases: `**{response['cases']}**\n`Deaths: `**{response['deaths']}**\n`Recoveries: `**{response['recovered']}**"
+        await event.reply(response_text)
+    else:
+        corona_data = get("https://corona.lmao.ninja/all").json()
+        response_text = f"`Global stats`\n\n`Cases: `**{corona_data['cases']}**\n`Deaths: `**{corona_data['deaths']}**\n`Recoveries: `**{corona_data['recovered']}**"
+        await event.reply(response_text)

ubot/modules/evaluation.py

@@ -11,13 +11,6 @@ from ubot.micro_bot import micro_bot
 ldr = micro_bot.loader
 
 
-@ldr.add(pattern="corona")
-async def corona(event):
-    corona_data = get("https://corona.lmao.ninja/all").json()
-    response_text = f"`Cases: `**{corona_data['cases']}**\n`Deaths: `**{corona_data['deaths']}**\n`Recoveries: `**{corona_data['recovered']}**"
-    await event.reply(response_text)
-
-
 @ldr.add(pattern="chatid")
 async def chatidgetter(event):
     if event.is_reply: