不要解析最后一个位置参数之后的选项

Question

我正在围绕 ssh 命令行客户端编写一个包装器。在作为命令一部分的第一个位置参数之后，所有其他选项也应被视为位置参数。

在 optparse 下，我相信这可以通过 完成disable_interspersed_args。

目前我有这样的事情：

parser = argparse.ArgumentParser()
parser.add_argument('--parallel', default=False, action='store_true')
# maybe allow no command? this would ssh interactively into each machine...
parser.add_argument('command', nargs='+')
args = parser.parse_args()

但是，如果选项作为命令的一部分传递（例如 my_wrapper ls -l），它们会被 ArgumentParser 解释为未知选项。 错误：无法识别的参数：-l

如果我使用 parse_known_args()，选项可能会乱序。

p = argparse.ArgumentParser()
p.add_argument('-a', action='store_true')
p.add_argument('command', nargs='+')
print(p.parse_known_args())

$ python3 bah.py -b ls -l -a
(Namespace(a=True, command=['ls']), ['-b', '-l'])

这里可以看到-b在ls之前的位置已经丢失，并且-a已经从命令中解析出来了，即不想要的。

我怎样才能：

• 防止参数在某一点之后被解析？
• 禁用散布参数的解析？
• 允许带有前缀的参数作为位置参数使用吗？

Answer 1

我也有同样的问题。我在 argparse bug 跟踪器上找到了解决方案：http://code.google .com/p/argparse/issues/detail?id=52

解决方案很简单：将 nargs='+' （或 '*'）替换为nargs=argparse.REMAINDER。这个特殊值没有记录，但它可以满足您的需求。

Answer 2

我认为开始解决这些问题的最佳选择是在所有可选参数之后尝试 -- 。 -- 是一个伪参数，它告诉 ArgumentParser 之后的所有内容都是位置参数。文档位于此处

至于防止参数在特定时间后被解析一点，您可以将 argv 的一部分传递给 parse_args。与一些内省相结合可以用来限制解析的内容。

Answer 3

@dcolish 建议的是通用方法。这是一个示例实现，它也支持标准 -- 分隔符，但正确解析不需要使用它。

结果：

# ./parse-pos.py -h
usage: parse-pos.py [-h] [-qa] [-qb] COMMAND [ARGS...]

# ./parse-pos.py -qa ls -q -h aa /bb
try_argv = ['-qa', 'ls']
cmd_rest_argv = ['-q', '-h', 'aa', '/bb']
parsed_args = Namespace(command='ls', qa=True, qb=False)

代码：

#!/usr/bin/python3

import argparse
import sys
from pprint import pprint

class CustomParserError(Exception):
    pass

class CustomArgumentParser(argparse.ArgumentParser):
    def error(self, message):
        raise CustomParserError(message)
    def original_error(self, message):
        super().error(message)

def parse_argv():
    parser = CustomArgumentParser(description='Example')
    parser.add_argument('command', metavar='COMMAND [ARGS...]', help='the command to be executed')
    parser.add_argument('-qa', action='store_true') # "ambiguous option" if you specify just "-q"
    parser.add_argument('-qb', action='store_true') # "ambiguous option" if you specify just "-q"

    def parse_until_positional(parser, _sys_argv = None):
        if _sys_argv is None:
            _sys_argv = sys.argv[1:] # skip the program name
        for i in range(0, len(_sys_argv) + 1):
            try_argv = _sys_argv[0:i]
            try:
                parsed_args = parser.parse_args(try_argv)
            except CustomParserError as ex:
                if len(try_argv) == len(_sys_argv):
                    # this is our last try and we still couldn't parse anything
                    parser.original_error(str(ex)) # sys.exit()
                continue
            # if we are here, we parsed our known optional & dash-prefixed parameters and the COMMAND
            cmd_rest_argv = _sys_argv[i:]
            break
        return (parsed_args, cmd_rest_argv, try_argv)

    (parsed_args, cmd_rest_argv, try_argv) = parse_until_positional(parser)

    # debug
    pprint(try_argv)
    pprint(cmd_rest_argv)
    pprint(parsed_args)

    return (parsed_args, cmd_rest_argv)

def main():
    parse_argv()

main()

Answer 4

另一种选择是使用 parse_known_args，它会在以下情况下停止解析：遇到未知参数。