首页 > 其他 > 详细

数据与代码分离:excel管理测试数据(二)

时间:2020-05-24 22:27:49      阅读:74      评论:0      收藏:0      [点我收藏+]

为了接口测试用例方便别人观看,加个标题

 

 

 技术分享图片

 

 1 from openpyxl import load_workbook
 2 
 3 
 4 class DoExcel:
 5     def __init__(self, filename, sheetname):
 6         self.filename = filename
 7         self.sheetname = sheetname
 8 
 9     def get_header(self):
10         """获取标题行"""
11         wb = load_workbook(self.filename)
12         sheet = wb[self.sheetname]
13         header = []  # 存储我们的标题行
14         for j in range(1, sheet.max_column+1):
15             header.append(sheet.cell(1, j).value)
16         return header
17 
18     def get_data(self):
19         wb = load_workbook(self.filename)
20         sheet = wb[self.sheetname]
21         header = self.get_header()
22         test_data = []
23 
24         for i in range(2, sheet.max_row+1):
25             sub_data = {}
26             for j in range(1, sheet.max_column+1):
27                 sub_data[header[j-1]] = sheet.cell(i, j).value
28             test_data.append(sub_data)
29         return test_data
30 
31 
32 if __name__ == __main__:
33     res = DoExcel("test.xlsx", "python").get_data()
34     print(res)

进阶,不要求掌握

 

注意:excel文件要和py文件放在同个文件夹,不然会报错

from openpyxl import load_workbook


class DoExcel:
def __init__(self, filename, sheetname):
self.filename = filename
self.sheetname = sheetname

def get_header(self):
"""获取标题行"""
wb = load_workbook(self.filename)
sheet = wb[self.sheetname]
header = [] # 存储我们的标题行
for j in range(1, sheet.max_column+1):
header.append(sheet.cell(1, j).value)
return header

def get_data(self):
wb = load_workbook(self.filename)
sheet = wb[self.sheetname]
header = self.get_header()
test_data = []

for i in range(2, sheet.max_row+1):
sub_data = {}
for j in range(1, sheet.max_column+1):
sub_data[header[j-1]] = sheet.cell(i, j).value
test_data.append(sub_data)
return test_data


if __name__ == ‘__main__‘:
res = DoExcel("test.xlsx", "python").get_data()
print(res)

数据与代码分离:excel管理测试数据(二)

原文:https://www.cnblogs.com/albeexu/p/12952589.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!