引用
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后面没有等号=,表示后面不带参数,其他两个后面有=,表示后面需要参数
注意:定义命令行参数时,要先定义带'-'选项的参数,再定义没有‘-’的参数