from argparse import ArgumentParser, Namespace
from pathlib import Path


QUEENS_FILE = "queens.txt"
PICKS_FILE_GLOB = "picks-*.txt"
ELIMINATIONS_FILE = "eliminations.txt"
ROSTERS_FILE = "rosters.json"


def parse_args() -> Namespace:
    parser = ArgumentParser()
    parser.add_argument(
        "season_dir",
        type=Path,
        default=Path("."),
        help="Directory containing season data",
    )
    parser.add_argument(
        "--eliminate", type=str, help="Eliminate a queen", metavar="QUEEN"
    )
    parser.add_argument("--draft", action="store_true", help="Run the initial draft")
    parser.add_argument("--print", action="store_true", help="Print the current status")

    return parser.parse_args()