Укажите формат для входных аргументов в Python argparse


у меня есть скрипт python, который требует некоторых входов командной строки, и я использую argparse для их разбора. Я нашел документацию немного запутанной и не смог найти способ проверить формат во входных параметрах. Что я имею в виду под проверкой формата объясняется с помощью этого примера сценария:

parser.add_argument('-s', "--startdate", help="The Start Date - format YYYY-MM-DD ", required=True)
parser.add_argument('-e', "--enddate", help="The End Date format YYYY-MM-DD (Inclusive)", required=True)
parser.add_argument('-a', "--accountid", type=int, help='Account ID for the account for which data is required (Default: 570)')
parser.add_argument('-o', "--outputpath", help='Directory where output needs to be stored (Default: ' + os.path.dirname(os.path.abspath(__file__)))

мне нужно проверить опцию -s и -e что вход пользователя в формате YYYY-MM-DD. Есть ли вариант в argparse, о котором я не знаю добиться этого.

3 65

3 ответа:

на документация:

The type аргумент ключевого слова add_argument() позволяет выполнять любые необходимые проверки типов и преобразования типов ... type= может принимать любой вызываемый объект, который принимает один строковый аргумент и возвращает преобразованное значение

вы могли бы сделать что-то вроде:

def valid_date(s):
    try:
        return datetime.strptime(s, "%Y-%m-%d")
    except ValueError:
        msg = "Not a valid date: '{0}'.".format(s)
        raise argparse.ArgumentTypeError(msg)

потом type:

parser.add_argument("-s", 
                    "--startdate", 
                    help="The Start Date - format YYYY-MM-DD", 
                    required=True, 
                    type=valid_date)

просто чтобы добавить к ответу выше, вы можете использовать лямбда-функцию, если хотите сохранить ее в одном лайнере. Например:

parser.add_argument('--date', type=lambda d: datetime.strptime(d, '%Y%m%d'))

старый поток, но вопрос был по-прежнему актуален для меня, по крайней мере!

для других, кто попал в это через поисковые системы: в Python 3.7, вы можете использовать стандартный .fromisoformat метод класса вместо изобретения колеса для ISO-8601 совместимых дат, например:

parser.add_argument('-s', "--startdate",
    help="The Start Date - format YYYY-MM-DD",
    required=True,
    type=datetime.date.fromisoformat)
parser.add_argument('-e', "--enddate",
    help="The End Date format YYYY-MM-DD (Inclusive)",
    required=True,
    type=datetime.date.fromisoformat)