命令行参数归属于多个参数

Question

我正在尝试构建一个命令行解析器，它能够在参数之间共享传递的值，以避免必须多次键入它们。否则，我希望两个参数的名称空间相同：

import argparse

class PrintAction(argparse.Action):
    def __init__(self, option_strings, dest, **kwargs):
        super(PrintAction, self).__init__(option_strings, dest, **kwargs)

    def __call__(self, parser, namespace, values, option_string=None):
        for val in values:
            print(val)

parser = argparse.ArgumentParser(description='A foo that foos and a bar that bars')
parser.add_argument('--foo', action=PrintAction)
parser.add_argument('bar', nargs='+')

args = parser.parse_args(['--foo', 'a', 'b', 'c']) # Case 1
args = parser.parse_args(['a', 'b', 'c']) # Case 2

然后我想要一个解决方案，在 bar 两种情况下存储 ['a', 'b', 'c']，但在提供 --foo 的情况下，然后将打印 a、b 和 c。

目前，我得到的是 foo 仅打印 a 和 bar 仅将 b 和 c 存储在 case 1 中，正确的结果存储在 case 2 中。

Answer 1

您需要将--foo 设为布尔标志。现在它是一个字符串参数，因为您没有另外声明。将action 设置为store_true 以获得布尔标志效果。

最终的解决方案如下：

def print_args(args):
    if args.foo:
        for val in args.bar:
            print(val)

parser = argparse.ArgumentParser(description='A foo that foos and a bar that bars')
parser.add_argument('--foo', action='store_true')
parser.add_argument('bar', nargs='+')

args = parser.parse_args(['--foo', 'a', 'b', 'c']) # Case 1
args = parser.parse_args(['a', 'b', 'c']) # Case 2

然后在第一种情况下调用print_args(args) 将打印a、b 和c，而在第二种情况下则不会。

Answer 2

您不能（轻易）欺骗argparse 重用argv 字符串。解析器为动作分配值。

默认的nargs 是None，这意味着使用下一个字符串作为参数。

parser.add_argument('--foo')

将设置foo='a' 和bar=['b','c']。

在您的操作中，values 将是您打印的['a']。在optparse 中，每个选项获取剩余的argv 列表，它可以根据需要使用它。在argparse 中，它只得到nargs 要求的values。

您可以在__init__ 中指定nargs=0，然后从sys.argv 打印。同样，正如@9000 建议的那样，将其设为store_true 并在解析后打印。查看store_true Action 类的代码。

另一个选项是给foo 和bar 一个*，并让foo 打印并保存到bar dest。然后foo 将消耗所有以下字符串。但是，如果bar 没有要保存的内容，它可能会将[] 写入命名空间。

无论如何，你能做的最好的就是伪造重复使用。

---

另一个想法是使用 2 个不同的解析器和 parse_known_args。解析器不会混淆sys.argv，因此它可以读取和解析多次。