main.py

import argparse
import asyncio
import logging
from typing import NoReturn

import chess.engine
import httpx

import config
from game_manager import GameManager
from lichess import Lichess

LOGO = """
                              __ _             _           _   
  __ _ ___ _   _ _ __   ___  / /(_) ___       | |__   ___ | |_ 
 / _` / __| | | | '_ \ / __|/ / | |/ _ \ _____| '_ \ / _ \| __|
| (_| \__ \ |_| | | | | (__/ /__| | (_) |_____| |_) | (_) | |_ 
 \__,_|___/\__, |_| |_|\___\____/_|\___/      |_.__/ \___/ \__|
           |___/                                               
           """

logger = logging.getLogger(__name__)


async def main(args: argparse.Namespace) -> NoReturn:
    logging_handlers = [logging.StreamHandler()]
    if args.log:
        logging_handlers.append(logging.FileHandler(args.log))
    logging.basicConfig(
        level=logging.DEBUG if args.verbose else logging.INFO,
        format="%(asctime)s %(levelname)-8s %(message)s",
        datefmt="%Y-%m-%d %H:%M:%S",
        handlers=logging_handlers,
    )

    logging.getLogger(httpx.__name__).setLevel(
        logging.DEBUG if args.verbose else logging.WARNING
    )

    config.load_config(args.config)

    print(LOGO)

    async with Lichess() as li:
        if args.upgrade:
            if li.title == "BOT":
                logger.warning(
                    f"{li.username} is already a BOT account. Run asyncLio-bot without the upgrade flag in the future."
                )
            else:
                await li.upgrade_account()
                logger.info(f"Upgraded {li.username} to a BOT account.")
                return

        if li.title != "BOT":
            logger.critical("asyncLio-bot can only be used by BOT accounts.")
            return

        logger.info(f"Logged in as {li.me}.")
        await GameManager(li).watch_event_stream()


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--upgrade", "-u", action="store_true", help="Upgrade account to BOT account."
    )
    parser.add_argument("--log", "-l", type=str, help="Log file.")
    parser.add_argument(
        "--config", "-c", type=str, default="config.yml", help="Log file."
    )
    parser.add_argument(
        "--verbose", "-v", action="store_true", help="Make output more verbose."
    )

    asyncio.set_event_loop_policy(chess.engine.EventLoopPolicy())
    asyncio.run(main(parser.parse_args()))