一篇文章教你自动化测试如何解析excel文件?

前言
自动化测试中我们存放数据无非是使用文件或者数据库,那么文件可以是csv,xlsx,xml,甚至是txt文件,通常excel文件往往是我们的首选,无论是编写测试用例还是存放测试数据,excel都是很方便的。那么今天我们就把不同模块处理excel文件的方法做个总结,直接做封装,方便我们以后直接使用,增加工作效率。

一、openpyxl
openpyxl是个第三方库,首先我们使用命令 pip install openpyxl 直接安装

注:openpyxl操作excel时,行号和列号都是从1开始计算的

封装代码

"""
from openpyxl import load_workbook
from openpyxl.styles import Font
from openpyxl.styles.colors import BLACK
from collections import namedtuple
class ParseExcel(object):"""解析excel文件"""def __init__(self, filename, sheet_name=None):try:self.filename = filenameself.sheet_name = sheet_nameself.wb = load_workbook(self.filename)if self.sheet_name is None:self.work_sheet = self.wb.activeelse:self.work_sheet = self.wb[self.sheet_name]except FileNotFoundError as e:raise edef get_max_row_num(self):"""获取最大行号"""max_row_num = self.work_sheet.max_rowreturn max_row_numdef get_max_column_num(self):"""获取最大列号"""max_column = self.work_sheet.max_columnreturn max_columndef get_cell_value(self, coordinate=None, row=None, column=None):"""获取指定单元格的数据"""if coordinate is not None:try:return self.work_sheet[coordinate].valueexcept Exception as e:raise eelif coordinate is None and row is not None and column is not None:if isinstance(row, int) and isinstance(column, int):return self.work_sheet.cell(row=row, column=column).valueelse:raise TypeError('row and column must be type int')else:raise Exception("Insufficient Coordinate of cell!")def get_row_value(self, row):"""获取某一行的数据"""column_num = self.get_max_column_num()row_value = []if isinstance(row, int):for column in range(1, column_num + 1):values_row = self.work_sheet.cell(row, column).valuerow_value.append(values_row)return row_valueelse:raise TypeError('row must be type int')def get_column_value(self, column):"""获取某一列数据"""row_num = self.get_max_column_num()column_value = []if isinstance(column, int):for row in range(1, row_num + 1):values_column = self.work_sheet.cell(row, column).valuecolumn_value.append(values_column)return column_valueelse:raise TypeError('column must be type int')def get_all_value_1(self):"""获取指定表单的所有数据(除去表头)"""max_row_num = self.get_max_row_num()max_column = self.get_max_column_num()values = []for row in range(2, max_row_num + 1):value_list = []for column in range(1, max_column + 1):value = self.work_sheet.cell(row, column).valuevalue_list.append(value)values.append(value_list)return valuesdef get_all_value_2(self):"""获取指定表单的所有数据(除去表头)"""rows_obj = self.work_sheet.iter_rows(min_row=2, max_row=self.work_sheet.max_row,values_only=True)  # 指定values_only 会直接提取数据不需要再使用cell().valuevalues = []for row_tuple in rows_obj:value_list = []for value in row_tuple:value_list.append(value)values.append(value_list)return valuesdef get_excel_title(self):"""获取sheet表头"""title_key = tuple(self.work_sheet.iter_rows(max_row=1, values_only=True))[0]return title_keydef get_listdict_all_value(self):"""获取所有数据,返回嵌套字典的列表"""sheet_title = self.get_excel_title()all_values = self.get_all_value_2()value_list = []for value in all_values:value_list.append(dict(zip(sheet_title, value)))return value_listdef get_list_nametuple_all_value(self):"""获取所有数据,返回嵌套命名元组的列表"""sheet_title = self.get_excel_title()values = self.get_all_value_2()excel = namedtuple('excel', sheet_title)value_list = []for value in values:e = excel(*value)value_list.append(e)return value_listdef write_cell(self, row, column, value=None, bold=True, color=BLACK):"""指定单元格写入数据:param work_sheet::param row: 行号:param column: 列号:param value: 待写入数据:param bold: 加粗, 默认加粗:param color: 字体颜色,默认黑色:return:"""try:if isinstance(row, int) and isinstance(column, int):cell_obj = self.work_sheet.cell(row, column)cell_obj.font = Font(color=color, bold=bold)cell_obj.value = valueself.wb.save(self.filename)else:raise TypeError('row and column must be type int')except Exception as e:raise e
if __name__ == '__main__':pe = ParseExcel('testdata.xlsx')# sheet = pe.get_sheet_object('testcase')column_row = pe.get_max_column_num()print('最大列号:', column_row)max_row = pe.get_max_row_num()print('最大行号:', max_row)#cell_value_1 = pe.get_cell_value(row=2, column=3)print('第%d行, 第%d列的数据为: %s' % (2, 3, cell_value_1))cell_value_2 = pe.get_cell_value(coordinate='A5')print('A5单元格的数据为: {}'.format(cell_value_2))value_row = pe.get_row_value(3)print('第{}行的数据为:{}'.format(3, value_row))value_column = pe.get_column_value(2)print('第{}列的数据为:{}'.format(2, value_column))#values_1 = pe.get_all_value_1()print('第一种方式获取所有数据\n', values_1)values_2 = pe.get_all_value_2()print('第二种方式获取所有数据\n', values_2)title = pe.get_excel_title()print('表头为\n{}'.format(title))dict_value = pe.get_listdict_all_value()print('所有数据组成的嵌套字典的列表:\n', dict_value)#namedtuple_value = pe.get_list_nametuple_all_value()print('所有数据组成的嵌套命名元组的列表:\n', namedtuple_value)pe.write_cell(1, 2, 'Tc_title')

上面这个封装如如果用来同时操作同一个excel文件的两个sheet写入数据时,会有点小bug(写完后你会发现两个表单有一个是没有数据的)

其实原因很简单:不同对象拥有自己独立的属性, 当你写操作的时候其实每个对象只针对自己的表单做了保存,所以最后一个对象写完数据后,只保存了自己的表单,其他的对象的表单实际是没有保存的。针对这个问题,对上面封装的代码进行了轻微改动

"""
from openpyxl import load_workbook
from openpyxl.styles import Font
from openpyxl.styles.colors import BLACK
from collections import namedtupleclass ParseExcel(object):"""解析excel文件"""def __init__(self, filename):try:self.filename = filenameself.__wb = load_workbook(self.filename)except FileNotFoundError as e:raise edef get_max_row_num(self, sheet_name):"""获取最大行号"""max_row_num = self.__wb[sheet_name].max_rowreturn max_row_numdef get_max_column_num(self, sheet_name):"""获取最大列号"""max_column = self.__wb[sheet_name].max_columnreturn max_columndef get_cell_value(self, sheet_name, coordinate=None, row=None, column=None):"""获取指定单元格的数据"""if coordinate is not None:try:return self.__wb[sheet_name][coordinate].valueexcept Exception as e:raise eelif coordinate is None and row is not None and column is not None:if isinstance(row, int) and isinstance(column, int):return self.__wb[sheet_name].cell(row=row, column=column).valueelse:raise TypeError('row and column must be type int')else:raise Exception("Insufficient Coordinate of cell!")def get_row_value(self, sheet_name, row):"""获取某一行的数据"""column_num = self.get_max_column_num(sheet_name)row_value = []if isinstance(row, int):for column in range(1, column_num + 1):values_row = self.__wb[sheet_name].cell(row, column).valuerow_value.append(values_row)return row_valueelse:raise TypeError('row must be type int')def get_column_value(self, sheet_name, column):"""获取某一列数据"""row_num = self.get_max_column_num(sheet_name)column_value = []if isinstance(column, int):for row in range(1, row_num + 1):values_column = self.__wb[sheet_name].cell(row, column).valuecolumn_value.append(values_column)return column_valueelse:raise TypeError('column must be type int')def get_all_value_1(self, sheet_name):"""获取指定表单的所有数据(除去表头)"""max_row_num = self.get_max_row_num(sheet_name)max_column = self.get_max_column_num(sheet_name)values = []for row in range(2, max_row_num + 1):value_list = []for column in range(1, max_column + 1):value = self.__wb[sheet_name].cell(row, column).valuevalue_list.append(value)values.append(value_list)return valuesdef get_all_value_2(self, sheet_name):"""获取指定表单的所有数据(除去表头)"""rows_obj = self.__wb[sheet_name].iter_rows(min_row=2, max_row=self.__wb[sheet_name].max_row, values_only=True)values = []for row_tuple in rows_obj:value_list = []for value in row_tuple:value_list.append(value)values.append(value_list)return valuesdef get_excel_title(self, sheet_name):"""获取sheet表头"""title_key = tuple(self.__wb[sheet_name].iter_rows(max_row=1, values_only=True))[0]return title_keydef get_listdict_all_value(self, sheet_name):"""获取所有数据,返回嵌套字典的列表"""sheet_title = self.get_excel_title(sheet_name)all_values = self.get_all_value_2(sheet_name)value_list = []for value in all_values:value_list.append(dict(zip(sheet_title, value)))return value_listdef get_list_nametuple_all_value(self, sheet_name):"""获取所有数据,返回嵌套命名元组的列表"""sheet_title = self.get_excel_title(sheet_name)values = self.get_all_value_2(sheet_name)excel = namedtuple('excel', sheet_title)value_list = []for value in values:e = excel(*value)value_list.append(e)return value_listdef write_cell(self, sheet_name, row, column, value=None, bold=True, color=BLACK):if isinstance(row, int) and isinstance(column, int):try:cell_obj = self.__wb[sheet_name].cell(row, column)cell_obj.font = Font(color=color, bold=bold)cell_obj.value = valueself.__wb.save(self.filename)except Exception as e:raise eelse:raise TypeError('row and column must be type int')if __name__ == '__main__':pe = ParseExcel('testdata.xlsx')print(pe.get_all_value_2('division'))print(pe.get_list_nametuple_all_value('division'))column_row = pe.get_max_column_num('division')print('最大列号:', column_row)max_row = pe.get_max_row_num('division')print('最大行号:', max_row)cell_value_1 = pe.get_cell_value('division', row=2, column=3)print('第%d行, 第%d列的数据为: %s' % (2, 3, cell_value_1))cell_value_2 = pe.get_cell_value('division', coordinate='A5')print('A5单元格的数据为: {}'.format(cell_value_2))value_row = pe.get_row_value('division', 3)print('第{}行的数据为:{}'.format(3, value_row))value_column = pe.get_column_value('division', 2)print('第{}列的数据为:{}'.format(2, value_column))values_1 = pe.get_all_value_1('division')print('第一种方式获取所有数据\n', values_1)values_2 = pe.get_all_value_2('division')print('第二种方式获取所有数据\n', values_2)title = pe.get_excel_title('division')print('表头为\n{}'.format(title))dict_value = pe.get_listdict_all_value('division')print('所有数据组成的嵌套字典的列表:\n', dict_value)namedtuple_value = pe.get_list_nametuple_all_value('division')print('所有数据组成的嵌套命名元组的列表:\n', namedtuple_value)pe.write_cell('division', 1, 2, 'Tc_title')

