[英]How to have sub-parser arguments in separate namespace with argparse?
我有以下测试代码
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("--verbose", default = 0, type=int)
subparsers = parser.add_subparsers(dest = "parser_name")
parser_lan = subparsers.add_parser('car')
parser_lan.add_argument("--boo")
parser_lan.add_argument("--foo")
parser_serial = subparsers.add_parser('bus')
parser_serial.add_argument("--fun")
print parser.parse_args()
它定义了两个子解析器,具有不同的参数集。 当我把测试代码称为
tester.py --verbose 3 car --boo 1 --foo 2
我得到了预期的结果
Namespace(boo='1', foo='2', parser_name='car', verbose=3)
我想要的是来自每个subparser的值在一个单独的命名空间或dict中,类似于
Namespace(subparseargs={boo:'1', foo:'2'}, parser_name='car', verbose=3)
这样每个subparser的参数与主解析器的参数逻辑分离(在本例中为verbose
)。
如何在同一名称空间中使用每个子分析器的参数( subparseargs
中的subparseargs
)来实现此subparseargs
。
你需要进入argparse
的一些内容,但是将脚本更改为以下内容应该可以解决问题:
import argparse
from argparse import _HelpAction, _SubParsersAction
class MyArgumentParser(argparse.ArgumentParser):
def parse_args(self, *args, **kw):
res = argparse.ArgumentParser.parse_args(self, *args, **kw)
from argparse import _HelpAction, _SubParsersAction
for x in parser._subparsers._actions:
if not isinstance(x, _SubParsersAction):
continue
v = x.choices[res.parser_name] # select the subparser name
subparseargs = {}
for x1 in v._optionals._actions: # loop over the actions
if isinstance(x1, _HelpAction): # skip help
continue
n = x1.dest
if hasattr(res, n): # pop the argument
subparseargs[n] = getattr(res, n)
delattr(res, n)
res.subparseargs = subparseargs
return res
parser = MyArgumentParser()
parser.add_argument("--verbose", default = 0, type=int)
subparsers = parser.add_subparsers(dest = "parser_name")
parser_lan = subparsers.add_parser('car')
parser_lan.add_argument("--boo")
parser_lan.add_argument("--foo")
parser_serial = subparsers.add_parser('bus')
parser_serial.add_argument("--fun")
print parser.parse_args()
我已经开始开发一种不同的方法(但类似于Anthon的建议)并提出了更短的代码。 但是,我不确定我的方法是解决问题的一般方法。
与Anthon提出的类似,我定义了一个新方法,它创建了一个“顶级”参数列表,这些参数保存在args
,而所有其他参数都作为附加字典返回:
class MyArgumentParser(argparse.ArgumentParser):
def parse_subargs(self, *args, **kw):
# parse as usual
args = argparse.ArgumentParser.parse_args(self, *args, **kw)
# extract the destination names for top-level arguments
topdest = [action.dest for action in parser._actions]
# loop over all arguments given in args
subargs = {}
for key, value in args.__dict__.items():
# if sub-parser argument found ...
if key not in topdest:
# ... remove from args and add to dictionary
delattr(args,key)
subargs[key] = value
return args, subargs
对这种方法的评论很受欢迎,尤其是我忽略的任何漏洞。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.