python接口自动化测试框架设计( 第二章)-封装Excel方法

如果你不太明白这篇文章是做什么的，点击下方进入介绍篇：

python接口自动化测试框架结构 ( 第一章)

点击跳转到第一章，你可以知道框架结构！！

前方高能！！正文开始！！

1、在utlis.py封装Excel读写方法 （文件位置跳转第一章看代码结构）

大家需要知道，读取Excel分两种情况，一种是写入，当然还有读取Excel

class Excel:
    """
    初始化方法 参数type：为r是读取excel，为w是写入excel,参数file_name是w文件路径
    """
    def __init__(self, type, file_name):
        """
        :param type: r:读 w:写
        :param file_name: 读取文件路径
        """
        # 读取excel
        if type == 'r':
            # 打开文件
            self.workbook = xlrd.open_workbook(file_name)
            # 获取到所有的sheet_names获取到所有，得到的是一个list
            self.sheet_names = self.workbook.sheet_names()
            # 装载所有数据的list
            self.list_data = []
        # 写入excel
        elif type == 'w':
            self.filename = file_name
            self.wb = load_workbook(self.filename) #读取存在文件
            self.ws = self.wb.active 

2、创建testcase.xlsx文件（文件位置跳转第一章看代码结构）

文件内容如下：

创建成功后新增read方法：

    def read(self)->list:
        # 根据sheet_name去读取用例，并获取文件的总行数获取到每行的内容
        for sheet_name in self.sheet_names:
            # 通过每个sheetname获取到每个页的内容
            sheet = self.workbook.sheet_by_name(sheet_name)
            # 获取总行数
            max_ros = sheet.nrows
            # 根据总行数进行读取
            for i in range(0, max_ros):
                rowvalues = sheet.row_values(i)
                # 每一行的内容添加进去
                self.list_data.append(rowvalues)
        #返回ex数据，格式为list
        return self.list_data

在utils.py底部加入如下代码：

if __name__ == '__main__':
    file = '../database/testcase.xlsx'
    e = Excel('r', file)
    list_read=e.read()
    for i in list_read:
        print(i)

输出结果为：

3、需进一步优化将ex表内表头修改为英文，并将每一行数据转换成一个个字典新建方法excel_dict：

def excel_dict(data):
    """
    1.将excel头部替换成英文的
    2.处理成dict格式
    """
    header = {
        '用例编号': 'id',
        '请求类型': 'get_type',
        '测试url': 'url',
        '测试接口': 'interface',
        '用例标题': 'title',
        '测试数据': 'data',
        '预期结果': 'expected',
        '请求头': 'header',
        '响应数据状态/json返回数据的code': 'code',
        '状态码': 'status',
        '响应状态': 'msg'
    }
    head = []
    list_dict_data = []
    for d in data[1]:
        # 获取到英文的头部内容如果为中文，则替换成英文 进行改成一个k
        # 传入两个参数的作用是 查到则返回查到的数据查不到则返回传入的原数据
        d = header.get(d, d)
        # 将去除的头部英文装进list中
        head.append(d)
    # 获取到数据进行切片处理，0坐标为标题，1坐标是头部
    for b in data[2:]:
        # 头部和内容拼接为json串
        dict_data = {}
        for i in range(len(head)):
            # 之所以判断类型，如果不进行判断会出现str的错误，strip去除空格也有转str的用法
            if isinstance(b[i], str):
                dict_data[head[i]] = b[i].strip()
            else:
                dict_data[head[i]] = b[i]
        # list里面是字典格式
        list_dict_data.append(dict_data)
    return list_dict_data

在utils.py底部加入如下代码：

if __name__ == '__main__':
    file = '../database/testcase.xlsx'
    e = Excel('r', file)
    list_read = e.read()
    a = excel_dict(list_read)
    print(a)
    print(a[0]['get_type']) # 获取列表内第一个字典的get_type的值

运行结果如下：

4、实现写入功能，新增方法write：

    def write(self, data):
        """
        数据写入ex表内
        :param data: 需要写入的数据：list
        :return:
        """
        self.ws.append(data)
        self.wb.save(self.filename)

在utils.py底部加入如下代码：

if __name__ == '__main__':
    # file = '../database/testcase.xlsx'
    file = '../results/results.xlsx'
    e = Excel('w', file)
    a = [1,2,3,4,5,6,7,8]
    e.write(a)

运行结果如下：

成功写入文件，需注意需写入的内容为list格式。

5、写入格式必须为例表格式、新增方法write_result，将所有要写入的内容处理成列表，代码如下：

返回为列表格式（可能多此一举，其余方法实现便可、这边考虑方便后续管控）

def write_result(value1=None, value2=None, value3=None,
                 value4=None, value5=None, value6=None,
                 value7=None, value8=None) -> list:
    """
    将写入的值返回：为list格式
    :param value1: 值1 - 编号
    :param value2: 值2 - 测试接口名字
    :param value3: 值3 - 测试用例标题
    :param value4: 值4 - 测试url地址
    :param value6: 值5 - 测试数据/json
    :param value6: 值6 - 预计结果
    :param value7: 值7 - 实际结果
    :param value8: 值8 - 执行情况
    :return:
    """
    result_list = []
    result_list.append(value1)
    result_list.append(value2)
    result_list.append(value3)
    result_list.append(value4)
    result_list.append(value5)
    result_list.append(value6)
    result_list.append(value7)
    result_list.append(value8)
    return result_list

