python接口自动化学习笔记(封装方法用于读取excel)

from collections import namedtuple from openpyxl import load_workbook # 1.如何将一些操作（Excel读写）进行封装？（写出封装的思路） # a、加载excel文件，定位表单； # b、定位单元格，读取所有用例数据； # c、遍历所有单元格数据，或使用ddt； # d、写入指定单元格数据，保存excel文件； class ReadExcelData(object): """封装读取excel表单数据的工具类""" def __init__(self, file_path, sheet_name=None): """ 初始化openpy加载excel文件 定位表单及初始化参数 """ self.file_path = file_path self.sheet_name = sheet_name # 打开已存在Excel文件 self.wb = load_workbook(self.file_path) #打开指定的excel # 定位表单 self.ws = self.wb[self.sheet_name] if self.sheet_name is not None else self.wb.active # 获取第一个表单 self.sheet_head_tuple = tuple(self.ws.iter_rows(max_row=self.ws.min_row, values_only=True))[0] self.cases_list = [] # 定义一个存放元组的对象 self.Cases = namedtuple("cases", self.sheet_head_tuple) # 创建一个命名元组类 def get_all_cases(self): """ 获取excel所有测试用例 """ for tuple_data in self.ws.iter_rows(min_row=self.ws.min_row + 1, values_only=True): # 每次遍历，返回由某行所有单元格值组成的一个元组 self.cases_list.append(self.Cases(*tuple_data)) return self.cases_list def get_one_case(self, row): """ 获取一条case :return一个case对象 """ if isinstance(row, int) and (self.ws.min_row + 1 <= row <= self.ws.max_row): return tuple(self.ws.iter_rows(min_row=row, max_row=row, values_only=True))[0] else: print("传入行号不正确，应为大于1的整数！") def write_file(self, row, actul_result, result_status): ''' 执行用例结果写入excel，并保存 ''' if isinstance(row, int) and (self.ws.min_row + 1 <= row < self.ws.max_row): self.ws.cell(row=row, column=self.ws.max_column - 1, value=actul_result) self.ws.cell(row=row, column=self.ws.max_column, value=result_status) self.wb.save(self.file_path) else: print("处理失败！") if __name__ == '__main__': file_path = u'E:\\heguanghua\\python\\接口自动化\\api.xlsx' sheet_name = 'Sheet1' res = ReadExcelData(file_path,sheet_name).get_all_cases() print(res) print('================================================') re = ReadExcelData(file_path,sheet_name).get_one_case(2) print(re)