From 7a09b0f7f3c17779366daa701383fe754a9fee70 Mon Sep 17 00:00:00 2001
From: flifloo <flifloo@gmail.com>
Date: Wed, 27 May 2020 23:51:31 +0200
Subject: [PATCH] Add db support

---
 extensions/reminders.py | 54 +++++++++++++++++++++--------------------
 1 file changed, 28 insertions(+), 26 deletions(-)

diff --git a/extensions/reminders.py b/extensions/reminders.py
index 11fa4e6..6dbeee3 100644
--- a/extensions/reminders.py
+++ b/extensions/reminders.py
@@ -7,6 +7,7 @@ from discord.ext.commands import CommandNotFound, BadArgument, MissingRequiredAr
 from discord.ext import tasks
 
 from bot_bde.logger import logger
+from bot_bde import db
 
 
 extension_name = "reminders"
@@ -26,7 +27,6 @@ def time_pars(s: str) -> timedelta:
 class Reminders(commands.Cog):
     def __init__(self, bot: commands.Bot):
         self.bot = bot
-        self.tasks = []
 
     @commands.group("reminder", pass_context=True)
     async def reminder(self, ctx: commands.Context):
@@ -47,13 +47,10 @@ class Reminders(commands.Cog):
     async def reminder_add(self, ctx: commands.Context, message: str, time: str):
         time = time_pars(time)
         now = datetime.now()
-        self.tasks.append({
-            "date": now + time,
-            "create": now,
-            "user": ctx.author.id,
-            "message": message,
-            "channel": ctx.channel.id,
-        })
+        s = db.Session()
+        s.add(db.Task(message, ctx.author.id, ctx.channel.id, now + time))
+        s.commit()
+        s.close()
 
         hours, seconds = divmod(time.seconds, 3600)
         minutes, seconds = divmod(seconds, 60)
@@ -65,37 +62,42 @@ class Reminders(commands.Cog):
     @reminder.group("list", pass_context=True)
     async def reminder_list(self, ctx: commands.Context):
         embed = Embed(title="Tasks list")
-        for i, t in enumerate(self.tasks):
-            if t["user"] == ctx.author.id:
-                embed.add_field(name=t["date"], value=f"N°{i} | {t['message']}", inline=False)
+        s = db.Session()
+        for t in s.query(db.Task).filter(db.Task.user == ctx.author.id).all():
+            embed.add_field(name=f"N°{t.id} | {t.date}", value=f"{t.message}", inline=False)
+        s.close()
         await ctx.send(embed=embed)
 
     @reminder.group("remove", pass_context=True)
     async def reminder_remove(self, ctx: commands.Context, n: int = None):
-        tasks =list(filter(lambda t: t["user"] == ctx.author.id, self.tasks))
         if n is None:
             await ctx.invoke(self.reminder_list)
-        elif n >= len(tasks):
-            raise BadArgument()
         else:
-            del self.tasks[n]
-            await ctx.message.add_reaction("\U0001f44d")
+            s = db.Session()
+            t = s.query(db.Task).filter(db.Task.id == n).first()
+            if t and t.user == ctx.author.id:
+                s.delete(t)
+                s.commit()
+                s.close()
+                await ctx.message.add_reaction("\U0001f44d")
+            else:
+                s.close()
+                raise BadArgument()
 
     @tasks.loop(minutes=1)
     async def reminders_loop(self):
-        trash = []
-        for t in self.tasks:
-            if t["date"] <= datetime.now():
-                self.bot.loop.create_task(self.reminder_exec(t))
-                trash.append(t)
+        s = db.Session()
+        for t in s.query(db.Task).filter(db.Task.date <= datetime.now()).all():
+            self.bot.loop.create_task(self.reminder_exec(t))
+            s.delete(t)
 
-        for t in trash:
-            del self.tasks[self.tasks.index(t)]
+        s.commit()
+        s.close()
 
-    async def reminder_exec(self, task: dict):
+    async def reminder_exec(self, task: db.Task):
         embed = Embed(title="You have a reminder !")
-        embed.add_field(name=task["date"], value=task["message"])
-        await self.bot.get_channel(task["channel"]).send(f"<@{task['user']}>", embed=embed)
+        embed.add_field(name=str(task.date), value=task.message)
+        await self.bot.get_channel(task.channel).send(f"<@{task.user}>", embed=embed)
 
     @commands.Cog.listener()
     async def on_command_error(self, ctx: commands.Context, error):