Название в значительной степени суммирует то, что я хотел бы, чтобы произошло.
Вот что у меня есть, и хотя программа не взорвалась от неположительного целого числа, я хочу, чтобы пользователь был проинформирован о том, что неположительное целое число в основном бессмысленно.
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)
int
не удастся, все еще будет читаемый вывод? Или вы должныtry
raise
конвертировать вручную для этого?error: argument foo: invalid check_positive value: 'foo=<whatever>'
. Вы могли бы просто добавитьtry:
...except ValueError:
вокруг него, который повторно вызывает исключение с лучшим сообщением об ошибке.type
будет рекомендуемым вариантом для обработки условий / проверок, как в ответе Юши.В вашем конкретном случае вы также можете использовать
choices
параметр, если ваш верхний предел также известен:Примечание: используйте
range
вместоxrange
Python 3.xисточник
if
намного быстрее.xrange
) и не требует дополнительного кода. Этот компромисс доступен. Каждый решает, в какую сторону идти.Быстрый и грязный способ, если у вас есть предсказуемый максимум и минимум для вашего аргумента, это использовать
choices
с диапазономисточник
Более простой альтернативой, особенно в случае
argparse.ArgumentParser
создания подклассов , является инициирование проверки изнутриparse_args
метода.Внутри такого подкласса:
Эта техника может быть не такой крутой, как пользовательская функция, но она делает свою работу.
О себе
ArgumentParser.error(message)
:Кредит: ответ Джонатаном
источник
print "-g/--games: must be positive."; sys.exit(1)
простоparser.error("-g/--games: must be positive.")
. (Использование, как в ответе Джонатана .)В случае, если кто-то (например, я) сталкивается с этим вопросом в поиске в Google, вот пример того, как использовать модульный подход, чтобы аккуратно решить более общую проблему разрешения целых чисел argparse только в указанном диапазоне :
Это позволяет вам сделать что-то вроде:
Теперь переменная
foo
допускает только положительные целые числа , как в запросе OP.Обратите внимание, что в дополнение к вышеперечисленным формам возможен только максимум с
IntRange
:источник