1 files changed, 41 insertions, 0 deletions
diff --git a/bot.py b/bot.py
new file mode 100644
index 0000000..9240d32
--- /dev/null
+++ b/bot.py
@@ -0,0 +1,41 @@
+import discord
+from discord import Interaction, Intents, Embed
+from discord.ext.commands import Bot, Context, when_mentioned
+from dotenv import load_dotenv
+from os import getenv
+from sys import exit
+from typing import Literal
+import httpx
+import nltk
+import asyncio
+
+load_dotenv()
+
+if getenv('TOKEN') is None or getenv('OWNER_ID') is None:
+    print('please set a TOKEN and OWNER_ID environment variable')
+    exit(1)
+
+class QBBBot(Bot):
+    def __init__(self) -> None:
+        super().__init__(command_prefix=when_mentioned, intents=Intents.default())
+
+    async def setup_hook(self):
+        await self.load_extension('cogs.tossup')
+        await self.load_extension('cogs.solo')
+        await self.load_extension('jishaku')
+
+
+bot = QBBBot()
+bot.qb_categories = Literal['Literature', 'History', 'Science', 'Fine Arts', 'Religion', 'Mythology', 'Philosophy',
+                            'Social Science', 'Current Events', 'Geography', 'Other Academic', 'Trash']
+
+
+@bot.command()
+async def sync(ctx: Context):
+    if str(ctx.author.id) == getenv('OWNER_ID'):
+        await bot.tree.sync()
+        await ctx.send('ok')
+    else:
+        await ctx.send('no')
+
+bot.run(getenv('TOKEN'))