From 52801b4de1d63cd01191acf7fcee137977140ec0 Mon Sep 17 00:00:00 2001 From: lexicade Date: Fri, 27 Jan 2023 21:06:30 +0000 Subject: Project init --- cogs/countdown.py | 63 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 63 insertions(+) create mode 100644 cogs/countdown.py (limited to 'cogs/countdown.py') diff --git a/cogs/countdown.py b/cogs/countdown.py new file mode 100644 index 0000000..f73e59d --- /dev/null +++ b/cogs/countdown.py @@ -0,0 +1,63 @@ +from discord.ext import commands +import discord +import importlib +import utils +from datetime import datetime, timedelta +import time +importlib.reload(utils) + + +class Countdown(commands.Cog): + def __init__(self, bot): + self.bot = bot + + @commands.defer(ephemeral=False) + @commands.group() + async def countdown(self, ctx): + """Get a countdown for this server.""" + + @commands.defer(ephemeral=False) + @countdown.command(aliases=[], application_command_meta=commands.ApplicationCommandMeta(options=[])) + async def show(self, ctx, reminder_name: str, year: int = datetime.now().year, month: int = datetime.now().month, day: int = datetime.now().day, hour: int = 0, minute: int = 0): + """Get a countdown for this server.""" + print(f"123") + unix = await utils.sql(f'SELECT epoch, title FROM "database1".synthy.countdown WHERE guild_id = %s and title ilike %s;', (ctx.guild.id, reminder_name,)) + if unix: + dt = timedelta(seconds=unix[0]['epoch'] - int(time.time())) + # hours, remainder = divmod(dt.seconds, 3600) + # minutes, seconds = divmod(remainder, 60) + + # f"Timer: {dt.days} days, {hours} hours, {minutes} minutes, {seconds} seconds." + emb = await utils.embed(ctx, f"Countdown for {unix[0]['title']}", + f"**Date:** \n" + f"**Timer:** ") + + await ctx.send(embed=emb) + else: + await ctx.send(content=f"Cannot find this countdown.") + + @commands.defer(ephemeral=False) + @countdown.command(aliases=[], application_command_meta=commands.ApplicationCommandMeta(options=[])) + async def new(self, ctx, reminder_name: str, year=datetime.now().year, month: int = datetime.now().month, day: int = datetime.now().day, hour: int = 0, minute: int = 0): + """Set a countdown for this server, all entries should be numeric.""" + slash_context = await self.bot.get_slash_context(ctx.interaction) + year = int(slash_context.given_values['year']) if 'year' in slash_context.given_values.keys() else int(datetime.now().year) + month = int(slash_context.given_values['month']) if 'month' in slash_context.given_values.keys() else int(month) + day = int(slash_context.given_values['day']) if 'day' in slash_context.given_values.keys() else int(day) + hour = int(slash_context.given_values['hour']) if 'hour' in slash_context.given_values.keys() else int(hour) + minute = int(slash_context.given_values['minute']) if 'minute' in slash_context.given_values.keys() else int(minute) + + dt = datetime(year=year, month=month, day=day, hour=hour, minute=minute, second=0) + unix_epoch = int(time.mktime(dt.timetuple())) + await utils.sql('INSERT INTO "database1".synthy.countdown (guild_id, title, epoch) VALUES (%s, %s, %s);', (ctx.guild.id, reminder_name, unix_epoch,)) + await ctx.send(content="Done!") + + +def setup(bot): + print("INFO: Loading [PFP]... ", end="") + bot.add_cog(Countdown(bot)) + print("Done!") + + +def teardown(bot): + print("INFO: Unloading [PFP]") -- cgit v1.2.3