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'))