如何在Python的argparse中对同一组参数调用两次parse_args()？

Question

简化的测试用例:我有一个带有两个参数的脚本.第一个是整数列表.第二个是一个整数,必须包含在第一个参数的整数集中.

例如:

$ python argtest.py --valid_nums 1 2 3 --num 2

应该工作,但是:

$ python argtest.py --valid_nums 1 2 3 --num 4

不应该工作,因为num不是in valid_nums.但是,我遇到了一些困难(阅读:我怀疑它比它的价值更麻烦,但我真的希望它能够工作)实现这个功能.

实施尝试:

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--valid_nums',
                    type=int,
                    nargs='+')

args = parser.parse_args()
print "Numbers: ", args.valid_nums


parser.add_argument('--num',
                    type=int,
                    choices=args.valid_nums)

args = parser.parse_args()
print args

实际产量:

$ python argtesttest.py --valid_nums 1 2 3 --num 2
usage: argtesttest.py [-h] [--valid_nums VALID_NUMS [VALID_NUMS ...]]
argtesttest.py: error: unrecognized arguments: --num 2

期望的输出:

$ python argtesttest.py --valid_nums 1 2 3 --num 2
Namespace(num=2, valid_nums=[1, 2, 3])

现在,(我认为)这里的问题是我在调用之后无法向解析器添加新参数parse_args(),这会产生关于无法识别的参数的错误,但我无法想到绕过它的方法.有没有办法打破论证的处理,使其不是一次全部？

显然,如果我只调用parse_args()一次并自己处理容器成员资格检查,这将是非常简单的事情,但我想让它"原生"使用argparse内置的错误检查.

有任何想法吗？

Answer 1

你需要使用parser.parse_known_args()而不是parser.parse_args(),或者在调用之前将所有参数添加到解析器中parse_args().

parse_args()需要了解所有当前的参数sys.argv,这些参数在--num首次解析时已经包含,因此是异常.

Answer 2

您可以使用命名空间和 parse_known_args 来完成此操作，并在最后调用 parse_args

class UserNamespace(object):
    pass
user_namespace = UserNamespace()

p = argparse.ArgumentParser() 
p.add_argument('-c', '--config', dest='config', default='')
p.parse_known_args(namespace=user_namespace)
if user_namespace.config == '':
    p.add_argument('-w', '--whatever', dest='whatever', default='')

parsed_args = p.parse_args(namespace=user_namespace)

对于具体情况：

num.py:

import argparse
class UserNamespace(object):
    pass
user_namespace = UserNamespace()

parser = argparse.ArgumentParser()
parser.add_argument('--valid_nums', dest='valid_nums',
                    type=int,
                    nargs='+')

parser.parse_known_args(namespace=user_namespace)

parser.add_argument('--num',
                    type=int,
                    choices=user_namespace.valid_nums)

args = parser.parse_args(namespace=user_namespace)

print "Numbers: ", user_namespace.valid_nums

所以你打电话：

$python num.py --valid_nums 1 2 3 --num 2  
Numbers:  [1, 2, 3]

$python num.py --valid_nums 1 2 3 --num 4  
usage: num.py [-h] [--valid_nums VALID_NUMS [VALID_NUMS ...]] [--num {1,2,3}]
num.py: error: argument --num: invalid choice: 4 (choose from 1, 2, 3)

参考资料： argparse 命名空间
argparse parse_known_args

再见！

2017 年 3 月 4 日更新 最好实例化 UserNamespace，否则其他功能（如 add_subparsers）无法正常工作。