main.py

from discord import Activity, ActivityType, Status, Intents
from discord.ext import commands
from sdb_lib import Log, Config
from os import listdir
from json import dump


bot = commands.Bot(
    command_prefix = Config.prefix,
    case_insensitive = True,
    help_command = None,
    intents = Intents.all()
)


def check_files():
    try:
        open("./config.json", "r").close()
    except FileNotFoundError:
        Log.error("No config file found, create a config file.")
        exit()

    try:
        open("./data.json", "r").close()
    except FileNotFoundError:
        Log.warning("No data file found, creating data file.")
        with open("./data.json", "w") as f:
            dump(
                {
                    "logging": {
                        "channel": 0
                    },
                    "tickets": {
                        "number": 0,
                        "category": 0,
                        "support": 0,
                        "tickets": []
                    }
                },
                f, indent = 4
            )


def load_exts():
    errors = 0

    for folder in listdir("./exts/"):
        if "." in folder:
            continue

        for ext in listdir(f"./exts/{folder}/"):
            if not ext.endswith(".py"):
                continue

            ext = ext.removesuffix(".py")

            try:
                bot.load_extension(f"exts.{folder}.{ext}")
                Log.info(f"Loaded '{ext}'")
            except Exception as exc:
                Log.error(f"Failed to load '{ext}':\n'{exc}'")
                errors += 1

    return errors


async def change_presence():
    modes = [Status.online, Status.idle, Status.dnd, Status.offline]
    mode = modes[Config.presence.mode]

    _activity = Config.presence.activity
    activities = [
        Activity(type = ActivityType.playing, name = _activity), Activity(type = ActivityType.streaming, name = _activity),
        Activity(type = ActivityType.listening, name = _activity), Activity(type = ActivityType.watching, name = _activity)
    ]
    activity = activities[Config.presence.status]

    return await bot.change_presence(status = mode, activity = activity)


@bot.event
async def on_ready():
    await change_presence()
    Log.info(f"Registered {len(bot.application_commands)} slash commands")
    Log.info(f"{bot.user.name} ({bot.application_id}) online")


Log.info("Checking files...")
check_files()

Log.info("Loading extensions...")
errors = load_exts()
if errors > 0:
    Log.warning(f"{errors} error(s) while loading")

bot.run(Config.token)