如何使python的argparse接受任意数量的[-R ab]，并将其聚合到列表中

Question

我希望能够调用foo.py -R ab -R cd -R ef并得到类似[('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)

但是当我将其foo.py -R ab -R cd -R ef为foo.py -R ab -R cd -R ef时['e', 'f']我得到了['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')])