import click import logging from tortoise import run_async log = logging.getLogger(__name__) @click.group() @click.option( "--log-level", default="INFO", type=click.Choice(["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"], case_sensitive=False), help="Set the log level." ) def cli(log_level: str): logging.basicConfig(level=log_level.upper()) @cli.command() def run(): """Runs the bot""" log.info("Starting bot.") from .main import bot run_async(bot.start(access_token=bot.cfg["bot"]["access_token"])) if __name__ == "__main__": cli()