Python 接口测试之Excel表格数据操作方法封装

2023-12-18 21:33:04

引言
我们在做接口测试,经常会用到excel去管理测试数据,对Excel的操作比较频繁,那么使用python如何操作Excel文件的读与写呢?由于之前讲的都是大的框框,没有讲这么小的模块使用,现在就化整为0的讲解。

读写模块介绍
python有三个模块是对Excel文件的操作,分别是:openpyxl,xlrd和xlwt。下面简单的介绍一下各个模块的特点及使用案例。

xlrd
xlrd是用来从Excel中读写数据的,但我平常只用它进行读操作,写操作会遇到些问题。用xlrd进行读取比较方便,流程和平常手动操作Excel一样,打开工作簿(Workbook),选择工作表(sheets),然后操作单元格(cell)。下面举个例子,例如要打开当前目录下名为”data.xlsx”的Excel文件,选择第一张工作表,然后读取第一行的全部内容并打印出来。Python代码如下:

import os
import xlrd
from xlutils.copy import copy
 
# 获取当前文件的绝对路径
curPath = os.path.abspath(os.path.dirname(__file__))
print(curPath)
# 获取项目根目录
rootPath = os.path.abspath(os.path.dirname(curPath))
print(rootPath)
 
# 获取文件路径
file_path = r'APItest_ddt\data\data.xls'
file_path = os.path.join(rootPath,file_path)
print(file_path)
#打开excel文件
data=xlrd.open_workbook(file_path)
#获取第一张工作表(通过索引的方式)
table=data.sheets()[0]
#data_list用来存放数据
data_list=[]
#将table中第一行的数据读取并添加到data_list中
data_list.extend(table.row_values(0))
#打印出第一行的全部数据
for item in data_list:
    print(item)
 
 
# 写入数据
copy_data = copy(data)
# 读取复制的excel的sheet页
copy_data_to_sheet = copy_data.get_sheet(0)
# 通过get_sheet()获取的sheet有write()方法,写入数据
copy_data_to_sheet.write(5,5,"这是我写入的测试数据:哈哈")
# 保存数据
copy_data.save(file_path)

运行结果:

C:\Users\Administrator\Desktop\APItest_ddt
C:\Users\Administrator\Desktop
C:\Users\Administrator\Desktop\APItest_ddt/data/data.xls
地址

打开data.xls文件:

在处理excel数据时发现了xlwt的局限性–不能写入超过65535行、256列的数据(因为它只支持Excel 2003及之前的版本,在这些版本的Excel中行数和列数有此限制),这对于实际应用还是不够的。为此经过一番寻找发现了一个支持07/10/13版本Excel的openpyxl,虽然功能很强大,但是操作起来感觉没有xlwt方便。以上是xlrd的几个简单操作,并且可以发现,xlrd可以读,也可以写的。只是写的话,没那么方便。

xlwt
如果说xlrd不是一个单纯的Reader(如果把xlrd中的后两个字符看成Reader,那么xlwt后两个字符类似看成Writer),那么xlwt就是一个纯粹的Writer了,因为它只能对Excel进行写操作。xlwt和xlrd不光名字像,连很多函数和操作格式也是完全相同。下面简要归纳一下常用操作。

import xlwt
# 新建一个Excel文件(只能通过新建写入)
data=xlwt.Workbook()
# 新建一个工作表
table=data.add_sheet('name')
# 写入数据到A1单元格
table.write(0,0,u'呵呵')
# 注意:如果对同一个单元格重复操作,会引发overwrite Exception,想要取消该功能,需要在添加工作表时指定为可覆盖,像下面这样
table=data.add_sheet('sheet1',cell_overwrite_ok=True)
# 保存
data.save('test.xls')

?查看结果:

openpyxl

该模块支持最新版的Excel文件格式,对Excel文件具有响应的读写操作,对此有专门的Reader和Writer两个类,便于对Excel文件的操作。虽然如此,但我一般还是用默认的workbook来进行操作。常用操作归纳如下:

from openpyxl.reader.excel import load_workbook
 
wb=load_workbook(filename=r"openpyxl_test.xlsx")
 
import openpyxl
 
