Python认同互斥组
我需要的是:
pro [-a xxx | [-b yyy -c zzz]]
我试过这个,但不起作用。 有人能帮我吗?
group= parser.add_argument_group('Model 2') group_ex = group.add_mutually_exclusive_group() group_ex.add_argument("-a", type=str, action = "store", default = "", help="test") group_ex_2 = group_ex.add_argument_group("option 2") group_ex_2.add_argument("-b", type=str, action = "store", default = "", help="test") group_ex_2.add_argument("-c", type=str, action = "store", default = "", help="test")
谢谢!
add_mutually_exclusive_group
不会使整个组互斥。 它使组内的选项相互排斥。
你要找的是子命令 。 而不是编[-a xxxx | [-b yyy -c zzz]],你会有:
prog command 1 -a: ... command 2 -b: ... -c: ...
用第一组参数调用:
prog command_1 -a xxxx
用第二组参数调用:
prog command_2 -b yyyy -c zzzz
您也可以将sub命令参数设置为位置。
prog command_1 xxxx
有点像git或svn:
git commit -am git merge develop
工作示例
# create the top-level parser parser = argparse.ArgumentParser(prog='PROG') parser.add_argument('--foo', action='store_true', help='help for foo arg.') subparsers = parser.add_subparsers(help='help for subcommand') # create the parser for the "command_1" command parser_a = subparsers.add_parser('command_1', help='command_1 help') parser_a.add_argument('a', type=str, help='help for bar, positional') # create the parser for the "command_2" command parser_b = subparsers.add_parser('command_2', help='help for command_2') parser_b.add_argument('-b', type=str, help='help for b') parser_b.add_argument('-c', type=str, action='store', default='', help='test')
testing它
>>> parser.print_help() usage: PROG [-h] [--foo] {command_1,command_2} ... positional arguments: {command_1,command_2} help for subcommand command_1 command_1 help command_2 help for command_2 optional arguments: -h, --help show this help message and exit --foo help for foo arg. >>> >>> parser.parse_args(['command_1', 'working']) Namespace(a='working', foo=False) >>> parser.parse_args(['command_1', 'wellness', '-b x']) usage: PROG [-h] [--foo] {command_1,command_2} ... PROG: error: unrecognized arguments: -bx
祝你好运。
虽然Jonathan的答案对于复杂的选项是完全正确的,但是有一个非常简单的解决scheme可以用于简单的情况,例如1个选项不包括2个其他选项
command [- a xxx | [ -b yyy | -c zzz ]]
甚至在原来的问题中:
pro [-a xxx | [-b yyy -c zzz]]
这是我将如何做到这一点:
parser = argparse.ArgumentParser() # group 1 parser.add_argument("-q", "--query", help="query", required=False) parser.add_argument("-f", "--fields", help="field names", required=False) # group 2 parser.add_argument("-a", "--aggregation", help="aggregation", required=False)
我在这里使用给命令行包装查询一个mongodb的选项。 collection
实例可以调用方法aggregate
或方法find
与可选参数query
和fields
,因此你看到为什么前两个参数是兼容的,最后一个不是。
所以现在我运行parser.parse_args()
并检查它的内容:
args = parser().parse_args() print args.aggregation if args.aggregation and (args.query or args.fields): print "-a and -q|-f are mutually exclusive ..." sys.exit(2)
当然,这个小黑客只是在简单的情况下工作,如果你有很多互斥的选项和组,它将成为一个噩梦来检查所有可能的选项。 在这种情况下,你应该把自己的select权交给乔纳森build议的指挥组。
有一个python补丁(在开发中),可以让你做到这一点。
http://bugs.python.org/issue10984
这个想法是允许重叠的互斥组。 所以usage
可能如下所示:
pro [-a xxx | -b yyy] [-a xxx | -c zzz]
改变argparse代码,所以你可以创build这样的两个组是很容易的部分。 更改usage
格式代码需要编写一个自定义的HelpFormatter
。
在argparse
,操作组不会影响parsing。 他们只是一个help
格式化工具。 在help
,互斥组只影响usage
线。 parsing时, parser
使用相互排斥的组来构造潜在冲突的字典( a
不能与b
或c
发生, b
不能和a
等发生),如果发生冲突则会引发错误。
如果没有这个argparse补丁,我认为你最好的select是自己testing由parse_args
生成的命名空间(例如,如果a
和b
都有非默认值),并且引发你自己的错误。 你甚至可以使用parsing器自己的错误机制。
parser.error('custom error message')