背景
很多公司还是用excel去管理测试用例的,所以为了减少重复繁琐的导出导出工作,学会如何用代码操作excel表格很实用~
1、读取excel文件基本步骤
1、操作excel的一些库
1、xlrd:读取库,xlwt:写入,现在基本不用,因为只能处理.xls这种格式的数据
2、使用openpyxl库:不支持的 .xls格式,支持的格式:.xlsx、.xlsm、.xltx、.xltm,可以将.xls 转化为 xlsx格式
- 安装openpyxl,使用 pip install openpyxl
- 导入:import | from openpyxl import xxx
2、读取的步骤
1、读取单元格数据
获取表格的工作簿对象(用到 openpyxl中的load_workbook模块)得到表单,通过sheet名称 — sheet获取单元格 — cell获取单元格数据 — cell.value实战演练:
excel表格如下:
代码实战:
from openpyxl import load_workbookimport pathlib# 获取excel的相对路径cur_path = pathlib.Path(__file__).absolute()excel_path = cur_path.parent/"testcase64.xlsx"# 第一步,获取表格的工作簿对象wb = load_workbook(excel_path)print(wb) # <openpyxl.workbook.workbook.Workbook object at 0x03EFCD60># 第二步,得到表单——sheetsheet = wb["login"]print(sheet)# 第三步,获取单元格数据cell = sheet.cell(row=1,column=3)print(cell)# 第四步,获取单元格数据 -- valuecell_value = sheet.cell(row=2,column=2).valueprint(cell_value) # 输出:登录正常# 获取表单中所有的数据 -- 转化为列表# 因为 sheet.values的输出是“<generator object Worksheet.values at 0x03817840” > 就是一个生成器,可以转化为列表的形式datas = list(sheet.values)print(datas) # 是一个列表,每一个数据都是元组'''[('id', 'title', 'method', 'header', 'url', 'data', 'expected'), (1, '登录正常', 'post', '{"Content-Type":"application/json"}', 'https://openapiv5.ketangpai.com//UserApi/login', '{"email":"2378807189@qq.com ","password":"lemon123456 ","remember":"0","code":"","mobile":"","type":"login","reqtimestamp":1695045752424}', '{"status":1,"code":10000,"message":"访问成功","data":{"token":"skjdaskjhd"}'), (2, '登录异常', 'post', '{"Content-Type":"application/json"}', 'https://openapiv6.ketangpai.com//UserApi/login', '{"email":"2378807189@qq.com ","password":"lemon","remember":"0","code":"","mobile":"","type":"login","reqtimestamp":1695045752424}', '{"status":0,"code":"30508","message":"登录失败"}')]''''''全部读取的数据转化为项目里需要的格式数据'''# 第一步,取出标题title = datas[0]# 第二步,再一次获取后续的每一行数据,跟title压缩成字典case_list = []for case in datas[1:]: case_dict = dict(zip(title,case)) case_list.append(case_dict)print(case_list)# 通过列表推导式来遍历case_list1 = []case_list1 = [case_list1.append(dict(zip(title,case))) for case in datas]print(case_list)
2、读取excel文件的优化
优化思路,代码分层思想:按照不同的功能代码进行不同包的管理(自动化框架的结构):
工具层:tools/common/util,封装好的公共方法,类同于Jmeter里的函数助手,比如对excel读取数据的处理过程测试数据层:data/testdata,放测试数据,例如:txt文件、excel表格等测试用例层:主要维护用例(pytest框架主要做的事情)测试结果输出:如 测试报告、测试日志入口文件:main.py、run.py一般放在最外层优化实战:
1、按照分层思想,新建data层和tools层,并完善其中的文件,参考如下图片:
2、将获取excel_path和获取数据的方式分别写到handle_excel文件和handle_path两个文件中,如下:
''' handle_excel文件'''from openpyxl import load_workbookfrom d11_excel操作和pytest框架.tools.handle_path import exce_pathdef read_excel(filename, sheet) wb = load_workbook(filename) sheet = wb[sheet] datas_list = list(sheet.values) title = datas_list[0] list_case = [] for case in datas_list[1:]: tep_dict = dict(zip(title,case)) list_case.append(tep_dict) return list_case
'''handle_path.py'''import pathlibcur_path = pathlib.Path(__file__).absolute()exce_path = cur_path.parent.parent/"data"/"testcase64.xlsx"if __name__ == '__main__': print(exce_path)
3、一些扩展(如何快速查看excel表格中的数据类型)
可以通过debug的方式,看到我们从excel中读取的数据的类型,举例如下: