unittest（20）- 自动更新表格中数据

Posted 2021-03-14 come202011

 

# 1. get_data.py

from tools import project_path
import pandas as pd


class GetData:
    Cookie = None
    AddTel = pd.read_excel(project_path.test_case_path, sheet_name="phone").iloc[0, 0]


# df = pd.read_excel(project_path.test_case_path, sheet_name="phone")
# print(df.iloc[0, 0])  # 获取phone表单中的A1单元格的值

 

# 2. do_excel.py

# 从表格中获取手机号，进行值的传递
# 每次测试的手机号都不同，如果重复注册，可以使用手工测试，不必再单独写代码。
# 或者直接在表格用例中直接写死一个手机号，表格中只需要保留一个
from openpyxl import load_workbook
from tools.read_config import ReadConfig
from tools import project_path
from account_auto_incre_6.get_data import GetData


class DoExcel:
    @classmethod
    def get_data(cls, file):
        wb = load_workbook(file)
        # 读出来的配置项是字符串格式，需要转换成字典格式
        mode = eval(ReadConfig.get_config(project_path.case_config_path_auto_incre, "MODE", "mode"))

        tel = getattr(GetData, "AddTel")  # 从GetData中拿到数据
        # tel = cls.get_tel(project_path.test_case_path, "phone")  # 通过函数拿到数据
        test_data = []
        for key in mode:
            sheet = wb[key]  # 表单名
            if mode[key] == "all":
                for i in range(2, sheet.max_row+1):
                    row_data = {}
                    # 数值类型读出来还是数值，不用使用函数eval转换
                    row_data["case_id"] = sheet.cell(i, 1).value
                    row_data["url"] = sheet.cell(i, 2).value
                    row_data["method"] = sheet.cell(i, 3).value
                    # row_data["data"] = eval(sheet.cell(i, 4).value)
                    if sheet.cell(i, 4).value.find("${tel}") != -1:   # 表示在这个单元格的值中如果找到了${tel}字符串
                        row_data["data"] = sheet.cell(i, 4).value.replace("${tel}", str(tel))
                        tel = tel + 1
                    else:  # 如果没有找到匹配的字符串
                        row_data["data"] = sheet.cell(i, 4).value
                    row_data["title"] = sheet.cell(i, 5).value
                    row_data["expected"] = sheet.cell(i, 6).value
                    row_data["sheet_name"] = key  # 为了写回测试结果时定位表单
                    test_data.append(row_data)
                    cls.update_tel(file, "phone", tel)  # 写入新数据
            else:
                for case_id in mode[key]:
                    row_data = {}
                    row_data["case_id"] = sheet.cell(case_id+1, 1).value  # 注意行号和case_id的关系
                    row_data["url"] = sheet.cell(case_id+1, 2).value
                    row_data["method"] = sheet.cell(case_id+1, 3).value
                    # row_data["data"] = eval(sheet.cell(case_id+1, 4).value)
                    if sheet.cell(case_id+1, 4).value.find("${tel}") != -1:   # 表示在这个单元格的值中如果找到了${tel}字符串
                        row_data["data"] = sheet.cell(case_id+1, 4).value.replace("${tel}", str(tel))
                        tel = tel + 1
                    else:  # 如果没有找到匹配的字符串
                        row_data["data"] = sheet.cell(case_id+1, 4).value
                    row_data["title"] = sheet.cell(case_id+1, 5).value
                    row_data["expected"] = sheet.cell(case_id+1, 6).value
                    row_data["sheet_name"] = key  # 为了写回测试结果时定位表单
                    test_data.append(row_data)
                    
            # 也可以不在上述的if else分支中都写一次row_data,而是取完数据后统一替换
            # for item in test_data:
            #     if item[‘data‘][‘loginName‘].find("${tel}") != -1:   # 表示在这个单元格的值中如果找到了${tel}字符串
            #             row_data["data"][‘loginName‘] = row_data["data"][‘loginName‘].replace("${tel}", str(tel))
            #             tel = tel + 1

        return test_data

    def write_back(self, file, sheet, i, response, test_result):
        wb = load_workbook(file)
        sheet = wb[sheet]
        # 写入测试结果, value必须是数据或者字符串类型才能被写回，写入数据前必须关闭工作簿
        sheet.cell(i, 7).value = response
        sheet.cell(i, 8).value = test_result
        # 保存工作簿
        wb.save(file)

    @classmethod
    def update_tel(cls, file, sheet_name, tel):
        wb = load_workbook(file)
        sheet = wb[sheet_name]
        sheet.cell(2, 1).value = tel
        wb.save(file)

    # 不使用函数，写一个获取手机号的函数
    # @classmethod
    # def get_tel(cls, file, sheet_name):
    #     wb = load_workbook(file)
    #     sheet = wb[sheet_name]
    #     tel = sheet.cell(2,  1).value
    #     return tel


if __name__ == ‘__main__‘:
    test_data = DoExcel().get_data(project_path.test_case_path)
    print(test_data)
    print(len(test_data))  # 获取列表元素个数

 

表格数据如图：