python - 如何将一个参数传递给 argparse 来影响两个选项？

Question

假设我有以下解析器:

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')

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

<小时/>

这不是重复。我不想传递一份 list 。我已经有两个选择，我希望它们保留。我只是想要将它们结合起来的第三种选择。使用列表会让我重复我的输入，就像这样 python my_script.py --list_param bla bla我想去的地方python my_script.py --x bla

Answer 1

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")

这可以改进，但我希望你能了解基本的想法。