import logging from discord import Activity, ActivityType from libbot import config_get from libbot.sync import config_get as sync_config_get from modules.client import client from modules.scheduled import scheduler 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(sync_config_get("bot_token", "bot")) except KeyboardInterrupt: scheduler.shutdown() exit() if __name__ == "__main__": main()