背景
很多公司还是用excel去管理测试用例的,所以为了减少重复繁琐的导出导出工作,学会如何用代码操作excel表格很实用~
1、xlrd:读取库,xlwt:写入,现在基本不用,因为只能处理.xls这种格式的数据
2、使用openpyxl库:不支持的 .xls格式,支持的格式:.xlsx、.xlsm、.xltx、.xltm,可以将.xls 转化为 xlsx格式
- 安装openpyxl,使用 pip install openpyxl
- 导入:import | from openpyxl import xxx
1、读取单元格数据
实战演练:
excel表格如下:
代码实战:
from openpyxl import load_workbook
import pathlib
# 获取excel的相对路径
cur_path = pathlib.Path(__file__).absolute()
excel_path = cur_path.parent/"testcase64.xlsx"
# 第一步,获取表格的工作簿对象
wb = load_workbook(excel_path)
print(wb) #
# 第二步,得到表单——sheet
sheet = wb["login"]
print(sheet)
# 第三步,获取单元格数据
cell = sheet.cell(row=1,column=3)
print(cell)
# 第四步,获取单元格数据 -- value
cell_value = sheet.cell(row=2,column=2).value
print(cell_value) # 输出:登录正常
# 获取表单中所有的数据 -- 转化为列表
# 因为 sheet.values的输出是“ 就是一个生成器,可以转化为列表的形式
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":"[email protected] ","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":"[email protected] ","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)
优化思路,代码分层思想:按照不同的功能代码进行不同包的管理(自动化框架的结构):
优化实战:
1、按照分层思想,新建data层和tools层,并完善其中的文件,参考如下图片:
2、将获取excel_path和获取数据的方式分别写到handle_excel文件和handle_path两个文件中,如下:
'''
handle_excel文件
'''
from openpyxl import load_workbook
from d11_excel操作和pytest框架.tools.handle_path import exce_path
def 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 pathlib
cur_path = pathlib.Path(__file__).absolute()
exce_path = cur_path.parent.parent/"data"/"testcase64.xlsx"
if __name__ == '__main__':
print(exce_path)
可以通过debug的方式,看到我们从excel中读取的数据的类型,举例如下: