解析 argv 时忽略无法识别的选项？

Question

我正在编写一个脚本，充当插件类的 run 方法的代理。

该脚本将像这样调用：

> main.py -v --plugin=Foo --extra=bar -c

该命令的结果相当于：

plugin = my.module.Foo()
plugin.run(extra='bar', c=True)

请注意，--plugin 前面的任何内容都由 main.py 内部使用，不会传递给插件。 --plugin 之后的任何内容都会被 main.py 忽略，而是直接传递给插件。

我遇到的问题是我找不到类似 getopt 的类，它允许我解析 argv 而无需指定允许的选项列表。

我不想重写 getopt 并注释掉一行。还有更好的选择吗？

Answer 1

您想要的内容位于 argparse 库中，请参阅 http://docs.python.org/dev/library/argparse.html#partial-parsing：

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', action='store_true')
>>> parser.add_argument('bar')
>>> parser.parse_known_args(['--foo', '--badger', 'BAR', 'spam'])
(Namespace(bar='BAR', foo=True), ['--badger', 'spam'])

Answer 2

您可以做一些愚蠢的事情，只生成一个字典，您可以从中查找键等以确定如何处理每个项目......

#!/usr/bin/python
import sys 
options = dict()
for pair in map(lambda item: item.lstrip('-'), sys.argv[1:]):
    pair = pair.split('=')
    if len(pair) < 2:
        option, val = pair[0], True
    else:
        option, val = pair
    options[option] = val 
print options

Answer 3

你很幸运。我编写了一个修改后的 getopt正是如此。然而，一个限制是空头期权必须先于多头期权。这可能是可以解决的。

Answer 4

我刚刚发现 getopt 如果遇到 --：

Python 2.6.6 (r266:84292, Jun 16 2011, 16:59:16) 
Type "help", "copyright", "credits" or "license" for more information.
>>> from getopt import getopt
>>>
>>> argv = ['-v', '--plugin=foo', '--', '--extra=bar', '-c']
>>> opts, extra = getopt(argv, 'v', 'plugin=')
>>>
>>> opts
[('-v', ''), ('--plugin', 'foo')]
>>>
>>> extra
['--extra=bar', '-c']

请注意，上面的 argv 相当于调用：

> main.py -v --plugin=Foo -- --extra=bar -c

我特别喜欢这个解决方案，因为它为用户在订购方式上提供了一点额外的灵活性参数。

Answer 5

使用正则表达式搜索 'plugin=，如果找到，则分割该行并使用 getopt 解析每一半。