python - 调度Python的argparse子命令的最佳方式

我正在使用Python的argparse模块，以获得一个相当复杂且庞大的子命令结构。到目前为止，参数解析非常顺利，并且一切正常，但是我正在寻找一种更好的方法来管理子命令的执行方式。

这是我的虚拟/播放应用程序的示例：

def a_func(cmd_args):
    print(cmd_args)


def b_func(cmd_args):
    print(cmd_args)


CMD_DISPATCHER = {
    'a': a_func,
    'b': b_func
}


def parse_commands():
    # Create top-level parser
    parser = argparse.ArgumentParser()
    subparsers = parser.add_subparsers(dest='subcmd_name')

    # Create parser for the "a" sub-command
    parser_a = subparsers.add_parser('a')
    parser_a.add_argument('-bar', type=int)

    # Create parser for the "b" sub-command
    parser_b = subparsers.add_parser('b')
    parser_b.add_argument('-foo', type=int)

    args = parser.parse_args()

    CMD_DISPATCHER[args.subcmd_name](args)


def main():
    parse_commands()


if __name__ == '__main__':
    main()

如您所见，这里我使用一个简单的dict（CMD_DISPATCHER）作为将子命令名称与其功能关联的映射。

就像我说的那样，这只是我想要实现的一个简单示例，但是实际项目中将包含许多嵌套的子命令，因此我希望有一种更好的方法来管理这几个子命令。

您知道更好/更专业的方法吗？

最佳答案

好吧，经过更多的研究，我发现了一种适合我需求的好方法：set_defaults。我之前使用过此函数，但始终会设置默认的现有默认参数值。我没有注意到argparse documentation提供的明确示例：

>>> # sub-command functions
>>> def foo(args):
...     print(args.x * args.y)
...
>>> def bar(args):
...     print('((%s))' % args.z)
...
>>> # create the top-level parser
>>> parser = argparse.ArgumentParser()
>>> subparsers = parser.add_subparsers()
>>>
>>> # create the parser for the "foo" command
>>> parser_foo = subparsers.add_parser('foo')
>>> parser_foo.add_argument('-x', type=int, default=1)
>>> parser_foo.add_argument('y', type=float)
>>> parser_foo.set_defaults(func=foo)
>>>
>>> # create the parser for the "bar" command
>>> parser_bar = subparsers.add_parser('bar')
>>> parser_bar.add_argument('z')
>>> parser_bar.set_defaults(func=bar)
>>>
>>> # parse the args and call whatever function was selected
>>> args = parser.parse_args('foo 1 -x 2'.split())
>>> args.func(args)
2.0
>>>
>>> # parse the args and call whatever function was selected
>>> args = parser.parse_args('bar XYZYX'.split())
>>> args.func(args)
((XYZYX))

如您所见，行parser_bar.set_defaults(func=bar)将一个新变量设置为parser_bar参数。在这种情况下，bar是一个函数，最终将其用作args.func(args)行中的子命令执行程序。

我希望这对以后的人有所帮助。

关于python - 调度Python的argparse子命令的最佳方式，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/46022518/