if __name__ == '__main__':
 
    wb = load_workbook(filename=r"openpyxl_test.xlsx")
    # 显示所有工作表的名字
    sheet_name = wb.get_sheet_names()
    print(sheet_name)
    # 取第一张表
    sheetnames = wb.sheetnames
    ws = wb[sheetnames[0]]
    # 显示表名,表行数,表列数
    print("Work Sheet Title:", ws.title)
    print("Work Sheet Rows:", ws.max_row)
    print("Work Sheet Cols:", ws.max_column)
    # 获取指定行的值,如第三行
    row3_values = []
    row3_cell_list = list(ws.rows)[2]
    for cell in row3_cell_list:
        row3_values.append(cell.value)
    print(row3_values)  # output:['A3', 'B3', 'C3', 'D3']
 
    # 获取所有行的数据
    # 建立存储数据的字典
    data = {}
    # 获取表格所有值
    # 法1:
    for i in range(0, ws.max_row):
        every_row_values = []
        every_row_cell_list = list(ws.rows)[i]
        for cell in every_row_cell_list:
            every_row_values.append(cell.value)
        data[i + 1] = every_row_values
    print(data)
    # {1: ['A1', 'B1', 'C1', 'D1'], 2: ['A2', 'B2', 'C2', 'D2'], 3: ['A3', 'B3', 'C3', 'D3'], 4: ['A4', 'B4', 'C4', 'D4'], 5: ['A5', 'B5', 'C5', 'D5'],
    #  6: ['A6', 'B6', 'C6', 'D6'], 7: ['A7', 'B7', 'C7', 'D7'], 8: ['A8', 'B8', 'C8', 'D8'], 9: ['A9', 'B9', 'C9', 'D9'], 10: ['A10', 'B10', 'C10', 'D10'],
    # 11: ['A11', 'B11', 'C11', 'D11']}
    # 法2
    for row in ws.rows:
        line = [cell.value for cell in row]
    print(
        line)  # output:['A1', 'B1', 'C1', 'D1'],['A2', 'B2', 'C2', 'D2'], ['A3', 'B3', 'C3', 'D3'],['A4', 'B4', 'C4', 'D4'], ['A5', 'B5', 'C5', 'D5'],['A6', 'B6', 'C6', 'D6'],['A7', 'B7', 'C7', 'D7'],['A8', 'B8', 'C8', 'D8'],['A9', 'B9', 'C9', 'D9'],['A10', 'B10', 'C10', 'D10'],['A11', 'B11', 'C11', 'D11']
    # 获取某个区间的值,例:获得了以A3为左上角,C6为右下角矩形区域的所有单元格
    # 法1:使用range
    data1 = {}
    for i in range(2, 6):
        every_row_values = []
        every_row_cell_list = list(ws.rows)[i]
        for cell in every_row_cell_list:
            every_row_values.append(cell.value)
        data1[i + 1] = every_row_values
    print(data1)
    # output{3: ['A3', 'B3', 'C3', 'D3'], 4: ['A4', 'B4', 'C4', 'D4'], 5: ['A5', 'B5', 'C5', 'D5'], 6: ['A6', 'B6', 'C6', 'D6']}
 
    # 法2:使用切片
    data_list = []
    for row_cell in ws['A3':'C6']:
        every_row_value = []
        for cell in row_cell:
            every_row_value.append(cell.value)
        data_list.append(every_row_value)
    print(data_list)  # output:[['A3', 'B3', 'C3'], ['A4', 'B4', 'C4'], ['A5', 'B5', 'C5'], ['A6', 'B6', 'C6']]
 
 
 
 
 
    # 测试写入数据:
 
    wb = load_workbook(filename=r"openpyxl_test.xlsx")
    ws = wb.active
 
    # 第一个sheet是ws
    ws = wb.worksheets[0]
 
    # 设置ws的名称
    ws.title = "range names"
 
    # 向某个单元格中写入数据
    ws.cell(11,11).value = u'Test write data!!!'
 
    # 最后保存文件
    wb.save(filename=r"openpyxl_test.xlsx")

?结果如下:

封装读写

上面已经介绍了三个模块的使用,那么在搭建接口测试框架的时候,不可能这样写,一方面不利于代码可读性与维护,另一方面不雅观。所以,要对零散的代码进行二次封装,一下就是针对常用方法进行二次封装:

import xlrd
from xlutils.copy import copy
import os
# 获取当前文件的绝对路径
curPath = os.path.abspath(os.path.dirname(__file__))
print(curPath)
# 获取项目根目录
rootPath = os.path.abspath(os.path.dirname(curPath))
print(rootPath)
 