在utils.py底部加入如下代码：

if __name__ == '__main__':
    # file = '../database/testcase.xlsx'
    file = '../results/results.xlsx'
    e = Excel('w', file)
    e.write(write_result(1,2,3,4,5,6,7,8))

运行成功后结果如下：

写入数据成功！！

完整版代码如下：

"""公共方法类"""
import logging
import xlrd
from openpyxl import load_workbook


class Excel:
    """
    初始化方法 参数type：为r是读取excel，为w是写入excel获取不同的实例，参数file_name是将要读取的文件
    """

    def __init__(self, type, file_name):
        """
        :param type: r:读 w:写
        :param file_name: 文件路径
        """
        # 读取excel
        if type == 'r':
            # 打开文件
            self.workbook = xlrd.open_workbook(file_name)
            # 获取到所有的sheet_names,sheet1,sheet2获取到所有，获取到的是一个list
            self.sheet_names = self.workbook.sheet_names()
            # 装载所有数据的list
            self.list_data = []
        # 写入excel
        elif type == 'w':
            self.filename = file_name
            self.wb = load_workbook(self.filename)
            self.ws = self.wb.active

    def read(self) -> list:
        # 根据sheet_name去读取用例，并获取文件的总行数获取到每行的内容
        for sheet_name in self.sheet_names:
            # 通过每个sheetname获取到每个页的内容
            sheet = self.workbook.sheet_by_name(sheet_name)
            # 获取总行数
            rosw = sheet.nrows
            # 根据总行数进行读取
            for i in range(0, rosw):
                rowvalues = sheet.row_values(i)
                # 讲每一行的内容添加进去
                self.list_data.append(rowvalues)
            #     去除大标题第一行进行切割处理
        # 将得到的excel数据返回进行处理
        return self.list_data

    def write(self, data):
        """
        数据写入ex表内
        :param data: 需要写入的数据：list
        :return:
        """
        self.ws.append(data)
        self.wb.save(self.filename)
        logging.info('测试数据写入结果: 成功写入')


def excel_dict(data):
    """
    1.将excel头部替换成英文的
    2.处理成json/dict格式
    """
    header = {
        '用例编号': 'id',
        '请求类型': 'get_type',
        '测试url': 'url',
        '测试接口': 'interface',
        '用例标题': 'title',
        '测试数据': 'data',
        '预期结果': 'expected',
        '请求头': 'header',
        '响应数据状态/json返回数据的code': 'code',
        '状态码': 'status',
        '响应状态': 'msg'
    }
    head = []
    list_dict_data = []
    for d in data[1]:
        # 获取到英文的头部内容如果为中文，则替换成英文 进行改成一个k
        # 传入两个参数的作用是 查到则返回查到的数据查不到则返回传入的原数据
        d = header.get(d, d)
        # 将去除的头部英文装进list中
        head.append(d)
    # 获取到数据进行切片处理，0坐标为标题，1坐标是头部
    for b in data[2:]:
        # 头部和内容拼接为json串
        dict_data = {}
        for i in range(len(head)):
            # 之所以判断类型，如果不进行判断会出现str的错误，strip去除空格也有转str的用法
            if isinstance(b[i], str):
                dict_data[head[i]] = b[i].strip()
            else:
                dict_data[head[i]] = b[i]
        # list里面是字典格式
        list_dict_data.append(dict_data)
    return list_dict_data


def write_result(value1=None, value2=None, value3=None,
                 value4=None, value5=None, value6=None,
                 value7=None, value8=None) -> list:
    """
    将写入的值返回：为list格式
    :param value1: 值1 - 编号
    :param value2: 值2 - 测试接口名字
    :param value3: 值3 - 测试用例标题
    :param value4: 值4 - 测试url地址
    :param value6: 值5 - 测试数据/json
    :param value6: 值6 - 预计结果
    :param value7: 值7 - 实际结果
    :param value8: 值8 - 执行情况
    :return:
    """
    result_list = []
    result_list.append(value1)
    result_list.append(value2)
    result_list.append(value3)
    result_list.append(value4)
    result_list.append(value5)
    result_list.append(value6)
    result_list.append(value7)
    result_list.append(value8)
    return result_list


if __name__ == '__main__':
    # file = '../database/testcase.xlsx'
    file = '../results/results.xlsx'
    e = Excel('w', file)
    e.write(write_result(1, 2, 3, 4, 5, 6, 7, 8))

点击不迷路，python接口自动化测试框架结构 ( 第三章)
非常感谢您的阅读，第一次写博客文章，写的不好的地方多度包含，欢迎大佬指出代码中的不足，共同提升、共同进步！（要源代码 点赞关注后+VX dengwoi ）