В Python, используя argparse, допускают только положительные целые числа

Название в значительной степени суммирует то, что я хотел бы иметь.

Вот что у меня есть, и пока программа не взорвется на неположительное целое число, я хочу, чтобы пользователь был проинформирован о том, что неположительное целое является в основном абсурдом.

import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-g", "--games", type=int, default=162,
                    help="The number of games to simulate")
args = parser.parse_args()

И вывод:

python simulate_many.py -g 20
Setting up...
Playing games...
....................

Выход с отрицательным значением:

python simulate_many.py -g -2
Setting up...
Playing games...

Теперь, очевидно, я мог бы просто добавить if, если определить if args.games отрицательно, но мне было любопытно, есть ли способ уловить его на уровне argparse, чтобы воспользоваться автоматической печатью использования.

В идеале, он напечатает что-то похожее на это:

python simulate_many.py -g a
usage: simulate_many.py [-h] [-g GAMES] [-d] [-l LEAGUE]
simulate_many.py: error: argument -g/--games: invalid int value: 'a'

Так же:

python simulate_many.py -g -2
usage: simulate_many.py [-h] [-g GAMES] [-d] [-l LEAGUE]
simulate_many.py: error: argument -g/--games: invalid positive int value: '-2'

Пока я делаю это, и я доволен, что я счастлив:

if args.games <= 0:
    parser.print_help()
    print "-g/--games: must be positive."
    sys.exit(1)

Ответы

Ответ 1

Это должно быть возможно с использованием type. Вам все еще нужно определить фактический метод, который решит это за вас:

def check_positive(value):
    ivalue = int(value)
    if ivalue <= 0:
        raise argparse.ArgumentTypeError("%s is an invalid positive int value" % value)
    return ivalue

parser = argparse.ArgumentParser(...)
parser.add_argument('foo', type=check_positive)

Это просто адаптированный пример из функции perfect_square в документах на argparse.

Ответ 2

type будет рекомендуемым вариантом для обработки условий/проверок, как в ответе Yuushi.

В вашем конкретном случае вы также можете использовать параметр choices, если ваш верхний предел также известен:

parser.add_argument('foo', type=int, choices=xrange(5, 10))

Ответ 3

Быстрый и грязный способ, если у вас есть предсказуемый максимум, а также минимальный для вашего аргумента, используется choices с диапазоном

parser.add_argument('foo', type=int, choices=xrange(0, 1000))

Ответ 4

Более простая альтернатива, особенно если используется подкласс argparse.ArgumentParser, состоит в том, чтобы инициировать проверку изнутри метода parse_args.

Внутри такого подкласса:

def parse_args(self, args=None, namespace=None):
    """Parse and validate args."""
    namespace = super().parse_args(args, namespace)
    if namespace.games <= 0:
         raise self.error('The number of games must be a positive integer.')
    return namespace

Эта техника может быть не такой крутой, как пользовательская функция вызова, но она выполняет свою работу.

О ArgumentParser.error(message):

Этот метод печатает сообщение об использовании, включая сообщение о стандартной ошибке, и завершает программу с кодом состояния 2.

Кредит: ответ джонатана