首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >Python argparse互斥组

Python argparse互斥组
EN

Stack Overflow用户
提问于 2013-07-28 22:37:18
回答 4查看 60.5K关注 0票数 103

我需要的是:

代码语言:javascript
复制
pro [-a xxx | [-b yyy -c zzz]]

我试过了,但不起作用。有谁能帮帮我吗?

代码语言:javascript
复制
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")

谢谢!

EN

回答 4

Stack Overflow用户

回答已采纳

发布于 2013-07-28 22:59:21

add_mutually_exclusive_group不会使整个组相互排斥。它使组内的选项相互排斥。

你要找的是subcommands。不是prog [ yyy xxxx | -b yyy -c zzz],而是:

代码语言:javascript
复制
prog 
  command 1 
    -a: ...
  command 2
    -b: ...
    -c: ...

使用第一组参数调用:

代码语言:javascript
复制
prog command_1 -a xxxx

使用第二组参数调用:

代码语言:javascript
复制
prog command_2 -b yyyy -c zzzz

也可以将子命令参数设置为位置参数。

代码语言:javascript
复制
prog command_1 xxxx

有点像git或svn:

代码语言:javascript
复制
git commit -am
git merge develop

工作示例

代码语言:javascript
复制
# 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')

测试它

代码语言:javascript
复制
>>> 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: -b x

祝好运。

票数 116
EN

Stack Overflow用户

发布于 2014-12-28 18:38:16

虽然Jonathan's answer完全适用于复杂的选项,但对于简单的情况,有一个非常简单的解决方案,例如,1个选项排除了2个其他选项,如

代码语言:javascript
复制
command [- a xxx | [ -b yyy | -c zzz ]] 

或甚至如原问题所述:

代码语言:javascript
复制
pro [-a xxx | [-b yyy -c zzz]]

下面是我会怎么做:

代码语言:javascript
复制
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,也可以调用带有可选参数queryfields的方法find,因此您可以看到为什么前两个参数兼容,而最后一个参数不兼容。

所以现在我运行parser.parse_args()并检查它的内容:

代码语言:javascript
复制
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)

当然,这个小技巧只适用于简单的情况,如果你有许多互斥的选项和组,检查所有可能的选项将成为一场噩梦。在这种情况下,您应该像Jonathan建议的那样,将您的选项分解到命令组中。

票数 39
EN

Stack Overflow用户

发布于 2013-07-30 15:19:53

有一个python补丁(正在开发中)可以让你做到这一点。

http://bugs.python.org/issue10984

这个想法是为了允许重叠的互斥的组。因此,usage可能如下所示:

代码语言:javascript
复制
pro [-a xxx | -b yyy] [-a xxx | -c zzz]

更改argparse代码以便可以创建两个这样的组是比较容易的部分。要更改usage格式化代码,需要编写自定义HelpFormatter

argparse中,操作组不会影响解析。它们只是一个help格式化工具。在help中,互斥组仅影响usage线路。在解析时,parser使用互斥组来构建潜在冲突的字典(a不能出现在bc中,b不能出现在a中,等等),如果出现冲突,就会抛出错误。

如果没有这个argparse补丁,我认为您最好的选择是自己测试parse_args生成的名称空间(例如,如果ab都有非默认值),并引发您自己的错误。您甚至可以使用解析器自己的错误机制。

代码语言:javascript
复制
parser.error('custom error message')
票数 4
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/17909294

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档