From 20ad83ba2388fbdee59d43a9cebe2805bfb81167 Mon Sep 17 00:00:00 2001
From: Stijn De Clercq <declercq.stijn@outlook.com>
Date: Sat, 24 Apr 2021 19:15:43 +0200
Subject: [PATCH] Corona vaccinations

---
 cogs/corona.py  | 100 ++++++++++++++++++++++++++++++++++++++++++++++++
 files/help.json |   1 +
 2 files changed, 101 insertions(+)

diff --git a/cogs/corona.py b/cogs/corona.py
index 181eb29..0659884 100644
--- a/cogs/corona.py
+++ b/cogs/corona.py
@@ -36,6 +36,106 @@ class Corona(commands.Cog):
 
         await self.sendEmbed(ctx, dic, vaccine)
 
+    @corona.command(name="Vaccinations", aliases=["V", "Vacc", "Vax", "Vaxx"])
+    async def vaccinations(self, ctx):
+        population = 11632326
+
+        res = requests.get("https://covid-vaccinatie.be/api/v1/administered.json")
+        updated = requests.get("https://covid-vaccinatie.be/api/v1/last-updated.json")
+
+        if res.status_code != 200:
+            return
+
+        res = res.json()
+        administered = res["result"]["administered"]
+
+        first_dose = 0
+        second_dose = 0
+
+        # Get dose counts
+        for entry in administered:
+            first_dose += entry["first_dose"]
+            second_dose += entry["second_dose"]
+
+        new_info = self.get_new_vaccinations(administered)
+
+        # % of population that has received their vaccine
+        first_dose_perc = round(first_dose * 100 / population, 2)
+        second_dose_perc = round(second_dose * 100 / population, 2)
+
+        # % of population that has received their vaccine TODAY
+        first_today_perc = round(new_info["today"]["first_dose"] * 100 / population, 2)
+        second_today_perc = round(new_info["today"]["second_dose"] * 100 / population, 2)
+
+        # Difference compared to the day before
+        first_trend = self.trend(new_info, "first_dose")
+        second_trend = self.trend(new_info, "second_dose")
+
+        embed = discord.Embed(colour=discord.Colour.red(), title="Vaccinatiecijfers")
+
+        embed.add_field(name="Eerste Dosis", value="{:,} ({} %)".format(first_dose, first_dose_perc))
+        embed.add_field(name="Eerste Dosis (Vandaag)",
+                        value="{:,} ({} %)".format(new_info["today"]["first_dose"], first_today_perc))
+        embed.add_field(name="Eerste Dosis (Verschil)", value=first_trend)
+        embed.add_field(name="Tweede Dosis", value="{:,} ({} %)".format(second_dose, second_dose_perc))
+        embed.add_field(name="Tweede Dosis (Vandaag)",
+                        value="{:,} ({} %)".format(new_info["today"]["second_dose"], second_today_perc))
+        embed.add_field(name="Tweede Dosis (Verschil)", value=second_trend)
+
+        # Only add updated timestamp if the request succeeded
+        # this isn't really a big deal so the command doesn't fail
+        # if it didn't
+        if updated.status_code == 200:
+            embed.set_footer(text="Laatste update: {}".format(updated.json()["result"]["last_updated"]["updated"]))
+
+        return await ctx.send(embed=embed)
+
+    def get_new_vaccinations(self, data):
+        """
+        Finds the amount of new doses administered today & the day before
+        """
+        reversed_data = list(reversed(data))
+
+        # Find the most recent date that was added
+        # (not necessarily today)
+        latest_date = reversed_data[0]["date"]
+        date_before = ""
+
+        info = {
+            "today": {
+                "first_dose": 0,
+                "second_dose": 0
+            },
+            "yesterday": {
+                "first_dose": 0,
+                "second_dose": 0
+            }
+        }
+
+        # Find first date doses
+        for entry in reversed_data:
+            if entry["date"] == latest_date:
+                info["today"]["first_dose"] += entry["first_dose"]
+                info["today"]["second_dose"] += entry["second_dose"]
+            else:
+                # Find the date before the most recent one
+                # to calculate differences
+                date_before = entry["date"]
+                break
+
+        # Find second date doses
+        for entry in reversed_data:
+            # Info on first date was added above
+            if entry["date"] == latest_date:
+                continue
+            elif entry["date"] == date_before:
+                info["yesterday"]["first_dose"] += entry["first_dose"]
+                info["yesterday"]["second_dose"] += entry["second_dose"]
+            else:
+                break
+
+        return info
+
     @corona.command(aliases=["lb", "leaderboards"], hidden=True)
     async def leaderboard(self, ctx):
         """
diff --git a/files/help.json b/files/help.json
index 3fe70ee..fd0100b 100644
--- a/files/help.json
+++ b/files/help.json
@@ -30,6 +30,7 @@
   "coinflip": "Gooi een muntje, indien je Didier Dinks inzet kan je je inzet verdubbelen (of verliezen).",
   "config": "Past constanten in het config bestand aan.",
   "corona": "Coronatracker voor [Land].\nIndien je geen land opgeeft is dit standaard België.\nCorona Global voor wereldwijde cijfers.",
+  "corona vaccinations": "Vaccinatiecijfers voor België.",
   "dadjoke": "Didier vertelt een dad joke.",
   "define": "Geeft de definitie van [Woord] zoals het in de Urban Dictionary staat.\nZoektermen met spaties moeten **niet** tussen aanhalingstekens staan.",
   "detect": "Didier probeert de taal van [Tekst] te detecteren.",