main.py

import logging
import os
from datetime import UTC, datetime

import discord
from aiohttp import ClientSession
from discord.ext import commands

from config import config


class Bot(commands.AutoShardedBot):
    error_channel: discord.TextChannel
    session: ClientSession
    launch_time: int
    colour = 0xFF7000

    def __init__(self, *args, **kwargs):
        super().__init__(
            *args,
            **kwargs,
            command_prefix=commands.when_mentioned,
            help_command=None,
            intents=discord.Intents.default(),
            case_insensitive=True,
            allowed_mentions=discord.AllowedMentions(everyone=False),
            allowed_installs=discord.app_commands.AppInstallationType(
                guild=True, user=True
            ),
            allowed_contexts=discord.app_commands.AppCommandContext(
                guild=True, dm_channel=True, private_channel=True
            ),
        )

    async def setup_hook(self) -> None:
        await self.load_extension("jishaku")
        for cog in os.listdir("./cogs"):
            if cog.endswith(".py"):
                await self.load_extension(f"cogs.{cog[:-3]}")

        self.error_channel = await self.fetch_channel(config["error_channel"])
        self.session = ClientSession()
        self.launch_time = round(datetime.now(UTC).timestamp())

    async def on_ready(self) -> None:
        print(f"Logged in as {self.user} (ID: {self.user.id})")

    async def close(self) -> None:
        if hasattr(self, "session"):
            await self.session.close()
        await super().close()


bot = Bot()

if __name__ == "__main__":
    logging_level = logging.DEBUG if config.get("debug") else logging.WARNING

    bot.run(config["token"], log_level=logging_level, root_logger=True)