如何使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')])