# 类中使用装饰器 @classmethod定义方法,是类方法
# 类中使用装饰器 @staticmethod定义方法,是静态方法
 
 
class Operate_Excel(object):
    """
    操作excel类
 
    """
    # 定义构造函数,创建对象自动执行
    def __init__(self,file_path=None,sheet_id=None):
        """
 
        :param file_path:如果没传值,默认为excel路径
        :param sheet_id:如果没传值,默认为第一个sheet页
        """
        if file_path:
            # 成员变量
            self.file_path = file_path
            self.sheet_id = sheet_id
        else:
            self.file_path = r'data/util_data/operate_excel.xls'
            # 将文件目录拼接成绝对路径
            self.file_path = os.path.join(rootPath,self.file_path)
            print(self.file_path)
 
        if sheet_id:
            self.sheet_id = sheet_id
        else:
            self.sheet_id = 0
        # 调用成员方法
        self.sheet_table = self.get_sheet()
 
    """成员方法"""
    # 获取sheet页操作对象
    def get_sheet(self):
        data = xlrd.open_workbook(self.file_path)
        sheet_table = data.sheets()[self.sheet_id]
        return sheet_table
 
    # 获取sheet页的行数和列数,返回的是一个元组
    def get_sheet_nrows_ncols(self):
        return self.sheet_table.nrows,self.sheet_table.ncols
 
    # 获取sheet页的行数
    def get_sheet_nrows(self):
        return self.sheet_table.nrows
 
    # 获取sheet页的列数
    def get_sheet_ncols(self):
        return self.sheet_table.ncols
 
    # 获取具体单元格的数据
    def get_sheet_cell(self,row,col):
        """
        :params row: 单元格行值
        :params col: 单元格列值
        :return: cell_data
        """
        cell_data = self.sheet_table.cell_value(row,col)
        return cell_data
 
    # 写入数据到excel中
    def write_to_excel(self,row,col,values):
        # 打开excel文件读取数据句柄
        data = xlrd.open_workbook(self.file_path)
        # 复制excel
        copy_data = copy(data)
        # 读取复制的excel的sheet页
        copy_data_to_sheet = copy_data.get_sheet(0)
        # 通过get_sheet()获取的sheet有write()方法,写入数据
        copy_data_to_sheet.write(row,col,values)
        # 保存数据
        copy_data.save(self.file_path)
 
if __name__ == '__main__':
    read_xls = Operate_Excel()
    print("获取excel表的行数与列数,返回元组格式: ",read_xls.get_sheet_nrows_ncols())
    print("获取excel表的行数: ",read_xls.get_sheet_nrows())
    print("获取excel表的列数: ",read_xls.get_sheet_ncols())
    print("获取excel表的单元格(1,1)的值: ",read_xls.get_sheet_cell(1,1))
    print("获取excel表的单元格(1,2)的值: ",read_xls.get_sheet_cell(1,2))
    print("获取excel表的单元格(2,2)的值: ",read_xls.get_sheet_cell(2,2))
    print("写入excel表的单元格(2,2)的值: ",read_xls.write_to_excel(8,8,'test'))

运行结果:

获取excel表的行数与列数,返回元组格式:  (18, 9)
获取excel表的行数:  18
获取excel表的列数:  9
获取excel表的单元格(1,1)的值:  case_name01
获取excel表的单元格(1,2)的值:  case_method01
获取excel表的单元格(2,2)的值:  case_method02
写入excel表的单元格(2,2)的值:  None

excel文件:

到此,Excel读写功能已经介绍完,以上功能大部分满足日常使用,当然你也可以深入研究,继续加入其它方法进来。

总结:

感谢每一个认真阅读我文章的人!!!

作为一位过来人也是希望大家少走一些弯路,如果你不想再体验一次学习时找不到资料,没人解答问题,坚持几天便放弃的感受的话,在这里我给大家分享一些自动化测试的学习资源,希望能给你前进的路上带来帮助。

?

软件测试面试文档

我们学习必然是为了找到高薪的工作,下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料,并且有字节大佬给出了权威的解答,刷完这一套面试资料相信大家都能找到满意的工作。

视频文档获取方式:
这份文档和视频资料,对于想从事【软件测试】的朋友来说应该是最全面最完整的备战仓库,这个仓库也陪伴我走过了最艰难的路程,希望也能帮助到你!以上均可以分享,点下方小卡片即可自行领取。???????

文章来源:https://blog.csdn.net/2201_76100073/article/details/135063556
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。