如何禁止两个相互冲突的选择

时间:2016-04-30 08:49:05

标签: python command-line-arguments argparse

有没有办法为Python的ArgumentParser指定两个可选标志是冲突的?

arg_parser.add_argument('-c', '--clean', action='store_true')
arg_parser.add_argument('-d', '--dirty', action='store_true')

我希望用户能够既不指定也不指定一个。

没有其他条件是否可以实现?

1 个答案:

答案 0 :(得分:13)

如何添加mutually exclusive group

group = arg_parser.add_mutually_exclusive_group()
group.add_argument('-c', '--clean', action='store_true')
group.add_argument('-d', '--dirty', action='store_true')

有了这个我得到以下行为:

>>> arg_parser.parse_args(['--clean']) 
Namespace(clean=True, dirty=False)
>>> arg_parser.parse_args(['--dirty']) 
Namespace(clean=False, dirty=True)
>>> arg_parser.parse_args(['--dirty','--clean']) 
usage: PROG [-h] [-c | -d] PROG: error: argument -c/--clean: not allowed with argument -d/--dirty