python - 如何使用 argparse 在单独的命名空间中使用子解析器参数？

Question

我有以下测试代码

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)

我想要的是来自每个子解析器的值在一个单独的命名空间或字典中，类似于

Namespace(subparseargs={boo:'1', foo:'2'}, parser_name='car', verbose=3)

因此每个子解析器的参数与主解析器的参数在逻辑上是分开的(在这个例子中是verbose)。

如何使用相同命名空间(示例中的subparseargs)中的每个子解析器的参数实现这一点。

Answer 1

您需要深入了解 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()