20

我有以下代码(使用 Python 2.7):

# shared command line options, like --version or --verbose
parser_shared = argparse.ArgumentParser(add_help=False)
parser_shared.add_argument('--version', action='store_true')

# the main parser, inherits from `parser_shared`
parser = argparse.ArgumentParser(description='main', parents=[parser_shared])

# several subcommands, which can't inherit from the main parser, since
# it would expect subcommands ad infinitum
subparsers = parser.add_subparsers('db', parents=[parser_shared])

...

args = parser.parse_args()

现在我希望能够调用这个程序,例如--version附加到普通程序或一些子命令:

$ prog --version
0.1

$ prog db --version
0.1

基本上,我需要声明可选的子解析器。我知道这并不真正受支持,但有任何解决方法或替代方案吗?

编辑:我收到的错误消息:

$ prog db --version
# works fine

$ prog --version
usage: ....
prog: error: too few arguments
4

7 回答 7

21

根据文档,--versionwith action='version'(而不是 with action='store_true')会自动打印版本号:

parser.add_argument('--version', action='version', version='%(prog)s 2.0')
于 2011-12-15T14:29:42.690 回答
8

FWIW,我也遇到了这个问题,最终通过不使用子解析器来“解决”它(我已经有了自己的打印帮助系统,所以在那里没有丢失任何东西)。

相反,我这样做:

parser.add_argument("command", nargs="?",
                    help="name of command to execute")

args, subcommand_args = parser.parse_known_args()

...然后子命令创建自己的解析器(类似于子解析器),它只在subcommand_args.

于 2014-02-26T22:04:10.860 回答
6

这似乎实现了可选子解析器的基本思想。我们解析适用于所有子命令的标准参数。然后,如果剩下任何东西,我们就调用其余部分的解析器。主要参数是子命令的父级,因此 -h 显示正确。如果不存在子命令,我计划输入交互式提示。

import argparse

p1 = argparse.ArgumentParser( add_help = False )    
p1.add_argument( ‘–flag1′ )

p2 = argparse.ArgumentParser( parents = [ p1 ] )
s = p2.add_subparsers()
p = s.add_parser( ‘group’ )
p.set_defaults( group=True )

( init_ns, remaining ) = p1.parse_known_args( )

if remaining:
    p2.parse_args( args = remaining, namespace=init_ns )
else:
    print( ‘Enter interactive loop’ )

print( init_ns )
于 2015-04-04T01:15:45.083 回答
4

正如http://bugs.python.org/issue9253(argparse:可选子解析器)中所讨论的,从 Python 3.3 开始,子解析器现在是可选的。这是parse_args检查所需参数的方式发生变化的意外结果。

我发现了一个软糖,可以恢复以前的(必需的子解析器)行为,明确设置动作的required属性subparsers

parser = ArgumentParser(prog='test')
subparsers = parser.add_subparsers()
subparsers.required = True   # the fudge
subparsers.dest = 'command'
subparser = subparsers.add_parser("foo", help="run foo")
parser.parse_args()

有关更多详细信息,请参阅该问题。我希望如果并且当这个问题得到正确修补时,默认情况下将需要子解析器,并带有某种选项将其required属性设置为False. 但是有大量积压的argparse补丁。

于 2014-02-27T06:04:50.673 回答
2

是的,我刚刚检查了,它在文档svn中用作对象示例,它只支持主命令上的“--version”:add_subparsers()

python zacharyyoung$ svn log --version
Subcommand 'log' doesn't accept option '--version'
Type 'svn help log' for usage.

仍然:

# create common parser
parent_parser = argparse.ArgumentParser('parent', add_help=False)
parent_parser.add_argument('--version', action='version', version='%(prog)s 2.0')

# create the top-level parser
parser = argparse.ArgumentParser(parents=[parent_parser])
subparsers = parser.add_subparsers()

# create the parser for the "foo" command
parser_foo = subparsers.add_parser('foo', parents=[parent_parser])

产生:

python zacharyyoung$ ./arg-test.py --version
arg-test.py 2.0
python zacharyyoung$ ./arg-test.py foo --version
arg-test.py foo 2.0
于 2011-12-15T17:15:07.370 回答
1

等待此功能交付时,我们可以使用如下代码:

# Make sure that main is the default sub-parser
if '-h' not in sys.argv and '--help' not in sys.argv:
    if len(sys.argv) < 2:
        sys.argv.append('main')
    if sys.argv[1] not in ('main', 'test'):
        sys.argv = [sys.argv[0], 'main'] + sys.argv[1:]
于 2012-11-13T22:48:58.737 回答
0

尽管@eumiro 的答案解决了该--version选项,但它只能这样做,因为这是 optparse 的一种特殊情况。允许一般调用:

 prog
 prog --verbose
 prog --verbose main
 prog --verbose db 

并且具有与(and )prog --version相同的工作,您可以在 Argumentparser 中添加一个方法,并在调用之前使用默认子解析器的名称调用它:prog --verbose mainprog main --verboseparse_args()

import argparse
import sys

def set_default_subparser(self, name, args=None):
    """default subparser selection. Call after setup, just before parse_args()
    name: is the name of the subparser to call by default
    args: if set is the argument list handed to parse_args()

    , tested with 2.7, 3.2, 3.3, 3.4
    it works with 2.6 assuming argparse is installed
    """
    subparser_found = False
    for arg in sys.argv[1:]:
        if arg in ['-h', '--help']:  # global help if no subparser
            break
    else:
        for x in self._subparsers._actions:
            if not isinstance(x, argparse._SubParsersAction):
                continue
            for sp_name in x._name_parser_map.keys():
                if sp_name in sys.argv[1:]:
                    subparser_found = True
        if not subparser_found:
            # insert default in first position, this implies no
            # global options without a sub_parsers specified
            if args is None:
                sys.argv.insert(1, name)
            else:
                args.insert(0, name)

argparse.ArgumentParser.set_default_subparser = set_default_subparser

def do_main(args):
    print 'main verbose', args.verbose

def do_db(args):
    print 'db verbose:', args.verbose

parser = argparse.ArgumentParser()
parser.add_argument('--verbose', action='store_true')
parser.add_argument('--version', action='version', version='%(prog)s 2.0')
subparsers = parser.add_subparsers()
sp = subparsers.add_parser('main')
sp.set_defaults(func=do_main)
sp.add_argument('--verbose', action='store_true')
sp = subparsers.add_parser('db')
sp.set_defaults(func=do_db)

parser.set_default_subparser('main')
args = parser.parse_args()

if hasattr(args, 'func'):
    args.func(args)

set_default_subparser()方法是ruamel.std.argparse包的一部分。

于 2014-10-15T09:43:31.117 回答