rescriptoon/tooncher/_cli.py

import argparse
import os
import pathlib
import sys

import yaml

import tooncher

if sys.platform == "darwin":
    _TOONTOWN_ENGINE_DEFAULT_PATH = os.path.join(
        os.path.expanduser("~"),
        "Library",
        "Application Support",
        "Toontown Rewritten",
        "Toontown Rewritten",
    )
else:
    _TOONTOWN_ENGINE_DEFAULT_PATH = None


def run(
    username,
    config_path,
    engine_path=None,
    validate_ssl_certs=True,
    cpu_limit_percent=None,
):
    if os.path.exists(config_path):
        with open(config_path) as config_file:
            config = yaml.safe_load(config_file.read())
    else:
        config = {}
    if engine_path is None:
        if "engine_path" in config:
            engine_path = config["engine_path"]
        else:
            raise Exception("missing path to toontown engine")
    accounts = config["accounts"] if "accounts" in config else []
    for account in accounts:
        if account["username"] == username:
            tooncher.launch(
                engine_path=pathlib.Path(engine_path),
                username=account["username"],
                password=account["password"],
                validate_ssl_certs=validate_ssl_certs,
                cpu_limit_percent=cpu_limit_percent,
            )


def _init_argparser():
    argparser = argparse.ArgumentParser(description=None)
    argparser.add_argument("username")
    argparser.add_argument(
        "--config",
        "-c",
        metavar="path",
        dest="config_path",
        help="path to config file (default: %(default)s)",
        default=os.path.join(os.path.expanduser("~"), ".tooncher"),
    )
    argparser.add_argument(
        "--engine-path",
        "-e",
        metavar="path",
        dest="engine_path",
        default=_TOONTOWN_ENGINE_DEFAULT_PATH,
        help="\n".join(
            [
                "path to toontown engine.",
                "this overrides the one specified in config file",
                "(default: %(default)s)",
            ]
        ),
    )
    argparser.add_argument(
        "--no-ssl-cert-validation",
        "-k",
        dest="validate_ssl_certs",
        help="do not validate ssl certificates",
        action="store_false",
    )
    argparser.add_argument(
        "--cpu-limit",
        dest="cpu_limit_percent",
        type=int,
        default=None,
        help="maximally allowed cpu usage in percent"
        + " (requires cpulimit command, default: %(default)s)",
    )
    return argparser


def main() -> None:
    argparser = _init_argparser()
    args = argparser.parse_args()
    run(**vars(args))