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

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)
原文:https://www.cnblogs.com/albeexu/p/12952589.html