如何正确使用argparse subparsers？

Question

我一直在搜索这里和一般的子分析器示例,但似乎无法想象这个看似简单的事情.

我有两个var类型,其中一个有约束,所以认为subparser是要走的路.例如-t允许"A"或"B".如果用户通过"A",那么他们还需要指定它是"a1"还是"a2".如果他们只通过"B"则没有.

我可以这样做并让argparse返回我通过什么类型的"A"或者它只是"B"？

下面似乎工作但由于某种原因在subparse后传递任何东西时中断.

例如来自Linux终端

>> python test01.py -t A a1 -v 61

错误......

usage: test01.py a1 [-h]
test01.py a1: error: unrecognized arguments: -v

希望这是有道理的.

代码:

import argparse

parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers(help='types of A')

parser.add_argument("-t",
                    choices = ["A", "B"],
                    dest = "type",
                    required=True,
                    action='store',
                    help="Some help blah blah")

cam_parser = subparsers.add_parser('a1', help='Default')
cam_parser.set_defaults(which='a1')

cam_parser = subparsers.add_parser('a2', help='parse this instead of default')
cam_parser.set_defaults(which='a2')


parser.add_argument("-v",
                    nargs = '+',
                    required=True,
                    dest = "version",
                    type=int,
                    action='store',
                    help="some version help blah blah")   

argument = parser.parse_args()

print argument.type
print argument.version

Answer 1

子分类符是根据第一个位置参数的值调用的,因此您的调用看起来像

python test01.py A a1 -v 61

"A"触发相应的subparser,它将被定义为允许位置参数和-v选项.

因为argparse不会对参数和选项可能出现的顺序施加任何限制,并且没有简单的方法来修改解析开始后可能出现的参数/选项(涉及修改解析器实例的自定义操作可能有效),你应该考虑更换-t自己:

import argparse

parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers(help='types of A')
parser.add_argument("-v", ...)

a_parser = subparsers.add_parser("A")
b_parser = subparsers.add_parser("B")

a_parser.add_argument("something", choices=['a1', 'a2'])

由于-v是为主解析器定义的,因此必须在参数之前指定它,以指定哪个子解析器用于其余参数.