Python 中带有可选参数的命令行选项

Question

我想知道是否有一种简单的方法来解析 Python 中具有可选参数的命令行选项。例如，我希望能够以两种方式调用脚本：

> script.py --foo
> script.py --foo=bar

从Python getopt 文档看来我必须选择其中一种。

Answer 1

stdlib 中的 optparse 模块不支持开箱即用（并且不应该这样做，因为以这种方式使用命令行选项是一种不好的做法）。

正如 @Kevin Horn 指出的，你可以使用 argparse 模块（可通过 easy_install argparse 安装或直接获取 argparse.py 并将其放在 sys.path 中的任何位置）。

示例

#!/usr/bin/env python
from argparse import ArgumentParser

if __name__ == "__main__":
    parser = ArgumentParser(prog='script.py')
    parser.add_argument('--foo', nargs='?', metavar='bar', default='baz')

    parser.print_usage()    
    for args in ([], ['--foo'], ['--foo', 'bar']):
        print "$ %s %s -> foo=%s" % (
            parser.prog, ' '.join(args).ljust(9), parser.parse_args(args).foo)

输出

usage: script.py [-h] [--foo [bar]]
$ script.py           -> foo=baz
$ script.py --foo     -> foo=None
$ script.py --foo bar -> foo=bar

Answer 2

另请注意，标准库还有 optparse，这是一个更强大的选项解析器。

Answer 3

查看 argparse：
http://code.google.com/p/argparse/

尤其是“nargs”选项

Answer 4

optparse 中没有选项允许您执行此操作。但你可以扩展它来做到这一点：

http://docs.python .org/library/optparse.html#adding-new-actions

Answer 5

使用 optparse 包。