python - 在解析器/子解析器的开头使用 argparse.REMAINDER

Question

我想实现一个 arg 解析器，它允许我将单元测试作为子命令之一运行，盲目地将参数传递给 unittest.main()。例如，

$ foo.py unittest [args to pass to unittest.main()]

连同其他子命令:

$ foo.py foo ...
$ foo.py bar ...

按照 argparse 的例子，这是可行的:

#!/usr/bin/python                                                               
import argparse                                                                 

p = argparse.ArgumentParser(prog='PROG')                                        
p.add_argument('-v', '--verbose', action='store_true')                          
sub = p.add_subparsers(dest='cmd')                                              
foo = sub.add_parser('foo')                                                     
bar = sub.add_parser('bar')                                                     
unittest = sub.add_parser('unittest')                                           
unittest.add_argument('command') # Need to add this to make it work.                                              
unittest.add_argument('args', nargs=argparse.REMAINDER)                         

print(p.parse_args('unittest command -blah blah'.split()))       

输出:

Namespace(args=['-blah', 'blah'], cmd='unittest', command='command', verbose=False)

但这不是。它似乎首先需要一个“正常”参数:

#!/usr/bin/python                                                               
import argparse                                                                 

p = argparse.ArgumentParser(prog='PROG')                                        
p.add_argument('-v', '--verbose', action='store_true')                          
sub = p.add_subparsers(dest='cmd')                                              
foo = sub.add_parser('foo')                                                     
bar = sub.add_parser('bar')                                                     
unittest = sub.add_parser('unittest')                                           
unittest.add_argument('args', nargs=argparse.REMAINDER)                         

print(p.parse_args('unittest -blah blah'.split()))             

输出:

$ /tmp/foo.py    
usage: PROG [-h] [-v] {foo,bar,unittest} ...
PROG: error: unrecognized arguments: -blah

我可以做print(p.parse_args('unittest -- -f -g'.split()))，但需要-- 有点违背了 argparse.REMAINDER 的目的。

有没有办法让 argparse 做我想做的事？还是我只需要手动解析这种情况？

python 2.7.5

Answer 1

看起来与 http://bugs.python.org/issue17050 中讨论的问题相同, argparse.REMAINDER 不能作为第一个参数

我 4 年前的推论仍然成立 - -blah 甚至在 REMAINDER 有机会采取行动之前就被归类为可选标志。 '--' 被较早地解析，但是 ... 在某种意义上只是 '*' 的概括。而且不是广泛使用的。 'subparsers' Action 的值(value)在于它有一个 nargs='+...' 值 (argparse.PARSER) - 它类似于 REMAINDER，只是它至少需要一个字符串, 'cmd'。

http://bugs.python.org/issue9334 中的可能修复尚未采取行动。所以你要么需要自己处理'-blah'，要么使用'--'。 parse_known_args 也可能适用于您的情况。