main.py

import sys
import winsound
from arknights.chores.login import from_login_to_lobby, close_calendar
from arknights.chores.farm_stage import navigate_to_target_stage, start_farming
from arknights.cli import show_banner, parse_args
from arknights.logging import configureLogger
from arknights.stage import show_info, get_today_stages


if __name__ == "__main__":
    logger = configureLogger()

    show_banner()

    args = parse_args()

    skip_login = args.skip_login
    first_login = args.first_login
    login_only = args.login_only
    describe = args.describe
    today = args.today

    if today:
        print(f"Available today: {', '.join(get_today_stages())}")
        sys.exit()

    if describe is not None:
        show_info(describe)
        sys.exit()

    logger.info(
        f"Session initiated with skip_login - {skip_login}, first_login - {first_login} and login_only - {login_only}"
    )

    if skip_login and login_only:
        logger.error("Cannot specify skip login and login only at the same time")
        sys.exit()

    if not skip_login:
        from_login_to_lobby()

        if first_login:
            close_calendar()

        if login_only:
            logger.info("Login to lobby only has completed.")
            sys.exit()

    skip_navigation = args.skip_navigation
    if not skip_navigation:
        navigate_to_target_stage(args.stage)

    refill_count = args.refill_count
    start_farming(refill_count)

    print('\a')
    winsound.Beep(200, 8000)