假设我有以下解析器:
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 bla
python 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")
这可以打磨,但我希望你明白基本的想法。