From 730a99c7b6fa9ebd3102213a84d133d4682fcea9 Mon Sep 17 00:00:00 2001
From: flifloo <flifloo@gmail.com>
Date: Thu, 28 May 2020 14:45:31 +0200
Subject: [PATCH] Add optional calendar name argument on calendar notify list

---
 extensions/calendar.py | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/extensions/calendar.py b/extensions/calendar.py
index e171ef9..c5284f9 100644
--- a/extensions/calendar.py
+++ b/extensions/calendar.py
@@ -26,7 +26,7 @@ def query_calendar(name: str, guild: int) -> db.Calendar:
     return c
 
 
-def get_one_mention(ctx: commands.Context):
+async def get_one_mention(ctx: commands.Context):
     if ctx.message.channel_mentions and ctx.message.mentions:
         raise BadArgument()
     elif ctx.message.channel_mentions:
@@ -175,7 +175,7 @@ class Calendar(commands.Cog):
     @calendar_notify.group("add", pass_context=True)
     @commands.guild_only()
     async def calendar_notify_set(self, ctx: commands.Context, name: str):
-        m = get_one_mention(ctx)
+        m = await get_one_mention(ctx)
         s = db.Session()
         c = query_calendar(name, ctx.guild.id)
         n = s.query(db.CalendarNotify).filter(db.CalendarNotify.channel == m) \
@@ -192,14 +192,14 @@ class Calendar(commands.Cog):
 
     @calendar_notify.group("remove", pass_context=True)
     async def calendar_notify_remove(self, ctx: commands.Context, name: str):
-        m = get_one_mention(ctx)
+        m = await get_one_mention(ctx)
         s = db.Session()
         c = query_calendar(name, ctx.guild.id)
         n = s.query(db.CalendarNotify).filter(db.CalendarNotify.channel == m) \
             .filter(db.CalendarNotify.calendar_id == c.id) \
             .first()
         if n:
-            s.delete(db.CalendarNotify(m, c.id))
+            s.delete(n)
         else:
             s.close()
             raise BadArgument()
@@ -209,10 +209,14 @@ class Calendar(commands.Cog):
 
     @calendar_notify.group("list")
     @commands.guild_only()
-    async def calendar_notify_list(self, ctx: commands.Context):
+    async def calendar_notify_list(self, ctx: commands.Context, name: str = None):
         s = db.Session()
         embed = Embed(title="Notify list")
-        for c in s.query(db.Calendar).filter(db.Calendar.server == ctx.guild.id).all():
+        if name is None:
+            calendars = s.query(db.Calendar).filter(db.Calendar.server == ctx.guild.id).all()
+        else:
+            calendars = [query_calendar(name, ctx.guild.id)]
+        for c in calendars:
             notify = []
             for n in c.calendars_notify:
                 ch = self.bot.get_channel(n.channel)
@@ -220,7 +224,7 @@ class Calendar(commands.Cog):
                     notify.append(ch.mention)
                 elif type(ch) == DMChannel:
                     notify.append(ch.recipient.mention)
-            embed.add_field(name=c.name, value="\n".join(notify) or "Nothing here")
+            embed.add_field(name=c.name, value="\n".join(notify) or "Nothing here", inline=False)
         await ctx.send(embed=embed)
 
     @calendar_notify.group("trigger", pass_context=True)