二、xlrd
安装xlrd,此模块只支持读操作, 如果要写需要使用xlwt或者使用xlutils配合xlrd, 但是使用xlwt只能对新的excel文件进行写操作,无法对原有文件进行写, 所以这里选择使用xlutils

但是还有一个问题就是,如果使用xlutils, 那么我们的excel文件需要以.xls 为后缀。因为以xlsx为后缀无法实现写,会报错(亲测,因为formatting_info参数还没有对新版本的xlsx的格式完成兼容)

注:xlrd操作excel时,行号和列号都是从0开始计算的

封装代码

"""
import xlrd
from xlutils import copy
from collections import namedtupleclass ParseExcel(object):# xlrd 解析excel, 行号和列号都是从0开始的def __init__(self, filename, sheet):try:self.filename = filenameself.sheet = sheetself.wb = xlrd.open_workbook(self.filename, formatting_info=True)if isinstance(sheet, str):self.sheet = self.wb.sheet_by_name(sheet)elif isinstance(sheet, int):self.sheet = self.wb.sheet_by_index(sheet)else:raise TypeError('sheet must be int or str')except Exception as e:raise edef get_max_row(self):"""获取表单的最大行号"""max_row_num = self.sheet.nrowsreturn max_row_numdef get_max_column(self):"""获取表单的最大列号"""min_row_num = self.sheet.ncolsreturn min_row_numdef get_cell_value(self, row, column):"""获取某个单元格的数据"""if isinstance(row, int) and isinstance(column, int):values = self.sheet.cell(row-1, column-1).valuereturn valueselse:raise TypeError('row and column must be type int')def get_row_values(self, row):"""获取某一行的数据"""if isinstance(row, int):values = self.sheet.row_values(row-1)return valueselse:raise TypeError('row must be type int')def get_column_values(self, column):"""获取某一列的数据"""if isinstance(column, int):values = self.sheet.col_values(column-1)return valueselse:raise TypeError('column must be type int')def get_table_title(self):"""获取表头"""table_title = self.get_row_values(1)return table_titledef get_all_values_dict(self):"""获取所有的数据,不包括表头,返回一个嵌套字典的列表"""max_row = self.get_max_row()table_title = self.get_table_title()value_list = []for row in range(2, max_row):values = self.get_row_values(row)value_list.append(dict(zip(table_title, values)))return value_listdef get_all_values_nametuple(self):"""获取所有的数据,不包括表头,返回一个嵌套命名元组的列表"""table_title = self.get_table_title()max_row = self.get_max_row()excel = namedtuple('excel', table_title)value_list = []for row in range(2, max_row):values = self.get_row_values(row)e = excel(*values)value_list.append(e)return value_listdef write_value(self, sheet_index, row, column, value):"""写入某个单元格数据"""if isinstance(row, int) and isinstance(column, int):if isinstance(sheet_index, int):wb = copy.copy(self.wb)worksheet = wb.get_sheet(sheet_index)worksheet.write(row-1, column-1, value)wb.save(self.filename)else:raise TypeError('{} must be int'.format(sheet_index))else:raise TypeError('{} and {} must be int'.format(row, column))if __name__ == '__main__':pe = ParseExcel('testdata.xls', 'testcase')print('最大行号:', pe.get_max_row())print('最大列号:', pe.get_max_column())print('第2行第3列数据:', pe.get_cell_value(2, 3))print('第2行数据', pe.get_row_values(2))print('第3列数据', pe.get_column_values(3))print('表头:', pe.get_table_title())print('所有的数据返回嵌套字典的列表:', pe.get_all_values_dict())print('所有的数据返回嵌套命名元组的列表:', pe.get_all_values_nametuple())pe.write_value(0, 1, 3, 'test')

三、pandas
pandas是一个做数据分析的库, 总是感觉在自动化测试中使用pandas解析excel文件读取数据有点大材小用,不论怎样吧,还是把pandas解析excel文件写一下把

我这里只封装了读,写的话我这有点小问题,后面改好再追加代码吧。

请先pip install pandas安装pandas

封装代码

"""
import pandas as pd
class ParseExcel(object):def __init__(self, filename, sheet_name=None):try:self.filename = filenameself.sheet_name = sheet_nameself.df = pd.read_excel(self.filename, self.sheet_name)except Exception as e:raise edef get_row_num(self):"""获取行号组成的列表, 从0开始的"""row_num_list = self.df.index.valuesreturn row_num_listdef get_cell_value(self, row, column):"""获取某一个单元格的数据"""try:if isinstance(row, int) and isinstance(column, int):cell_value = self.df.ix[row-2, column-1] # ix的行参数是按照有效数据行,且从0开始return cell_valueelse:raise TypeError('row and column must be type int')except Exception as e:raise edef get_table_title(self):"""获取表头, 返回列表"""table_title = self.df.columns.valuesreturn table_titledef get_row_value(self, row):"""获取某一行的数据, 行号从1开始"""try:if isinstance(row, int):row_data = self.df.ix[row-2].valuesreturn row_dataelse:raise TypeError('row must be type int')except Exception as e:raise edef get_column_value(self, col_name):"""获取某一列数据"""try:if isinstance(col_name, str):col_data = self.df[col_name].valuesreturn col_dataelse:raise TypeError('col_name must be type str')except Exception as e:raise edef get_all_value(self):"""获取所有的数据,不包括表头, 返回嵌套字典的列表"""rows_num = self.get_row_num()table_title = self.get_table_title()values_list = []for i in rows_num:row_data = self.df.ix[i, table_title].to_dict()values_list.append(row_data)return values_list
if __name__ == '__main__':pe = ParseExcel('testdata.xlsx', 'testcase')print(pe.get_row_num())print(pe.get_table_title())print(pe.get_all_value())print(pe.get_row_value(2))print(pe.get_cell_value(2, 3))print(pe.get_column_value('Tc_title'))

