如何在 Python 中处理命令行参数？ [复制]

如果我期望像 001 或 999 这样的东西（这次让我们将期望限制在 001...999 范围），并且传递的其他参数很少，并且希望忽略任何意外的参数，那么处理命令行参数的简单表达式是什么？

我知道，例如，如果我需要查明“调试”是否在参数之间传递，那么它会是这样的：

if 'debug' in argv[1:]:
  print 'Will be running in debug mode.'

如何知道009或575是否通过？

所有这些都是预期的电话：

python script.py
python script.py 011
python script.py 256 debug
python script.py 391 xls
python script.py 999 debug pdf

此时我不关心这样的调用：

python script.py 001 002 245 568
python script.py some unexpected argument
python script.py 0001
python script.py 02

...第一个 - 因为多个“数字”参数；第二-因为……嗯，意想不到的争论；第三和第四 - 因为非 3 位数的参数。

正如其他人回答的那样， optparse 是最好的选择，但如果您只想快速编写代码，请尝试以下操作：

import sys, re

first_re = re.compile(r'^\d{3}$')

if len(sys.argv) > 1:

    if first_re.match(sys.argv[1]):
        print "Primary argument is : ", sys.argv[1]
    else:
        raise ValueError("First argument should be ...")

    args = sys.argv[2:]

else:

    args = ()

# ... anywhere in code ...

if 'debug' in args:
    print 'debug flag'

if 'xls' in args:
    print 'xls flag'

EDIT：这是一个 optparse 示例，因为很多人都在回答 optparse ，但没​​有真正解释原因，或解释必须更改哪些内容才能使其正常工作。

使用 optparse 的主要原因是它为您以后的扩展提供了更大的灵活性，并为您在命令行上提供了更大的灵活性。换句话说，您的选项可以以任何顺序出现，并且使用消息会自动生成。但是，要使其与 optparse 一起使用，您需要更改规范以将“-”或“--”放在可选参数前面，并且需要允许所有参数按任何顺序排列。

这是一个使用 optparse 的示例：

import sys, re, optparse

first_re = re.compile(r'^\d{3}$')

parser = optparse.OptionParser()
parser.set_defaults(debug=False,xls=False)
parser.add_option('--debug', action='store_true', dest='debug')
parser.add_option('--xls', action='store_true', dest='xls')
(options, args) = parser.parse_args()

if len(args) == 1:
    if first_re.match(args[0]):
        print "Primary argument is : ", args[0]
    else:
        raise ValueError("First argument should be ...")
elif len(args) > 1:
    raise ValueError("Too many command line arguments")

if options.debug:
    print 'debug flag'

if options.xls:
    print 'xls flag'

optparse 和您的规范的区别在于，现在您可以使用如下命令行：

python script.py --debug --xls 001

您可以通过调用 parser.add_option() 轻松添加新选项