【转载】Python 获取命令行参数：getopt模块

Python 提供了 getopt 模块来获取命令行参数，getopt模块是专门处理命令行参数的模块，用于获取命令行选项和参数。下面通过实际例子来详细解释如何操作。

类似于下面包含参数的命令行

例1- $ python test.py -i input.txt -o output.txt middle.txt final.txt
例2- $ python test.py --ifile=input.txt --ofile=output.txt middle.txt 

getopt.getopt 方法函数原型:

getopt.getopt(args, shortopts, longopts=[])

方法参数说明：

• ** args：args为需要解析的参数列表。一般使用sys.argv[1:]**,这样可以过滤掉第一个参数（ps：第一个参数是脚本的名称，即例子中的test.py文件名，它不应该作为参数进行解析）.
• ** shortopts：**简写参数列表，短格式 (-) ，冒号(:)表示该选项必须有附加的参数，不带冒号表示该选项不附加参数.
• ** longopts：**长参数列表，长格式(–) ，等号(=)表示如果设置该选项，必须有附加的参数，否则就不附加参数。

方法返回值：

• opts：分析出的(option, value)列表对。
• args：不属于格式信息的剩余命令行参数列表。

实例

假定我们创建这样一个脚本，可以通过命令行向脚本文件传递两个文件名，同时我们通过另外一个选项查看脚本的使用。脚本使用方法如下：

usage: test.py -i <inputfile> -o <outputfile>

test.py 文件代码如下所示：
#!/usr/bin/python

# -*- coding: UTF-8 -*-
import sys, getopt
def main(argv): 
    inputfile = '' 
    outputfile = '' 
    try: 
        opts, args = getopt.getopt(argv,"hi:o:",["help","ifile=","ofile="])       
    except getopt.GetoptError: 
        sys.exit(2) 
    for opt, arg in opts: 
        if opt in ("-h","--help"): 
            print 'test.py -i <inputfile> -o <outputfile>' 
            sys.exit() 
        elif opt in ("-i", "--ifile"): 
            inputfile = arg 
        elif opt in ("-o", "--ofile"): 
            outputfile = arg 
    print '输入的文件为：', inputfile 
    print '输出的文件为：', outputfile

if __name__ == "__main__": 
    main(sys.argv[1:])

执行以上代码，输出结果为：

$ python test.py -h
usage: test.py -i <inputfile> -o <outputfile>

$ python test.py -i input.txt -o output.txt middle.txt final.txt
输入的文件为： input.txt
输出的文件为： output.txt

$ python test.py --ifile=input.txt --ofile=output.txt middle.txt final.txt
输入的文件为： input.txt
输出的文件为： output.txt

对代码的理解：

• main(sys.argv[1:]) 调用main函数，getopt.getopt获取命令行参数-i input.txt -o
  output.txt middle.txt final.txt，
• 返回值opts是个包含元祖的列表，每个元祖是分析出来的格式信息，比如
  [(’-i’,‘input.txt’),(’-o’,‘output.txt’)] ;
• 返回值args是个列表，包含那些没有‘-’或‘–’的参数，比如：[‘middle.txt’,‘final.txt’]
• “hi⭕️” 短格式 h 后面没有冒号：表示后面不带参数，i和 o后面有冒号表示后面需要参数
• [“help”,“ifile=”,“ofile=”] 长格式
  help后面没有等号=，表示后面不带参数，其他两个后面有=，表示后面需要参数

注意：定义命令行参数时，要先定义带’-'选项的参数，再定义没有‘-’的参数