4

我试图弄清楚如何使用 argparser 来执行以下操作：

$ python test.py executeBuild --name foobar1 executeBuild --name foobar2 ....

getBuild本身就是一个子命令。我的目标是让脚本能够链接一系列子命令（executeBuild作为其中之一）并按顺序执行它们。在上面的示例中，它将执行构建，然后设置环境，然后再次执行构建。如何使用 argparse 完成此操作？我尝试了以下方法：

    main_parser = argparse.ArgumentParser(description='main commands')
    subparsers = main_parser.add_subparsers(help='SubCommands', dest='command')

    build_parser = subparsers.add_parser('executeBuild')
    build_parser.add_argument('--name', action='store', nargs=1, dest='build_name')
    check_parser = subparsers.add_parser('setupEnv')

    args, extra=main_parser.parse_known_args()

但是，似乎每当我这样做时，它都会进入子命令executeBuild并报告它不知道是什么executeBuild。我已经尝试解析额外的，以便我可以重复调用/链，但是，第一个视图属性似乎已被覆盖，所以我什至不能只保存额外的选项并迭代。

4

2 回答 2

4

您在问argparse一些不是为它编写的东西：它擅长解析一个命令行（但只有一个），并且您想在一行中解析多个命令。恕我直言，您必须对参数数组进行初始拆分，然后argparse在每个子命令上使用。下面的函数接受一个参数列表（可能是sys.argv），跳过第一个参数并从每个已知子命令开始拆分剩余的数组。然后，您可以在每个子列表上使用 argparse ：

def parse(args, subcommands):
    cmds = []
    cmd = None
    for arg in args[1:]:
        if arg in (subcommands):
            if cmd is not None:
                cmds.append(cmd)
            cmd = [arg]
        else:
            cmd.append(arg)
    cmds.append(cmd)
    return cmds

在您的示例中：

parse(['test.py', 'executeBuild', '--name', 'foobar1', 'executeBuild', '--name', 'foobar2'],
    ('executeBuild',))

=>

[['executeBuild', '--name', 'foobar1'], ['executeBuild', '--name', 'foobar2']]

限制：子命令用作保留字，不能用作选项参数。

于 2014-06-30T08:11:04.193 回答
2

事先拆分sys.argv是一个很好的解决方案。但它也可以在使用带有 的参数进行解析时完成nargs=argparse.REMAINDER。这种类型的参数获取其余的字符串，无论它们是否看起来像标志。

parse_known_args用以下代码替换：

...
build_parser.add_argument('rest', nargs=argparse.REMAINDER)
check_parser.add_argument('rest', nargs=argparse.REMAINDER)
extras = 'executeBuild --name foobar1 setupEnv executeBuild --name foobar2'.split()
# or extras = sys.argv[1:]
while extras:
    args = main_parser.parse_args(extras)
    extras = args.rest
    delattr(args,'rest')
    print args
    # collect args as needed

印刷：

Namespace(build_name=['foobar1'], command='executeBuild')
Namespace(command='setupEnv')
Namespace(build_name=['foobar2'], command='executeBuild')

在文档中：

argparse.REMAINDER。所有剩余的命令行参数都被收集到一个列表中。这对于调度到其他命令行实用程序的命令行实用程序通常很有用：

一个问题REMAINDER是它可能过于贪婪。http://bugs.python.org/issue14174。因此build_parser，check_parser不能有其他位置参数。

绕过贪婪的方法REMAINDER是使用argparse.PARSER. 这是使用的nargs值subparsers（未记录）。就像REMAINDER，除了第一个字符串必须看起来像一个“参数”（没有“-”），并且匹配choices（如果给定）。 PARSER不像 贪婪REMAINDER，因此子解析器可以有其他位置参数。

有一些额外的代码涉及“退出”字符串和虚拟解析器。这是为了解决这个PARSER论点是“必需”的事实（有点像nargs='+'）

from argparse import ArgumentParser, PARSER, SUPPRESS

main_parser = ArgumentParser(prog='MAIN')
parsers = {'exit': None}
main_parser.add_argument('rest',nargs=PARSER, choices=parsers)

build_parser = ArgumentParser(prog='BUILD')
parsers['executeBuild'] = build_parser
build_parser.add_argument('cmd')
build_parser.add_argument('--name', action='store', nargs=1, dest='build_name')
build_parser.add_argument('rest',nargs=PARSER, choices=parsers, help=SUPPRESS)

check_parser = ArgumentParser(prog='CHECK')
parsers['setupEnv'] = check_parser
check_parser.add_argument('cmd')
check_parser.add_argument('foo')
check_parser.add_argument('rest',nargs=PARSER, choices=parsers, help=SUPPRESS)

argv = sys.argv[1:]
if len(argv)==0:
    argv = 'executeBuild --name foobar1 setupEnv foo executeBuild --name foobar2'.split()
argv.append('exit') # extra string to properly exit the loop
parser = main_parser
while parser:
    args = parser.parse_args(argv)
    argv = args.rest
    delattr(args,'rest')
    print(parser.prog, args)
    parser = parsers.get(argv[0], None)

样本输出：

('MAIN', Namespace())
('BUILD', Namespace(build_name=['foobar1'], cmd='executeBuild'))
('CHECK', Namespace(cmd='setupEnv', foo='foo'))
('BUILD', Namespace(build_name=['foobar2'], cmd='executeBuild'))

另一种可能性是用于'--'分隔命令块：

'executeBuild --name foobar1 -- setupEnv -- executeBuild --name foobar2'

但是有几个时会出现问题：http '--': //bugs.python.org/issue13922

于 2014-06-30T16:40:17.967 回答