自动化不知如何参数化？xlrd来帮你解决

平时在做自动化测试的时候，一直都是要求数据与业务逻辑分离。把测试数据都写在业务里面的话，比较混杂。为了方便管理测试数据，所以引入了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, '': ''}

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

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

巴特西