bot.py

from __future__ import annotations
from pathlib import Path
from traceback import print_exception

import discord
from discord.ext import commands
from loguru import logger

from app_cmds import cmds, groups
from internal.bot import CrajyBot
from secret.TOKEN import TOKEN
from secret.constants import BOT_TEST_SERVER


intents = discord.Intents.default()
intents.members = True  # TO DO: Investigate feature-loss with Member intents disabled
intents.message_content = True

bot = CrajyBot(
    command_prefix=commands.when_mentioned_or("."),
    activity=discord.Activity(
        type=discord.ActivityType.watching, name="cute panda videos or something."
    ),
    intents=intents,
    owner_id=271586885346918400,
)

for cmd in cmds:
    bot.tree.add_command(cmd, guild=discord.Object(BOT_TEST_SERVER))

for group in groups:
    bot.tree.add_command(group, guild=discord.Object(BOT_TEST_SERVER))


async def main(exts: list[str] | None = None) -> None:
    if exts is not None:
        for ext in exts:
            await bot.load_extension(ext)

    else:
        for ext in Path("./exts").iterdir():
            if ext.name in ["__pycache__", "__init__.py"]:
                continue
            try:
                await bot.load_extension(f"exts.{ext.name}")
            except Exception as e:
                logger.error(f"Error occurred while loading {ext=}")
                print_exception(type(e), e, e.__traceback__)
                continue

    try:
        await bot.load_extension("jishaku")
    except Exception as e:
        print_exception(type(e), e, e.__traceback__)
        logger.warning(f"Couldn't load jishaku: {e}")

    logger.info("Finished loading extensions")

    async with bot:
        await bot.start(TOKEN)