python - 如何使 python 的 argparse 接受任意数量的 [-R a b]，并将它们聚合到一个列表中

Question

我希望能够调用 foo.py -R a b -R c d -R e f并得到类似 [('a', 'b'), ('c', 'd'), ('e', 'f')] 的内容在一个变量中。我可以使用 foo.py -R a=b -R c=d -R e=f并手动进行分割，但我不想这样做，因为我正在围绕另一个程序构建一个包装器，并且想模仿它的命令行选项输入格式。

我尝试过以下方法:

#!/usr/bin/env python
import argparse
parser = argparse.ArgumentParser(description='Foo')
parser.add_argument('-R', metavar=('A', 'B'), dest='libnames', type=str, default=('.', 'Top'), nargs=2)

if __name__ == '__main__':
    args = parser.parse_args()
    print(args.libnames)

但后来我得到 ['e', 'f']当我将其称为foo.py -R a b -R c d -R e f时.

Answer 1

您可以使用自定义argparse.Action类。

https://docs.python.org/3/library/argparse.html#argparse.Action

import argparse

class Pairs(argparse.Action):
    def __call__(self, parser, namespace, values, opts, **kwargs):
        lst = getattr(namespace, self.dest)
        if lst is None:
            lst = []
            setattr(namespace, self.dest, lst)
        lst.append(tuple(values))

parser = argparse.ArgumentParser()   
parser.add_argument('-R', nargs='+', dest='libnames', action=Pairs)
print parser.parse_args("-R a b -R c d -R e f".split())

输出:

Namespace(libnames=[('a', 'b'), ('c', 'd'), ('e', 'f')])