在Python中，使用argparse仅允许正整数
在Python中，使用argparse仅允许正整数
提问于 2013-01-02 13:47:22
标题很好地概括了我希望发生的事情。
这是我所拥有的，虽然程序不会在非正整数上崩溃，但我想让用户知道非正整数基本上是无意义的。
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)回答 3
Stack Overflow用户
回答已采纳
发布于 2013-01-02 13:58:05
使用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上docs中的perfect_square函数的示例。
Stack Overflow用户
发布于 2013-01-02 14:05:20
如果您arg有一个可预测的最大值和最小值，那么最快和最坏的方法就是使用带有范围的choices
parser.add_argument('foo', type=int, choices=xrange(0, 1000))Stack Overflow用户
发布于 2021-12-16 16:53:49
根据Yuushi的回答，您还可以定义一个简单的helper函数，该函数可以检查各种数值类型的数字是否为正：
def positive(numeric_type):
def require_positive(value):
number = numeric_type(value)
if number <= 0:
raise ArgumentTypeError(f"Number {value} must be positive.")
return number
return require_positivehelper函数可用于注释任何数值参数类型，如下所示：
parser = argparse.ArgumentParser(...)
parser.add_argument("positive-integer", type=positive(int))
parser.add_argument("positive-float", type=positive(float))页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/14117415
复制相关文章
相似问题