30 lines
585 B
Python
30 lines
585 B
Python
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()
|