如何将一个参数传递给会影响两个选项的参数?



假设我有以下解析器:

def parse():
p = ArgumentParser()
p.add_argument('-c', 'client', help='')
p.add_argument('-s', 'server', help='')
args = vars(p.parse_args())
return args

我必须用python my_script.py -s bla -c bla来称呼它.我想添加第三个选项,它将结合它们,允许我调用python my_script.py -x bla并从函数中获取{'s': 'bla, 'c': 'bla'},就像我从第一次调用中获得的那样。

我该怎么做?


我可以做相反的事情,即创建 2 个具有相同名称的参数:

In [14]: p.add_argument('--x', dest='z') 
...: p.add_argument('--y', dest='z') 
...: p.parse_args('--x 1 --y 2'.split())                                                                                                
Out[14]: Namespace(z='2')

但这并没有真正帮助我...


这不是重复。我不想传递列表。我已经有 2 个选项,我希望他们留下来。我只想有第三个选项将它们结合起来。使用列表会让我重复输入,就像我想要python my_script.py --x blapython my_script.py --list_param bla bla

argparse中没有直接支持您想要的内容,但您可以通过注册自定义操作来获取它:

class BothAction(argparse.Action):
def __call__(self, parser, namespace, val, option_string=None):
setattr(namespace, "client", val)
setattr(namespace, "server", val)
del namespace.x
...
parser.register("action", "both", BothAction)
parser.add_argument('-x', action="both")

这可以打磨,但我希望你明白基本的想法。