龙空技术网

Python自动化测试之Excel文件操作

苏沐OvO 2105

前言:

现在大家对“excel根据文件名读取单元格值”可能比较重视,姐妹们都想要了解一些“excel根据文件名读取单元格值”的相关知识。那么小编在网络上搜集了一些对于“excel根据文件名读取单元格值””的相关文章,希望同学们能喜欢,同学们一起来学习一下吧!

首先为什么要讲Excel文件操作呢,因为一般用到的自动化框架中大部分都是用Excel存放测试用例数据,是数据驱动的来源。

安装

本次分享的是通过openpyxl 这个模块来操作Excel文件,所以首先需要安装该模块,通过下面命令安装。

pip install openpyxl 

(左右滑动查看完整代码)

表结构

首先Excel表格里有sheet表、单元格,在Python里可以把整张Excel表 sheet表、单元格都当做是个对象,我们这里先准备一份练手的Excel文件并简单写一些数据进去。

操作步骤

1、单独读取单元格值

from openpyxl import load_workbook# 从openpyxl中导入 load_workbook#load_workbook参数:#filename:文件名称,#read_only:只读模式#data_only:读取数据的时候如果遇到计算公式是否要计算后再读取数据,False:读取未计算的结果(公式读出来) True:读取计算后的记过#一般读取用例只需要用到filename这个参数就行,如果测试用例文件在当前目录下只需传文件名,#如果在其他目录下需要传文件的绝对路径不然会提示找不到文件# 获取表格对象wb = load_workbook(filename="test.xlsx")# 获取sheet对象sheet = wb["sheet1"]# 获取单元格对象cell = sheet["B2"]# 获取值print(cell.value)# 关闭数据流,注意:每次操作完文件后都需要关闭操作流,不然下次操作会报错。wb.close()

(左右滑动查看完整代码)

运行结果如下:

可以看到这时候已经把test文件中sheet1表里的B2单元格已经读出来了。

2、行操作

# 获取行数据用的是iter_rows()这个方法这个方法有五个参数(min_row,max_row,min_col,max_col,values_only)# min_row=None: 行的起始索引值,默认是1,必须是int类型 # max_row=None: 行的结束索引值,默认是1,必须是int类型# min_col=None: 列的起始索引值,默认是1,必须是int类型# max_col=None:列的结束索引值,默认是1,必须是int类型 # values_only=False:返回是对象 ,True:返回对应的值 默认是False 我们用到的时候需要设置成Trueresult = sheet.iter_rows(values_only=True) # 不填写min_row,max_row,min_col,max_col 则返回所有行列数据print(result)# 这里如果想用数据得转换成list 因为读取出来的数据默认是返回的内存地址需要转成listprint(list(result)) # 可遍历输出数据# for i in list(result):#     print(i)

(左右滑动查看完整代码)

运行结果如下:

3、列操作(一般用不着,竖着读数据)

# 这里的操作参数和行操作一致# 获取列数据用的是iter_cols()这个方法这个方法一样有五个参数用法与行操作一致(min_row=1,max_row=2,min_col=1,max_col=4,values_only=True)result = sheet.iter_cols(min_row=1,max_row=2,min_col=1,max_col=4,values_only=True)print(result)print(list(result))

(左右滑动查看完整代码)

运行结果如下:

在做自动化测试的时候用到的都是行读取,列读取我们稍作了解就行。

但是我们现在通过行读取出来的数据并不能直接用作数据驱动使用,所以我们得进一步把数据处理成下面这种格式。

代码如下:

# 要把数据处理成上图的效果,一个字典对应一条用例 我们需要先把标题行 和用例数据先拆出来,并用zip函数进行压缩 再转成字典形式result = list(sheet.iter_rows(values_only=True))title = result [0] # 通过之前返回数据可以看出第一行是标题 我们直接索引获取第一行datas = result [1:] # 剩下除了第一行就是用例数据了 直接截取# print(title)# print(data_list)case_list=[] # 定义一个空列表保存用例for case in datas:    res = dict(zip(title,case))    print(res)    case_list.append(res) # 将处理好的字典用例保存到列表中print(case_list)# 打印结果

(左右滑动查看完整代码)

运行结果如下:

这时候已经可以拿到用例并处理成自己想要的格式了,但是这种代码比较low是写死的,这时候我们就可以封装一下,封装成类和方法。代码如下:

from openpyxl import load_workbookimport osclass GetExcel():    def __init__(self,filename,sheet_name):        # 获取用例文件绝对路径 casedata存放用例的目录 test是当前文件运行的目录 demo.py是封装的文件替换一下就是用例文件的绝对路径了        # 如果你用例文件和封装文件在同一个目录下 只需要传文件名即可        # filepath = os.path.abspath(__file__).replace("casedata","test").replace("demo.py",filename)        # print(filepath)        # 获取表        #  self.wb = load_workbook(filename=filepath)        self.wb = load_workbook(filename=filename)        # 获取表对象        self.sheet = self.wb[sheet_name]    def __close_excel(self):        self.wb.close()    def __get_title_and_data(self):        """        获取表头和用例并返回        :return:        """        data_list = list(self.sheet.iter_rows(values_only=True))        title = data_list[0]        datas = data_list[1:]        return title,datas    def get_data(self):        """        获取所有用例并保存到列表中        :return:        """        title,data = self.__get_title_and_data()        case_list = []        for i in data:            res = dict(zip(title, i))            case_list.append(res)        self.__close_excel()        return case_listif __name__ == '__main__':    cl = GetExcel('test.xlsx','sheet1')    res = cl.get_data()    print(res)

(左右滑动查看完整代码)

运行结果如下:

通过封装成类和方法,我们可以反复使用读取不同的Excel表里用例数据。

下次分享如何用读取出来的数据通过数据驱动的方式去执行。

-End-

更多相关资料有需要的小伙伴回 01 即可获取

标签: #excel根据文件名读取单元格值