总结
使用了3种方法,4个库 xlrd,openpyxl,xlwt,pandas 操作excel文件,个人感觉还是使用openpyxl比较适合在自动化中使用,当然不同人有不同选择,用哪个区别也不是很大。

以上3种方法,都可以拿来直接使用,不需要再做封装了 !

最后感谢每一个认真阅读我文章的人,礼尚往来总是要有的,虽然不是什么很值钱的东西,如果你用得到的话可以直接拿走:

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.rhkb.cn/news/145315.html

如若内容造成侵权/违法违规/事实不符,请联系长河编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

SpringBoot使用Docker并上传至DockerHub

我的新书《Android App开发入门与实战》已于2020年8月由人民邮电出版社出版,欢迎购买。点击进入详情 文章目录 1.系列文章2.构建docker镜像的方式3.docker操作3.1 安装docker3.2 查看docker镜像3.3 本地运行docker3.4 修改tag3.5 推送docker镜像3.6 远端server拉取d…

Linux 集锦 之 最常用的几个命令

Linux最常用的几个命令 ​ Linux系统中的命令那是相当地丰富,不同的版本可能还有不同的命令,不过Linux核心自带的命令大概有几百个,这个不管是什么发行版一般都是共用的。 ​ 如果希望探索Linux的所有命令,可能不太实际&#xf…

