From 13ee0683582498cedc66d2e3a14cb32510392135 Mon Sep 17 00:00:00 2001 From: Parker Date: Tue, 21 Jan 2025 21:06:33 -0600 Subject: Fix database creation --- code/bot.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) (limited to 'code/bot.py') diff --git a/code/bot.py b/code/bot.py index b067f73..919f394 100644 --- a/code/bot.py +++ b/code/bot.py @@ -1,5 +1,5 @@ import discord -from discord.ext import commands +from discord.ext import commands, tasks import os from utils.database import Base, engine @@ -14,6 +14,7 @@ class MyBot(commands.Bot): ) async def setup_hook(self): + delete_accounts_task.start() for ext in os.listdir("./code/cogs"): if ext.endswith(".py"): await self.load_extension(f"cogs.{ext[:-3]}") @@ -28,7 +29,14 @@ async def on_ready(): config.LOG.info(f"{bot.user} has connected to Discord.") -if __name__ == "__main__": +@tasks.loop(minutes=1) +async def delete_accounts_task(): + from utils.jellyfin_delete import delete_accounts + Base.metadata.create_all(bind=engine) + delete_accounts() + + +if __name__ == "__main__": config.load_config() bot.run(config.BOT_TOKEN) -- cgit v1.2.3-70-g09d2