seekers.py

from __future__ import annotations

import argparse
import sys

from seekers import *

def parse_config_overrides(overrides: list[str]) -> dict[str, str]:
    parsed = {}

    for override in overrides:
        if "=" not in override:
            raise ValueError(f"Invalid config override {override!r}. "
                             f"Use the form option=value, e.g. global.seed=43")

        option, value = override.split("=", maxsplit=1)

        parsed[option.strip()] = value.strip()

    return parsed


def main():
    parser = argparse.ArgumentParser(description="Run python seekers AIs.")
    parser.add_argument("--no-grpc", action="store_true", help="Don't host a gRPC server.")
    parser.add_argument("--no-kill", action="store_true", help="Don't kill the process after the game is over.")
    parser.add_argument("--debug", action="store_true", help="Enable debug mode. This will enable debug drawing.")
    parser.add_argument("--address", "-a", type=str, default="localhost:7777",
                        help="Address of the server. (default: localhost:7777)")
    parser.add_argument("--config", "-c", type=str, default="config.ini",
                        help="Path to the config file. (default: config.ini)")
    parser.add_argument("--config-override", "--override", "-o", action="append",
                        help="Override a config option. Use the form option=value, e.g. global.seed=43.")
    parser.add_argument("--loglevel", "--log", "-l", type=str, default="INFO",
                        choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"])
    parser.add_argument("ai_files", type=str, nargs="*", help="Paths to the AIs.")

    args = parser.parse_args()

    if args.no_grpc and not args.ai_files:
        raise ValueError("At least one AI file must be provided if gRPC is disabled.")

    config = Config.from_filepath(args.config)
    for option, value in parse_config_overrides(args.config_override or []).items():
        section, key = option.split(".", maxsplit=1)

        try:
            config.import_option(section, key, value)
        except KeyError as e:
            raise ValueError(f"Invalid config option {e.args[0]!r}.") from e

    logging.basicConfig(level=args.loglevel, style="{", format=f"[{{name}}] {{levelname}}: {{message}}",
                        stream=sys.stdout)
    address = args.address if not args.no_grpc else False

    seekers_game = SeekersGame(
        local_ai_locations=args.ai_files,
        config=config,
        grpc_address=address,
        debug=args.debug,
        dont_kill=args.no_kill
    )
    seekers_game.listen()
    seekers_game.start()


if __name__ == "__main__":
    main()