树莓派基本配置(2)

安装motion $sudo apt-get update $sudo apt-get install motion配置motion sudo nano /etc/default/motionsudo nano /etc/motion/motion.conf主要改这些参数 //让Motion作为守护进程运行 daemon on ... //用这个端口号来读取数据 stream_port 8081 ... //网络上其它主机…

力扣刷题-哈希表-求两个数组的交集

349 求两个数组的交集 题意&#xff1a;给定两个数组&#xff0c;编写一个函数来计算它们的交集。注意&#xff1a;输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。 提示&#xff1a; 1 < nums1.length, nums2.length < 1000 0 < nums1[i], …

nodejs在pdf中绘制表格

需求 之前我已经了解过如何在pdf模板中填写字段了 nodejs根据pdf模板填入中文数据并生成新的pdf文件https://blog.csdn.net/ArmadaDK/article/details/132456324 但是当我具体使用的时候&#xff0c;我发现我的模板里面有表格&#xff0c;表格的长度是不固定的&#xff0c;所…

WPF 03

staticResource和dynamicResource的区别 首先看一个案例 MainWindow.xaml <Window x:Class"WpfDay03.MainWindow"xmlns"http://schemas.microsoft.com/winfx/2006/xaml/presentation"xmlns:x"http://schemas.microsoft.com/winfx/2006/xaml&quo…

