Python处理命令行参数模块optpars用法实例分析

简介

在Python程序开发中，经常需要处理用户从命令行传入的参数。Python官方库提供了Optparse模块，可方便地解析和处理命令行参数。本文将通过实例来讲解Optparse的使用方法。

optparse的基本用法

Optparse提供了OptionParser类，可用于定义和解析命令行选项。

定义选项

OptionParser类的add_option()方法可用于定义选项。方法的参数如下：

• opt_str：短选项或长选项，形如"-a"或"--alpha"。
• dest：选项的属性名
• action：选项的行为，如store表示将选项值存储到dest属性中。
• help：选项的帮助信息

示例代码：

from optparse import OptionParser

usage = "usage: %prog [options] arg1 arg2"
parser = OptionParser(usage=usage)
parser.add_option("-f", "--file", dest="filename", help="input file name")
parser.add_option("-q", "--quiet", action="store_false", dest="verbose", default=True, help="do not print status messages to stdout")

(options, args) = parser.parse_args()

在上述代码中，我们定义了两个选项：-f和--file用于指定文件名，-q和--quiet用于设置程序不输出运行状态信息，具体的选项信息如下：

-f FILENAME, --file=FILENAME   input file name
-q, --quiet                    do not print status messages to stdout

解析选项

OptionParser类的parse_args()方法可用于解析命令行选项，方法会自动检查sys.argv中的命令行参数，并赋值给所定义的选项。

示例代码：

(options, args) = parser.parse_args()
if len(args) != 2:
    parser.error("incorrect number of arguments")

在上述代码中，我们通过parse_args()方法解析命令行参数，并使用parser.error()方法提示参数个数是否正确。

示例1：统计文件行数

我们来看一个简单的示例，该示例用于统计输入的文件中的行数，代码如下：

from optparse import OptionParser

usage = "usage: %prog [options] filename"
parser = OptionParser(usage=usage)
parser.add_option("-s", "--skip_blank", action="store_true", dest="skip_blank", default=False, help="skip blank lines")

(options, args) = parser.parse_args()

if len(args) != 1:
    parser.error("incorrect number of arguments")

filename = args[0]

lines = 0
with open(filename) as f:
    for line in f:
        if options.skip_blank and not line.strip():
            continue
        lines += 1

print("There are %d lines in file %s" % (lines, filename))

在上述代码中，我们定义了一个skip_blank选项用于跳过空白行。执行结果如下：

$ python line_counter.py --skip_blank test.txt
There are 7 lines in file test.txt

示例2：合并文件

另一个例子是将两个文件合并成一个新文件，代码如下：

from optparse import OptionParser

usage = "usage: %prog [options] infile1 infile2 outfile"
parser = OptionParser(usage=usage)

(options, args) = parser.parse_args()

if len(args) != 3:
    parser.error("incorrect number of arguments")

infile1, infile2, outfile = args

with open(infile1) as f1, open(infile2) as f2, open(outfile, "w") as f3:
    f3.write(f1.read())
    f3.write(f2.read())

print("Files %s and %s have been merged into %s" % (infile1, infile2, outfile))

在上述代码中，我们通过定义了三个参数：infile1、infile2和outfile，并在程序中使用这些参数，将两个输入文件合并到输出文件中。

执行结果如下：

$ python merge_files.py file1.txt file2.txt merged.txt
Files file1.txt and file2.txt have been merged into merged.txt

总结

本文介绍了Optparse模块的基本用法，并提供了两个实例说明。实际上，Optparse提供了更多高级用法，更多细节可以参考官方文档。