×

Python 获取命令行参数:getopt模块

96
孙宾宾
2016.12.02 21:54* 字数 651

引用

Python 命令行参数,修改并增加了部分内容,原文链接地址ttp://www.runoob.com/python/python-command-line-arguments.html,如涉及版权问题请及时联系我!

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:o:"
    短格式 h 后面没有冒号:表示后面不带参数,i和 o后面有冒号表示后面需要参数

  • ["help","ifile=","ofile="]
    长格式 help后面没有等号=,表示后面不带参数,其他两个后面有=,表示后面需要参数

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

吊儿郎当学python
Web note ad 1