是否可以在不使用子解析器的情况下表示一组 argparse 的参数？

Question

我有一个argparser，有两个参数子集，输入和参数。我的命令看起来像

program -input1 <input1> -input2 <input2> -p1 <param1> -p2 <param2>

我愿意

args = parser.parse_args()
params = vars(args.params)

而不是

params = {"p1": args.p1, "p2": args.p2, etc...}

有没有一种优雅的方式来做类似的事情：

parser.add_argument("-p1", dest='p1', part_of=params)

Subparsers 似乎不是为此而生的。还是他们？

Argparse 有argument groups，但它们似乎只是用于帮助文本。

谢谢！

Answer 1

这听起来像是一个变体 - 我如何接受任意的“键=值”对？多年来，人们以各种方式提出了这个问题，并给出了各种答案。

最近的一篇：

Parsing "python foo.py -DVAR1=9 -DVAR2=Off" with argparse

我的解决方案是将-DVAR1=9 拆分为('VAR1',9) 并将其附加到D 属性中。使用自定义type。

Using variable arg names with argparse - 在将sys.argv 传递给解析器之前对其进行处理。

python argparse store --foo=bar as args.key='foo', args.value='bar'

建议自定义 Action 类。

我认为我们还建议了一个自定义 Namespace 类。

用于分组值的内置机制是nargs 和append 操作。您可以一起获取列表列表的属性。

JSON 字符串也可用于输入复杂的数据结构。

---

class MyAction(argparse._StoreAction):
    def __call__(self, parser, namespace, values, option_string=None):
        print('storing', option_string)
        arg = getattr(namespace, self.dest)
        if arg is None:
            arg = {}
        arg[option_string] = values
        setattr(namespace, self.dest, arg)

In [135]: p=argparse.ArgumentParser()
In [136]: p.add_argument('--p1',action=MyAction,dest='p');
In [137]: p.add_argument('--p2',action=MyAction,dest='p');
In [138]: p.parse_args('--p1 one --p2 two'.split())
storing --p1
storing --p2
Out[138]: Namespace(p={'--p2': 'two', '--p1': 'one'})
In [139]: _.p
Out[139]: {'--p1': 'one', '--p2': 'two'}

显然，这可以通过各种方式进行优化 - 将键修剪为“p1”甚至“1”、保存到列表、嵌套命名空间或其他自定义结构等。

这种方法仍然需要您为每个“--pn”变体定义一个add_argument。

另一种方法是保持argparse 简单，生成如下命名空间：

In [141]: argparse.Namespace(p1='one', p2='two', input1=1, input2=3)
Out[141]: Namespace(input1=1, input2=3, p1='one', p2='two')

然后进行自己的分组。

In [142]: args=argparse.Namespace(p1='one', p2='two', input1=1, input2=3)
In [143]: {key:getattr(args,key) for key in ['p1','p2']}
Out[143]: {'p1': 'one', 'p2': 'two'}