自动化不知如何参数化(一)?xlrd来帮你解决

时间:2024-01-27 10:25:12

平时在做自动化测试的时候,一直都是要求数据与业务逻辑分离。把测试数据都写在业务里面的话,比较混杂。为了方便管理测试数据,所以引入了python的一个扩展库--xlrd。该库使用简单,能满足自动化测试的数据分离的需求。

闲语少聊,我们直接来看,xlrd是如何完成数据读取的。

安装

安装很简单,方式①:直接命令pip install xlrd;方式②:在Pycharm中直接安装即可。

基础命令

①打开excel文件并创建对象

excel_object = xlrd.open_workbook(excel_path)

②根据工作表名称获取数据

sheet_object = excel_object.sheet_by_name(\'Sheet1\')

③获取excel文件中所有工作表名称

print(excel_object.sheet_names())

④获取有效行数

row_nrows = sheet_object.nrows

⑤获取有效列数

col_ncols = sheet_object.ncols

⑥获取当前行的单元格长度

row_length = sheet_object.row_len(1)

⑦获取第一行数据

rows_cell_value = sheet_object.row_values(0)

⑧获取坐标为第一行、第一列单元格数据

row_cell_value1 = sheet_object.cell_value(0, 0)# 坐标起始位从0开始
或者
row_cell_value1 = sheet_object.cell(0, 0).value
或者
row_cell_value1 = sheet_object.row(0)[0].value

⑨获取坐标为第二行、第四列单元格数据

row_cell_value2 = sheet_object.cell_value(1, 3)

⑩获取第四列数据

cols_cell_value = sheet_object.col_values(3)

单元格类型

单元格类型区分为六类,读取单元格类型如下所示:

0 empty,1 string, 2 number, 3 date, 4 boolean, 5 error
cell_value_type1 = sheet_object.cell_type(2, 0)
或者
cell_value_type1 = sheet_object.cell(2, 0).value

命令实践

我们来写个例子验证下这些命令,示例代码如下:

\'\'\'
@author: wenyihuqingjiu
@project: Pytest_Auto_Test
@file: demo
@time: 2020-07-23 21:40
@desc:
\'\'\'

import xlrd
import os

current_path = os.path.dirname(__file__)
excel_path = os.path.join(current_path, \'..\', \'file/excel/testcase.xlsx\')
excel_object = xlrd.open_workbook(excel_path)
sheet_object = excel_object.sheet_by_name(\'Sheet1\')
# 获取第一行数据
rows_cell_value = sheet_object.row_values(0)
print(\'获取到的第一行数据为:\' + str(rows_cell_value))
# 获取第一行、第一列单元格数据
row_cell_value1 = sheet_object.cell_value(0, 0)
print(\'获取坐标为第一行、第一列单元格的数据为:\' + row_cell_value1)
# 获取第二行、第四列单元格数据
row_cell_value2 = sheet_object.cell_value(1, 3)
print(\'获取坐标为第二行、第四列单元格的数据为:\' + row_cell_value2)
# 获取第四列数据
cols_cell_value = sheet_object.col_values(3)
print(\'获取第四列的数据为:\' + str(cols_cell_value))
# 获取单元格类型
# 0 empty,1 string, 2 number, 3 date, 4 boolean, 5 error
cell_value_type1 = sheet_object.cell_type(2, 0)
print(\'单元格类型为:\' + str(cell_value_type1))
cell_value_type2 = sheet_object.cell_type(1, 0)
print(\'单元格类型为:\' + str(cell_value_type2))
cell_value_type3 = sheet_object.cell_type(1, 7)
print(\'单元格类型为:\' + str(cell_value_type3))
cell_value_type4 = sheet_object.cell_type(1, 8)
print(\'单元格类型为:\' + str(cell_value_type4))
cell_value_type5 = sheet_object.cell_type(2, 8)
print(\'单元格类型为:\' + str(cell_value_type5))
# 获取有效行数
row_nrows = sheet_object.nrows
print(\'当前sheet页中有效行数为:\' + str(row_nrows))
# 获取有效列数
col_ncols = sheet_object.ncols
print(\'当前sheet页中有效列数为:\' + str(col_ncols))
# 获取当前行的单元格长度
row_length = sheet_object.row_len(0)
print(\'第一行的有效单元格长度为:\' + str(row_length))
# 获取所有工作表名称
print(\'当前excel中的工作表名称为:\'+ str(excel_object.sheet_names()))

excel文档内容如下所示:

 执行代码,我们结合excel文档内容来对比看下输出结果,如下所示:

我们从上述简单的命令了解到,读取数据的方式其实很简单,代码中加上逻辑便可获取整个sheet页的数据,如下所示,获取当前sheet页的数据代码:

# 获取有效行数
row_index = sheet_object.nrows
# 获取有效列数
col_index = sheet_object.ncols
# 定义空数组
all_data_list = []
# 获取首行数据
first_row = sheet_object.row_values(0)
for row in range(1, row_index):
    # 定义空列表
    row_dict = {}
    for col in range(col_index):
        c_cell = sheet_object.cell_value(row, col)
        # 循环每一个有效的单元格,将字段与值对应存储到字典中
        row_dict[first_row[col]] = c_cell
    # 再将字典追加到列表中
    all_data_list.append(row_dict)
    print(str(row_dict))

查看控制台,打印出来的数据如下所示:

excel文件内容还是如上截图,可以对比查看下。

{\'模块\': \'搜索测试\', \'用例编号\': \'搜索测试01\', \'输入说明\': \'1.输入:\', \'操作输入值\': \'温一壶清酒 博客园\', \'按钮操作\': \'2.点击搜索按钮\', \'断言\': \'温一壶清酒 博客园_百度搜索\', \'预期值\': \'正确断言\', \'测试序号\': 1.0, \'\': 44035.0}
{\'模块\': \'\', \'用例编号\': \'搜索测试02\', \'输入说明\': \'\', \'操作输入值\': \'温一壶清酒 抓包\', \'按钮操作\': \'\', \'断言\': \'\', \'预期值\': \'错误断言\', \'测试序号\': 2.0, \'\': 1}
{\'模块\': \'\', \'用例编号\': \'搜索测试03\', \'输入说明\': \'\', \'操作输入值\': \'温一壶清酒 appium\', \'按钮操作\': \'\', \'断言\': \'温一壶清酒 博客园\', \'预期值\': \'\', \'测试序号\': 3.0, \'\': \'\'}

我们从上述的打印数据来看,细心的博友们应该发现问题了。

我们可以看出,合并单元格的数据,只有首个单元格有数据显示;单元格数据类型不正确。

数据转换

今天,我们先来解决单元格数据类型显示不正确的问题。

解决该问题的方式很简单,就是将单元格类型获取后,判断单元格类型并转换格式,示例代码如下所示:

\'\'\'
@author: wenyihuqingjiu
@project: Pytest_Auto_Test
@file: demo
@time: 2020-07-23 21:40
@desc:
\'\'\'

import xlrd
import os
import datetime
from xlrd import xldate_as_tuple

current_path = os.path.dirname(__file__)
excel_path = os.path.join(current_path, \'..\', \'file/excel/testcase.xlsx\')
excel_object = xlrd.open_workbook(excel_path)
sheet_object = excel_object.sheet_by_name(\'Sheet1\')

# 获取有效行数
row_index = sheet_object.nrows
# 获取有效列数
col_index = sheet_object.ncols
# 定义空数组
all_data_list = []
# 获取首行数据
first_row = sheet_object.row_values(0)
for row in range(1, row_index):
    # 定义空列表
    row_dict = {}
    for col in range(col_index):
        c_cell = sheet_object.cell_value(row, col)
        # 获取单元格数据类型
        c_type = sheet_object.cell(row, col).ctype
        if c_type == 2 and c_cell % 1 == 0:  # 如果是整形
            c_cell = int(c_cell)
        elif c_type == 3:
            # 转成datetime对象
            date = datetime.datetime(*xldate_as_tuple(c_cell, 0))
            c_cell = date.strftime(\'%Y-%m-%d\')
        elif c_type == 4:
            c_cell = True if c_cell == 1 else False
        # 循环每一个有效的单元格,将字段与值对应存储到字典中
        row_dict[first_row[col]] = c_cell
    # 再将字典追加到列表中
    all_data_list.append(row_dict)
    print(str(row_dict))

运行代码,我们再来看下打印结果,如下所示:

{\'模块\': \'搜索测试\', \'用例编号\': \'搜索测试01\', \'输入说明\': \'1.输入:\', \'操作输入值\': \'温一壶清酒 博客园\', \'按钮操作\': \'2.点击搜索按钮\', \'断言\': \'温一壶清酒 博客园_百度搜索\', \'预期值\': \'正确断言\', \'测试序号\': 1, \'\': \'2020-07-23\'}
{\'模块\': \'\', \'用例编号\': \'搜索测试02\', \'输入说明\': \'\', \'操作输入值\': \'温一壶清酒 抓包\', \'按钮操作\': \'\', \'断言\': \'\', \'预期值\': \'错误断言\', \'测试序号\': 2, \'\': True}
{\'模块\': \'\', \'用例编号\': \'搜索测试03\', \'输入说明\': \'\', \'操作输入值\': \'温一壶清酒 appium\', \'按钮操作\': \'\', \'断言\': \'温一壶清酒 博客园\', \'预期值\': \'\', \'测试序号\': 3, \'\': \'\'}

格式转换后,数据就打印正常了,数据类型错误的问题就解决了。

好了,今天的分享就到此结束啦,希望对博友们有帮助,下期分享获取合并单元格数据问题,敬请期待。