52 lines
1.2 KiB
Python
52 lines
1.2 KiB
Python
import asyncio
|
|
|
|
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.",
|
|
)
|
|
@click.option(
|
|
"--log-file",
|
|
default=None,
|
|
type=click.Path(),
|
|
help="Log to a file as well as stdout",
|
|
)
|
|
def cli(log_level: str, log_file: str | None):
|
|
logging.basicConfig(
|
|
level=log_level.upper(),
|
|
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
|
|
datefmt="%Y-%m-%d %H:%M:%S",
|
|
)
|
|
if log_file is not None:
|
|
file_handler = logging.FileHandler(log_file)
|
|
file_handler.setLevel(logging.DEBUG)
|
|
file_handler.setFormatter(
|
|
logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
|
|
)
|
|
logging.getLogger().addHandler(file_handler)
|
|
|
|
|
|
@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"]))
|
|
asyncio.run(bot.start(access_token=bot.cfg["bot"]["access_token"]))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
cli()
|