我觉得 argparse 中的 sub_parsers 有一点限制,如果说,你有一套工具可能有类似的选项,这些选项可能会分布在不同的级别。这种情况可能很少见,但如果您正在编写可插入/模块化的代码,它可能会发生。
我有以下示例。这是牵强的,目前还没有很好地解释,因为它已经很晚了,但它是这样的:
Usage: tool [-y] {a, b}
a [-x] {create, delete}
create [-x]
delete [-y]
b [-y] {push, pull}
push [-x]
pull [-x]
from argparse import ArgumentParser
parser = ArgumentParser()
parser.add_argument('-x', action = 'store_true')
parser.add_argument('-y', action = 'store_true')
subparsers = parser.add_subparsers(dest = 'command')
parser_a = subparsers.add_parser('a')
parser_a.add_argument('-x', action = 'store_true')
subparsers_a = parser_a.add_subparsers(dest = 'sub_command')
parser_a_create = subparsers_a.add_parser('create')
parser_a_create.add_argument('-x', action = 'store_true')
parser_a_delete = subparsers_a.add_parser('delete')
parser_a_delete.add_argument('-y', action = 'store_true')
parser_b = subparsers.add_parser('b')
parser_b.add_argument('-y', action = 'store_true')
subparsers_b = parser_b.add_subparsers(dest = 'sub_command')
parser_b_create = subparsers_b.add_parser('push')
parser_b_create.add_argument('-x', action = 'store_true')
parser_b_delete = subparsers_b.add_parser('pull')
parser_b_delete.add_argument('-y', action = 'store_true')
print parser.parse_args(['-x', 'a', 'create'])
print parser.parse_args(['a', 'create', '-x'])
print parser.parse_args(['b', '-y', 'pull', '-y'])
print parser.parse_args(['-x', 'b', '-y', 'push', '-x'])
输出
Namespace(command='a', sub_command='create', x=True, y=False)
Namespace(command='a', sub_command='create', x=True, y=False)
Namespace(command='b', sub_command='pull', x=False, y=True)
Namespace(command='b', sub_command='push', x=True, y=True)
如您所见,很难区分每个参数设置在链的哪个位置。您可以通过更改每个变量的名称来解决此问题。例如,您可以将 'dest' 设置为 'x'、'a_x'、'a_create_x'、'b_push_x' 等,但这会很痛苦并且很难分开。
另一种方法是让 ArgumentParser 在到达子命令后停止并将剩余的参数传递给另一个独立的解析器,以便它可以生成单独的对象。您可以尝试通过使用 'parse_known_args()' 而不是为每个子命令定义参数来实现这一点。但是,这并不好,因为之前任何未解析的参数仍然存在并且可能会混淆程序。
我觉得一个稍微便宜但有用的解决方法是让 argparse 将以下参数解释为列表中的字符串。这可以通过将前缀设置为空终止符“\0”(或其他一些“难以使用”字符)来完成——如果前缀为空,代码将抛出错误,至少在 Python 2.7 中是这样。 3.
例子:
parser = ArgumentParser()
parser.add_argument('-x', action = 'store_true')
parser.add_argument('-y', action = 'store_true')
subparsers = parser.add_subparsers(dest = 'command')
parser_a = subparsers.add_parser('a' prefix_chars = '\0')
parser_a.add_argument('args', type = str, nargs = '*')
print parser.parse_args(['-xy', 'a', '-y', '12'])
输出:
Namespace(args=['-y', '12'], command='a', x=True, y=True)
请注意,它不使用第二个-y
选项。然后,您可以将结果“args”传递给另一个 ArgumentParser。
缺点:
- 帮助可能处理不好。将不得不对此做出更多解决方法
- 遇到的错误可能很难追踪,并且需要一些额外的努力来确保错误消息被正确链接。
- 与多个 ArgumentParsers 相关的开销要多一些。
如果有人对此有更多意见,请告诉我。