From 821385cff381a4770dd0605aa14984d770aa6941 Mon Sep 17 00:00:00 2001
From: flifloo <flifloo@gmail.com>
Date: Thu, 28 May 2020 15:14:18 +0200
Subject: [PATCH] Add calendar notify loop

---
 extensions/calendar.py | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/extensions/calendar.py b/extensions/calendar.py
index c5284f9..8281b96 100644
--- a/extensions/calendar.py
+++ b/extensions/calendar.py
@@ -1,5 +1,6 @@
 import re
-from datetime import datetime, timedelta
+from datetime import datetime, timedelta, timezone
+from operator import xor
 
 import ics
 import requests
@@ -234,6 +235,15 @@ class Calendar(commands.Cog):
         now = datetime.now()
         await c.notify(self.bot, c.events(now, now)[0])
 
+    @tasks.loop(minutes=1)
+    async def calendar_notify_loop(self):
+        s = db.Session()
+        now = datetime.now().replace(tzinfo=timezone.utc).astimezone(tz=None)
+        for c in s.query(db.Calendar).all():
+            for e in c.events(now, now):
+                if xor(e.begin >= now - timedelta(minutes=30), e.begin >= now - timedelta(minutes=10)):
+                    self.bot.loop.create_task(await c.notify(self.bot, e))
+
     @commands.Cog.listener()
     async def on_command_error(self, ctx: commands.Context, error):
         if ctx.invoked_with == extension_name or \
@@ -247,11 +257,16 @@ class Calendar(commands.Cog):
                 await ctx.send("An error occurred !")
                 raise error
 
+    def cog_unload(self):
+        self.calendar_notify_loop.stop()
+
 
 def setup(bot):
     logger.info(f"Loading...")
     try:
-        bot.add_cog(Calendar(bot))
+        calendar = Calendar(bot)
+        bot.add_cog(calendar)
+        calendar.calendar_notify_loop.start()
     except Exception as e:
         logger.error(f"Error loading: {e}")
     else: