【第三章】优化excel_utils 类，封装testcase_data_utils类 读取测试数据

优化excel_utils 类

代码地址:https://gitee.com/todayisgoodday/PythonRequest

• 优化excel_utils

　　上一章节我们说到了读取excel文件数据,并且处理合并单元格的数据,这节课我们优化一下excel_utils 将它调整为我们的指定模板格式

　　期望的数据格式 :  

 [{'做菜步骤': '蛋炒饭', '步骤序号': 'step_01', '操作': '油锅烧热', '是否完成': 100.0}, {'做菜步骤': '蛋炒饭', '步骤序号': 'step_02', '操作': '放油', '是否完成': 100.0}, {'做菜步骤': '蛋炒饭', '步骤序号': 'step_03', '操作': '放鸡蛋', '是否完成': 100.0}, {'做菜步骤': '蛋炒饭', '步骤序号': 'step_04', '操作': '放米饭', '是否完成': 80.0}, {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_01', '操作': '油锅烧热', '是否完成': 100.0}, {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_02', '操作': '放油', '是否完成': 100.0}, {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_03', '操作': '放肉', '是否完成': 100.0}, {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_04', '操作': '放辣椒', '是否完成': 80.0}]

　　

• 首先我们导入昨天封装好的 excel_utils

• excel_demo_02.py

# -*- coding: utf-8 -*-
# @Time : 2021/12/9 10:23
# @Author : Limusen
# @File : excel_demo_02


import xlrd3

from common.excel_utils import ExcelUtils

excel_data_list = []
data = ExcelUtils("做饭秘籍.xlsx", "Sheet1")

expect_list = [
    {'做菜步骤': '蛋炒饭', '步骤序号': 'step_01', '操作': '油锅烧热', '是否完成': 100.0},
    {'做菜步骤': '蛋炒饭', '步骤序号': 'step_02', '操作': '放油', '是否完成': 100.0},
    {'做菜步骤': '蛋炒饭', '步骤序号': 'step_03', '操作': '放鸡蛋', '是否完成': 100.0},
    {'做菜步骤': '蛋炒饭', '步骤序号': 'step_04', '操作': '放米饭', '是否完成': 80.0},
    {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_01', '操作': '油锅烧热', '是否完成': 100.0},
    {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_02', '操作': '放油', '是否完成': 100.0},
    {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_03', '操作': '放肉', '是否完成': 100.0},
    {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_04', '操作': '放辣椒', '是否完成': 80.0}]


# # 1.获取行的数据 i为循环的次数
# for i in range(1, data.get_row_count()):
#     print(data.get_merged_value(i, 0))

# # 2.获取列的数据
# for b in range(0, data.get_col_count()):
#     print(data.get_merged_value(0,b))

# 3.组装
# print(data.get_row_count())
# print(data.get_col_count())

# 先循环行
for i in range(1, data.get_row_count()):
    # 先定义一个空字典
    row_dict = {}
    # 再根据行数 循环列数
    for j in range(data.get_col_count()):
        # 重新赋值
        row_dict[data.get_merged_value(0, j)] = data.get_merged_value(i, j)
    excel_data_list.append(row_dict)

print(excel_data_list)

• 做好之后封装到excel_utils类中

• excel_utils.py   新增 get_all_excel_data 方法

# -*- coding: utf-8 -*-
# @Time : 2021/12/8 16:40
# @Author : Limusen
# @File : excel_utils

import os
import xlrd3

# 优化获取文件路径的方式
current_path = os.path.dirname(os.path.abspath(__file__))
# 这里分开写是为了适配mac跟windows之间的// 转义,这样更兼容
excel_file_path = os.path.join(current_path, '..', 'testcase_data', '做饭秘籍.xlsx')
sheet_name = "Sheet1"


class ExcelUtils:

    def __init__(self, excel_file_path=excel_file_path, sheet_name=sheet_name):
        self.excel_file_path = excel_file_path
        self.sheet_name = sheet_name
        self.sheet_obj = self.get_sheet_obj()

    def get_sheet_obj(self):
        """
        创建工作蒲对象
        :return:
        """
        workbook_obj = xlrd3.open_workbook(excel_file_path)
        sheet_obj = self.workbook_obj.sheet_by_name(sheet_name)
        return self.sheet_obj

    def get_row_count(self):
        """
        获取总行数
        :return:
        """
        return self.sheet_obj.nrows

    def get_col_count(self):
        """
        获取总列数
        :return:
        """
        return self.sheet_obj.ncols

    def get_merged_value(self, row_index, col_index):
        """
        查询指定单元格信息
        :param row_index: 想要查询的行坐标
        :param col_index: 想要查询的类坐标
        :return:
        """
        for (min_row, max_row, min_col, max_col) in self.sheet_obj.merged_cells:

            if row_index >= min_row and row_index < max_row:
                if col_index >= min_col and col_index < max_col:
                    # 是合并单元格,单元格的值等于 合并单元格中的第一个单元格的值
                    cell_value = self.sheet_obj.cell_value(min_row, min_col)
                    break
                else:
                    cell_value = self.sheet_obj.cell_value(row_index, col_index)
            else:
                cell_value = self.sheet_obj.cell_value(row_index, col_index)
        return cell_value

    def get_all_excel_data(self):
        """
        获取全部excel表格数据 用列表组装
        :return: excel数据
        """
        excel_data_list = []
        for i in range(1, self.get_row_count()):
            # 先定义一个空字典
            row_dict = {}
            # 再根据行数 循环列数
            for j in range(self.get_col_count()):
                # 重新赋值
                row_dict[self.get_merged_value(0, j)] = self.get_merged_value(i, j)
            excel_data_list.append(row_dict)
        return excel_data_list

if __name__ == '__main__':
    # 每次做完都需要检查一下自己代码是否正确噢!
    ex = ExcelUtils()
    print(ex.get_merged_value(7, 0))

    print(ex.get_all_excel_data())

到此优化完成excel_utils 读取到我们想要的数据格式 

封装testcase_data_utils类

 　　excel文件长这样

• 上面说到我们可以获取到excel的全部数据,现在讲它组装成我们测试数据需要的格式
• 格式如下:　

• 

　这里说到一个新的知识  字典的setdefault()

　　Python 字典 setdefault() 函数和 get()方法 类似, 如果键不存在于字典中，将会添加键并将值设为默认值。

　　如果字典中包含有给定键，则返回该键对应的值，否则返回为该键设置的值。

• 学习 setdefault() 方法

testcase_demo_03.py

# -*- coding: utf-8 -*-
# @Time : 2021/12/9 11:10
# @Author : Limusen
# @File : testcase_demo_03


# 字典设置默认值

dict_01 = {"name": "王二狗", "age": 180}
print(dict_01)

# 使用setdefault可以新增数据到字典
dict_01.setdefault("sex", "男")
print(dict_01)

# 尝试一下如果有值的情况，再试一下setdefault
dict_01.setdefault("age", 29)
# 这个时候发现 setdefault不会改变已有的值
print(dict_01)

testdata_list = [
    {'做菜步骤': '蛋炒饭', '步骤序号': 'step_01', '操作': '油锅烧热', '是否完成': 100.0},
    {'做菜步骤': '蛋炒饭', '步骤序号': 'step_02', '操作': '放油', '是否完成': 100.0},
    {'做菜步骤': '蛋炒饭', '步骤序号': 'step_03', '操作': '放鸡蛋', '是否完成': 100.0},
    {'做菜步骤': '蛋炒饭', '步骤序号': 'step_04', '操作': '放米饭', '是否完成': 80.0},
    {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_01', '操作': '油锅烧热', '是否完成': 100.0},
    {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_02', '操作': '放油', '是否完成': 100.0},
    {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_03', '操作': '放肉', '是否完成': 100.0},
    {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_04', '操作': '放辣椒', '是否完成': 80.0}
]
# 改造
tmp_dict = {}
for i in testdata_list:
    tmp_dict.setdefault(i['做菜步骤'], []).append(i)

print(tmp_dict)

• 　继续优化 

• testcase_demo_04.py

# -*- coding: utf-8 -*-
# @Time : 2021/12/9 13:36
# @Author : Limusen
# @File : testcase_demo_04


testcase_list = {'蛋炒饭': [{'做菜步骤': '蛋炒饭', '步骤序号': 'step_01', '操作': '油锅烧热', '是否完成': 100.0},
                         {'做菜步骤': '蛋炒饭', '步骤序号': 'step_02', '操作': '放油', '是否完成': 100.0},
                         {'做菜步骤': '蛋炒饭', '步骤序号': 'step_03', '操作': '放鸡蛋', '是否完成': 100.0},
                         {'做菜步骤': '蛋炒饭', '步骤序号': 'step_04', '操作': '放米饭', '是否完成': 80.0}],
                 '辣椒炒肉': [{'做菜步骤': '辣椒炒肉', '步骤序号': 'step_01', '操作': '油锅烧热', '是否完成': 100.0},
                          {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_02', '操作': '放油', '是否完成': 100.0},
                          {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_03', '操作': '放肉', '是否完成': 100.0},
                          {'做菜步骤': '辣椒炒肉', '步骤序号': 'step_04', '操作': '放辣椒', '是否完成': 80.0}]}

data_list = []

for key, value in testcase_list.items():
    # 定义一个空字典
    dict_value = {}
    dict_value["菜名"] = key
    dict_value["做菜步骤"] = value
    data_list.append(dict_value)

print(data_list)

• 接下来封装testcase_data_utils.py 完成我们的封装读取测试数据类

　　将测试数据转换成字典格式

testcase_data_utils.py

# -*- coding: utf-8 -*-
# @Time : 2021/12/9 13:42
# @Author : Limusen
# @File : testcase_data_utils

import os
from common.excel_utils import ExcelUtils

current = os.path.dirname(os.path.abspath(__file__))
excel_file_path = os.path.join(current, '..', 'testcase_data', 'testcase_infos.xlsx')
excel_sheet_name = "Sheet1"


class TestCaseDataUtils:

    def __init__(self):
        self.ex = ExcelUtils(excel_file_path=excel_file_path,
                             sheet_name=excel_sheet_name)

    def test(self):
        print(self.ex.get_all_excel_data())

    def convert_data_info_dict(self):
        """
        将测试数据转换成字典
        :param testcase_info:
        :return:
        """
        test_case_dict = {}
        for testcase in self.ex.get_all_excel_data():
            test_case_dict.setdefault(testcase['测试用例编号'], []).append(testcase)
        return test_case_dict


if __name__ == '__main__':
    print(TestCaseDataUtils().convert_data_info_dict())

　　

• 继续优化,封装转换成列表包夹字典

testcase_data_utils.py

# -*- coding: utf-8 -*-
# @Time : 2021/12/9 13:42
# @Author : Limusen
# @File : testcase_data_utils

import os
from common.excel_utils import ExcelUtils

current = os.path.dirname(os.path.abspath(__file__))
excel_file_path = os.path.join(current, '..', 'testcase_data', 'testcase_infos.xlsx')
excel_sheet_name = "Sheet1"


class TestCaseDataUtils:

    def __init__(self):
        self.ex = ExcelUtils(excel_file_path=excel_file_path,
                             sheet_name=excel_sheet_name)

    def test(self):
        print(self.ex.get_all_excel_data())

    def convert_data_info_dict(self):
        """
        将测试数据转换成字典
        :param testcase_info:
        :return:
        """
        test_case_dict = {}
        for testcase in self.ex.get_all_excel_data():
            test_case_dict.setdefault(testcase['测试用例编号'], []).append(testcase)
        return test_case_dict

    def convert_data_info_list(self):
        test_case_list = []
        for key, value in self.convert_data_info_dict().items():
            type_dict = {}
            type_dict["case_id"] = key
            type_dict["case_step"] = value
            test_case_list.append(type_dict)
        return test_case_list

if __name__ == '__main__':
    print(TestCaseDataUtils().convert_data_info_list())

　　到此我们的excel封装就结束咯 ~

•  现在的项目框架图

　　

　　读取excel跟testcase都已经好了 下一章节要对request请求进行封装

欢迎转载 请注明出处: https://www.cnblogs.com/yushengaqingzhijiao/p/15665812.html