在接口测试中,有的时候我们需要把测试数据存放在excel里,便于测试数据管理,但是数据的读取对于刚写代码的同学可能有点困难,今天我就整理一下。以下代码就是解决了如何用python读取excel里面的数据。以字典的形式保存在列表里面。
# 先导入openpyxl,没安装的先安装一下,安装命令:pip install openpyxl
import openpyxll
# 封装一个读取数据的类
class ReadExcel(object):
#为了便于操作,我们写一个__init__方法,用于保存文件名和表单
def __init__(self, file_name, sheet_name):
self.file_name = file_name
self.sheet_name = sheet_name
# 写一个读取数据的方法
def read_excel(self):
wb = openpyxl.load_workbook(self.file_name)
sh = wb[self.sheet_name]
#定义一个对象,用来保存excel里面数据的格子
data = list(sh.rows)
#把表单第一行作为字典的KEY
title = [r1.value for r1 in data[0]]
#定义一个空列表用来保存测试数据,后面我们会把测试数据以字典的形式保存在这个列表中
li = []
#去掉第一行,开始遍历
for value in data[1:]:
r = [i.value for i in value]
#聚合打包,将title,r以字典的形式保存
x = dict(zip(title, r))
# 把每次得到的r,添加到列表中
li.append(x)
#返回li
return (li)
#定义一个类对象
str = ReadExcel("case.XLSX", "Sheet1")
#打印读取到的数据
print(str.read_excel())
# 下图就是我们打印出来的结果,我们看一下。