cmdparser.py

import argparse
from config import (
    SINGLEPLAYER_MODE, HOT_MULTIPLAYER_MODE, CO_OP_MODE, DIFFICULTY_EASY,
    DIFFICULTY_HARD, DIFFICULTY_NORMAL, DIFFICULTY_NIGHTMARE)


def parse_game_cmd_args():
    """ Initializes and parses cmd args. """
    parser = argparse.ArgumentParser(
        description='Capture the Flag game with Tanks!')

    parser.add_argument(
        "--game-mode",
        type=str,
        default=SINGLEPLAYER_MODE,
        choices=(SINGLEPLAYER_MODE, HOT_MULTIPLAYER_MODE, CO_OP_MODE),
        help="set the game mode"
    )

    parser.add_argument(
        "--map",
        type=str,
        default="map0.json",
        help="set the map to play."
    )

    parser.add_argument(
        "--difficulty", type=str, default=DIFFICULTY_NORMAL,
        choices=(DIFFICULTY_EASY, DIFFICULTY_NORMAL, DIFFICULTY_HARD,
                 DIFFICULTY_NIGHTMARE),
        help="set the map to play.")

    return parser.parse_args()