Python - 规避 argparse nargs 错误

Question

我有一个像这样工作的程序：

prog.py filename -r

使用我给的默认默认值

prog.py filename -r 0 500 20

使用0, 500 and 20

我已经设法实现了这一点：

class RdistAction(argparse.Action):
    def __call__(self,parser,namespace,values,option_string=None):
        if not values:
            setattr(namespace,self.dest,[0, 1000, 50])
        else:
            setattr(namespace,self.dest,values)

parser = argparse.ArgumentParser()
parser.add_argument("-r", "--rdist", action=RdistAction, nargs='*', type=int)
args = parser.parse_args()

但我想固执一点，因为我最初的目标是将 nargs 设置为 3。但是当我在上面的代码中使用 nargs=3 时，我收到一条错误消息，指出需要 3 个参数。

我用谷歌搜索了一下，结果我的直觉告诉我，我必须添加def __init__ 并修改该函数中的某些内容。是否有可能在nargs='*' 时得到与上面代码相​​同的结果，但使用nargs=3 时得到的结果？

Answer 1

如果我将您的 add_argument 行更改为 nargs='3'，我想我得到了您所犯的错误：

Traceback (most recent call last):
  File "python", line 12, in <module>
ValueError: length of metavar tuple does not match nargs

如果我设置nargs=3（不带引号），那么它对我有用：

import argparse

class RdistAction(argparse.Action):
    def __call__(self,parser,namespace,values,option_string=None):
      if not values:
        setattr(namespace,self.dest,[0, 1000, 50])
      else:
        setattr(namespace,self.dest,values)

parser = argparse.ArgumentParser()
parser.add_argument("-r", "--rdist", action=RdistAction, nargs=3, type=int)
print parser.parse_args('-r 0 500 20'.split())

给予

Namespace(rdist=[0, 500, 20])

这就是你要找的吗？

这里的诀窍是，如果 nargs 不是特殊字符之一（'?'、'*'、'+' 等），那么它必须是整数，而不是字符串。

请注意，the documentation 确实指出了这一点。

Answer 2

正如我在上一个问题中指出的那样，只需省略 -r 即可将 default 放入 args 命名空间中。你不必做任何特别的事情。

nargs='?' 有第三个选项，const。

您的自定义操作有点像这样，当-r 不带任何字符串使用时，它会为命名空间分配一个自定义值。

解析大致如下：

• 为命名空间分配默认值
• 依次循环输入、处理位置和选项
• 对于每个参数，调用它的Action.__call__
• values 传递的数量取决于nargs 参数

您的自定义RdistAction 无法控制values 的数量，除非通过其nargs 属性。 nargs=3 表示正好 3，nargs='*' 表示“尽可能多地给我”（即 0 或到末尾或到下一个标志）。

在optparse 中，Action 给出了所有剩余的argv 字符串，并且可以根据需要使用任意数量的字符串。但在argparse 中，分配是在更高级别完成的，Action 使用它给定的内容。

我在 Python 错误/问题中探索了扩展 nargs 以包含 regex 之类的范围，例如[0,3]。但这需要更改解析器的核心。

你还没有解释为什么需要这个 -r 选项。