Springboot中slf4j日志的简单应用

1、注入依赖&#xff08;pom.xml&#xff09; <!-- https://mvnrepository.com/artifact/org.slf4j/slf4j-api --> <dependency><groupId>org.slf4j</groupId><artifactId>slf4j-api</artifactId><version>2.0.9</version> &…

从MVC到DDD,该如何下手重构?

作者&#xff1a;付政委 博客&#xff1a;bugstack.cn 沉淀、分享、成长&#xff0c;让自己和他人都能有所收获&#xff01;&#x1f604; 大家好&#xff0c;我是技术UP主小傅哥。多年的 DDD 应用&#xff0c;使我开了技术的眼界&#xff01; MVC 旧工程腐化严重&#xff0c;…

排序算法之【希尔排序】

&#x1f4d9;作者简介&#xff1a; 清水加冰&#xff0c;目前大二在读&#xff0c;正在学习C/C、Python、操作系统、数据库等。 &#x1f4d8;相关专栏&#xff1a;C语言初阶、C语言进阶、C语言刷题训练营、数据结构刷题训练营、有感兴趣的可以看一看。 欢迎点赞 &#x1f44d…

C++ -- 学习系列 std::deque 的原理与使用

一 deque 是什么? std::deque 是 c 一种序列式容器&#xff0c;其与 vector 类似&#xff0c;其底层内存都是连续的&#xff0c;不同的地方在于&#xff0c; vector 是一端开口&#xff0c;在一端放入数据与扩充空间&#xff0c;而 deque 是双端均开口&#xff0c;都可以放…

