flifloo/Administrator-py
Archived
1
0
Fork 0
Code Activity

Add a calendar notify list command

Browse source
This commit is contained in:
Ethanell 2020-05-28 14:32:34 +02:00
parent ae5f706cf5
commit 7c4bdfc1f3
1 changed files with 17 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
extensions/calendar.py
View file

@ -3,7 +3,7 @@ from datetime import datetime, timedelta
import ics
import requests
from discord import Embed
from discord import Embed, DMChannel, TextChannel
from discord.ext import commands
from discord.ext import tasks
from discord.ext.commands import CommandNotFound, BadArgument, MissingRequiredArgument
@ -183,6 +183,22 @@ class Calendar(commands.Cog):
s.close()
await ctx.message.add_reaction("\U0001f44d")
@calendar_notify.group("list")
@commands.guild_only()
async def calendar_notify_list(self, ctx: commands.Context):
s = db.Session()
embed = Embed(title="Notify list")
for c in s.query(db.Calendar).filter(db.Calendar.server == ctx.guild.id).all():
notify = []
for n in c.calendars_notify:
ch = self.bot.get_channel(n.channel)
if type(ch) == TextChannel:
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")
await ctx.send(embed=embed)
@calendar_notify.group("trigger", pass_context=True)
@commands.guild_only()
async def calendar_notify_trigger(self, ctx: commands.Context, name: str):