如何使用 nargs+ 和子命令进行 argparse

Question

我正在尝试创建一个类似的命令

prog [-h] [-i ID [ID ...]] | -x [SOMETHING] 
     {cmd1,cmd2,cmd3}...

所以基本上在顶层，我有一个解析器，它对 -i 和 -x 选项有一个互斥组，然后按照这些（可能还有其他）选项，我有一个要运行的命令。每个命令都有自己使用的一组选项。我可以使用add_subparsers() 使命令正常工作，但我遇到的问题是当我尝试向具有nargs='+' 的根解析器添加参数时。当我这样做时，它会吞下-i 的所有参数，认为该命令是一个参数而不是一个 ID。

有没有办法解决这个问题？似乎它必须查看 -i 的参数以查找命令字，然后告诉 argparse 它应该在该点恢复解析。

Answer 1

我不得不多次阅读您的描述，但我认为这是问题所在：

prog -i id1 id2 cmd1 -foo 3 ....

它给出了一些关于找不到{cmd1,cmd2,cmd3} 的警告。确切的错误可能会有所不同，因为在某些版本中 subparsers 实际上并不需要。

在任何情况下，-i 的参数都是 ['id1','id2','cmd1']，直到下一个 - 标志。对于主解析器，子解析器参数只是另一个positional 一个（带有choices）。将字符串分配给-i 时，它不会检查字符串是否与cmds 之一匹配。它只是看它是否以-开头。

在上下文中使用nargs='+'（或“*”）的唯一方法是包含其他一些标记的参数，例如

prog -i id1 id2 -x 3 cmd1 --foo ...

我知道这与您的 mutually_exclusive 群组不符。

基本点是非标志字符串是根据位置分配的，而不是值。对于变量nargs，您必须有某种明确的列表终止符。

从侧边栏

Argparse nargs="+" is eating positional argument

除了你的下一个位置是 subparsers cmd 之外，它是相似的。

==============

带有 '+' 的位置将在子解析器 cmd 之前工作

usage: prog [-h] foo [foo ...] {cmd1,cmd2} ...

In [160]: p1.parse_args('1 22 3 cmd1'.split())
Out[160]: Namespace(cmd='cmd1', foo=['1', '22', '3'])

但这是因为foo 和cmd 的字符串分配有一个regex 模式测试。

在

usage: prog [-h] [--bar BAR [BAR ...]] {cmd1,cmd2} ...

字符串被分配给bar，而不考虑以下位置cmd的需要。如http://bugs.python.org/issue9338 的建议补丁中所示，更改此行为并非易事。它需要一个额外的前瞻性试错循环。