在Python中，使用Argparse，只允许正整数

Question

标题几乎概括了我希望发生的事情。

这是我所拥有的，虽然程序不会在非正整数上爆炸，但我希望用户被告知非正整数基本上是无稽之谈。

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参数。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)

Answer 1

如果arg有一个可预测的最大值和最小值，那么最快捷的方法就是使用选项和范围

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

Answer 2

type将是处理条件/检查的推荐选项，如Yuushi的答案所示。

在特定情况下，如果已知上限，也可以使用choices参数：

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

注意：对于python 3. x使用range而不是xrange

Answer 3

这应该可以利用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)

这基本上只是argparse上文档中的perfect\u square函数的一个改编示例。