64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
import asyncio
|
|
import time
|
|
|
|
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()
|
|
@click.option(
|
|
"--resume/--no-resume",
|
|
default=False,
|
|
help="Whether to do a full sync, or resume from previous"
|
|
)
|
|
def run(resume: bool):
|
|
"""Runs the bot"""
|
|
log.info("Starting bot.")
|
|
from .main import bot
|
|
|
|
if resume is False:
|
|
bot.config.store_sync_tokens = False
|
|
bot.loaded_sync_token = ""
|
|
log.critical("Not resuming from previous sync. This may be slow!")
|
|
time.sleep(3)
|
|
|
|
# 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()
|