3D孪生场景搭建:模型区域摆放

前面介绍完了NSDT场景编辑器的线性绘制和阵列绘制&#xff0c;本章将讲述下编辑器的另一种绘制方式&#xff1a;区域绘制。 1、区域绘制功能简介 在场景中绘制资产时&#xff0c;除使用上述两个的方式外&#xff0c;NSDT 编辑器还支持使用区域绘制的方式进行绘制。先选取需要…

GEO生信数据挖掘(一)数据集下载和初步观察

检索到目标数据集后&#xff0c;开始数据挖掘&#xff0c;本文以阿尔兹海默症数据集GSE1297为例 目录 GEOquery 简介 安装并加载GEOquery包 getGEO函数获取数据&#xff08;联网下载&#xff09; 更换下载数据源 对数据集进行初步观察处理 GEOquery 简介 GEOquery是一个…

聊聊并发编程——并发容器和阻塞队列

目录 一.ConcurrentHashMap 1.为什么要使用ConcurrentHashMap&#xff1f; 2.ConcurrentHashMap的类图 3.ConcurrentHashMap的结构图 二.阻塞队列 Java中的7个阻塞队列 ArrayBlockingQueue&#xff1a;一个由数组结构组成的有界阻塞队列。 LinkedBlockingQueue&#xf…

用go实现http服务端和请求端

一、概述 本文旨在学习记录下如何用go实现建立一个http服务器&#xff0c;同时构造一个专用格式的http客户端。 二、代码实现 2.1 构造http服务端 1、http服务处理流程 基于HTTP构建的服务标准模型包括两个端&#xff0c;客户端(Client)和服务端(Server)。HTTP 请求从客户端…

PHP8的静态变量和方法-PHP8知识详解

我们在上一课程讲到了public、private、protected这3个关键字&#xff0c;今天我们来讲解static关键字&#xff0c;明天再讲解final关键字。 如果不想通过创建对象来调用变量或方法&#xff0c;则可以将该变量或方法创建为静态变量或方法&#xff0c;也就是在变量或方法的前面…

【PyTorch实战演练】使用Cifar10数据集训练LeNet5网络并实现图像分类(附代码)

文章目录 0. 前言1. Cifar10数据集1.1 Cifar10数据集下载1.2 Cifar10数据集解析 2. LeNet5网络2.1 LeNet5的网络结构2.2 基于PyTorch的LeNet5网络编码 3. LeNet5网络训练及输出验证3.1 LeNet5网络训练3.2 LeNet5网络验证 4. 完整代码4.1 训练代码4.1 验证代码 0. 前言 按照国际…

C语言文件操作与管理

一、为什么使用文件 在我们前面练习使用结构体时&#xff0c;写通讯录的程序&#xff0c;当通讯录运行起来的时候&#xff0c;可以给通讯录中增加、删除数据&#xff0c;此时数据是存放在内存中&#xff0c;当程序退出的时候&#xff0c;通讯录中的数据自然就不存在了&#xff…

Java 基于 SpringBoot 的在线学习平台

1 简介 基于SpringBoot的Java学习平台&#xff0c;通过这个系统能够满足学习信息的管理及学生和教师的学习管理功能。系统的主要功能包括首页&#xff0c;个人中心&#xff0c;学生管理&#xff0c;教师管理&#xff0c;课程信息管理&#xff0c;类型管理&#xff0c;作业信息…

大数据Doris(一):Doris概述篇

文章目录 Doris概述篇 一、前言 二、Doris简介

队列的各个函数的实现

1.第一个结构是存放链表的数据&#xff0c;第二个结构体是存放头节点和尾节点的以方便找到尾节点&#xff0c;存放头节点的是phead&#xff0c;尾节点的是ptail typedef struct QueueNode {struct QueueNode* next;//单链表QDataType data;//放数据 }QNode;typedef struct Queu…