Discord/main.py

import logging

from discord import Activity, ActivityType

from modules.client import client
from modules.scheduled import scheduler
from modules.utils import config_get
from modules.utils_sync import config_get_sync

logging.basicConfig(
    level=logging.INFO,
    format="%(name)s.%(funcName)s | %(levelname)s | %(message)s",
    datefmt="[%X]",
)

logger = logging.getLogger(__name__)

try:
    import uvloop  # type: ignore

    uvloop.install()
except ImportError:
    pass


@client.event
async def on_ready():
    logger.info("Logged in as %s", client.user)

    activity_type = await config_get("type", "status")
    activity_message = await config_get("message", "status")

    if activity_type == "playing":
        await client.change_presence(
            activity=Activity(type=ActivityType.playing, name=activity_message)
        )
    elif activity_type == "watching":
        await client.change_presence(
            activity=Activity(type=ActivityType.watching, name=activity_message)
        )
    elif activity_type == "listening":
        await client.change_presence(
            activity=Activity(type=ActivityType.listening, name=activity_message)
        )
    elif activity_type == "streaming":
        await client.change_presence(
            activity=Activity(type=ActivityType.streaming, name=activity_message)
        )
    elif activity_type == "competing":
        await client.change_presence(
            activity=Activity(type=ActivityType.competing, name=activity_message)
        )
    elif activity_type == "custom":
        await client.change_presence(
            activity=Activity(type=ActivityType.custom, name=activity_message)
        )
    else:
        return

    logger.info(
        "Set activity type to %s with message %s", activity_type, activity_message
    )


def main():
    client.load_extension("cogs")

    try:
        scheduler.start()
        client.run(config_get_sync("bot_token", "bot"))
    except KeyboardInterrupt:
        scheduler.shutdown()
        exit()


if __name__ == "__main__":
    main()