class ReadExcel(object):
def __init__(self,file_name, sheet_name):
"""
:param file_name: 文件名字
:param sheet_name: 表单名字
"""
self.file_name = file_name
self.sheet_name = sheet_name
def open_file(self):
self.workbook = openpyxl.load_workbook(self.file_name) # 打开工作簿
self.sheet = self.workbook[self.sheet_name] # 获取表单名
def readData(self):
# 调用打开文件的方法
self.open_file()
cases = [] # 定义空列表，存放测试用例的数据
# rows为按行获取到的数据，以列表形式展示
rows = list(self.sheet.rows) # 转换成列表形式
# 获取表头：键
titles = [] # 测试数据标题
for row in rows[0]:
titles.append(row.value)
# 获取数据：值
for row in rows[1:]:
datas=[]
for i in row:
datas.append(i.value)
cases.append(dict(zip(titles,datas))) # 打包成字典形式
return cases
# 写入数据的方法
def writeData(self, row, column, value):
"""
:param row: 写入的行
:param column: 写入的类
:param value: 写入的内容
:return:
"""
# 打开文件
self.open_file()
# 安装传入的行和列写入内容
self.sheet.cell(row=row, column=column, value=value)
# 保存工作簿
self.workbook.save(self.file_name)
