51Testing软件测试论坛

标题: Python命令行参数 [打印本页]

作者: 测试积点老人 时间: 2018-12-19 11:20
标题: Python命令行参数
本帖最后由测试积点老人于 2018-12-19 11:25 编辑

Python 获得命令行参数的方法

需要模块：sys
参数个数：len(sys.argv)
脚本名： sys.argv[0]
参数1： sys.argv[1]
参数2： sys.argv[2]

解析命令行参数

Python提供了一个getopt模块，可用于解析命令行选项和参数。

该模块提供了两个功能和异常，以启用命令行参数解析。

getopt.getopt方法此方法解析命令行选项和参数列表。以下是此方法的简单语法

-getopt.getopt(args, options, [long_options])

复制代码

Pythongetopt.GetoptError异常

当在参数列表中有一个无法识别的选项，或者当需要一个参数的选项不为任何参数时，会引发这个异常。
异常的参数是一个字符串，指示错误的原因。属性msg和opt给出错误消息和相关选项。

import sys, getopt
def main(argv):
inputfile = ''
outputfile = ''
try:
opts, args = getopt.getopt(argv, "hi:o:g:", ['help', 'ifile=', "ofile="])
# 长选项'='相当于短选项的'：'
print(opts) # 打印参数与值 [('-i', 'in'), ('-o', 'out')]
except getopt.GetoptError as e:
print(e)
sys.exit(-2)
for opt, arg in opts:
if opt in ('-h', '--help'):
print('usage: command_line_usage.py -i <inputfile> -o <outputfile>')
sys.exit()
elif opt in ("-i", "--ifile"):
inputfile = arg
elif opt in ("-o", "--ofile"):
outputfile = arg
print('Input file is "', inputfile)
print('Output file is "', outputfile)
if __name__ == "__main__":
main(sys.argv[1:]) # 要处理的参数列表，sys.argv[0]为脚本名，所以用sys.argv[1:]过滤掉脚本名。

复制代码

代码解释：
a) sys.argv[1:]为要处理的参数列表，sys.argv[0]为脚本名，所以用sys.argv[1:]过滤掉脚本名。
b) "hi: o :": 当一个选项只是表示开关状态时，即后面不带附加参数时，在分析串中写入选项字符。当选项后面是带一个附加参数时，在分析串中写入选项字符同时后面加一个":"号。所以"hi : o:"就表示"h"是一个开关选项；"i :"和" o : "则表示后面应该带一个参数。
c) 调用getopt函数。函数返回两个列表：opts和args。opts为分析出的格式信息。args为不属于格式信息的剩余的命令行参数。opts是一个两元组的列表。每个元素为：(选项串,附加参数)。如果没有附加参数则为空串''。

getopt函数的第三个参数[, long_options]为可选的长选项参数，上面例子中的都为短选项(如-i -o)

长选项格式举例:

--version
--file=error.txt

让一个脚本同时支持短选项和长选项
getopt.getopt(sys.argv[1:], "hi: o:", ["version", "file="])

D:\TestFPython>python 1_1_getopt.py -i in -o out
[('-i', 'in'), ('-o', 'out')]
Input file is " in
Output file is " out

复制代码

作者: Miss_love 时间: 2021-1-5 17:01
支持分享

欢迎光临 51Testing软件测试论坛 (http://bbs.51testing.com/)