163 lines
4.5 KiB
Python
163 lines
4.5 KiB
Python
# -*- coding: utf-8 -*-
|
||
# @Time : 2019/11/13 9:31
|
||
# @Author : kira
|
||
# @Email : 262667641@qq.com
|
||
# @File : do_excel.py
|
||
# @Project : risk_project
|
||
import sys
|
||
|
||
sys.path.append("../../")
|
||
sys.path.append("../../common")
|
||
from openpyxl import load_workbook
|
||
from common.utils.singleton import singleton
|
||
from common.file_handling import logger
|
||
|
||
|
||
@singleton
|
||
class DoExcel:
|
||
|
||
def __init__(self, file_name):
|
||
self.file_name = file_name
|
||
self.wb = load_workbook(self.file_name)
|
||
self.init_sheet = self.wb["init"]
|
||
|
||
# def __enter__(self):
|
||
# self.wb = load_workbook(self.file_name)
|
||
# self.init_sheet = self.wb['init']
|
||
# return self
|
||
#
|
||
# def __exit__(self, exc_type, exc_val, exc_tb):
|
||
# self.wb.save(self.file_name)
|
||
# self.wb.close()
|
||
|
||
@logger.log_decorator()
|
||
def do_excel(self):
|
||
"""
|
||
通过 title 定位单元格，获取所有测试数据
|
||
Returns: 读取每一条测试用用例分别保存到字典中，然后再将所有用例保存到列表中，如[{用例1},{用例2},{用例3}]
|
||
[{"":""},{},{}]
|
||
"""
|
||
sheets = eval(self.get_excel_init().get("sheets"))
|
||
test_data = []
|
||
for sheet_name in sheets: # 遍历存在配置文件里面的字典，sheet_name == 每一个 excel 中的 sheetName
|
||
sheet = self.wb[sheet_name] # 获取所有 sheet 句柄
|
||
max_row = sheet.max_row # 获取最大行
|
||
max_column = sheet.max_column # 获取最大列
|
||
fist_header = [] # 获取第一行标题所有值
|
||
for i in range(1, max_column + 1):
|
||
fist_header.append(sheet.cell(1, i).value)
|
||
# 定位单元格
|
||
for i in range(2, max_row + 1):
|
||
sub_data = {} # 列表内的字典（也就是测试数据）
|
||
for k in range(1, max_column + 1):
|
||
sub_data[fist_header[k - 1]] = sheet.cell(i, k).value
|
||
sub_data["sheet"] = sheet_name
|
||
test_data.append(sub_data) # 将所有单元格 title 对应的值组成字典添加到列表中。
|
||
return test_data
|
||
|
||
# @logger.log_decorator()
|
||
def do_excel_yield(self):
|
||
"""
|
||
读取excel数据的生成器
|
||
Returns:
|
||
|
||
"""
|
||
sheets = eval(self.get_excel_init().get("sheets"))
|
||
for sheet_name in sheets:
|
||
sheet = self.wb[sheet_name]
|
||
max_row = sheet.max_row
|
||
max_column = sheet.max_column
|
||
first_header = []
|
||
for i in range(1, max_column + 1):
|
||
first_header.append(sheet.cell(1, i).value)
|
||
for i in range(2, max_row + 1):
|
||
sub_data = {}
|
||
for k in range(1, max_column + 1):
|
||
sub_data[first_header[k - 1]] = sheet.cell(i, k).value
|
||
sub_data["sheet"] = sheet_name
|
||
yield sub_data
|
||
|
||
@logger.log_decorator()
|
||
def write_back(self, sheet_name, i, **kwargs):
|
||
"""
|
||
|
||
Args:
|
||
sheet_name:sheet 名称
|
||
i:用例 Id
|
||
response_value: 响应结果
|
||
test_result: 测试结果
|
||
assert_log: 报错结果
|
||
Returns:
|
||
"""
|
||
response_value = kwargs.get("response")
|
||
test_result = kwargs.get("test_result")
|
||
assert_log = kwargs.get("assert_log")
|
||
sheet = self.wb[sheet_name]
|
||
sheet.cell(i + 1, 22).value = response_value
|
||
sheet.cell(i + 1, 23).value = test_result
|
||
sheet.cell(i + 1, 24).value = assert_log
|
||
self.wb.save(self.file_name)
|
||
|
||
@logger.log_decorator()
|
||
def clear_date(self):
|
||
"""
|
||
执行清空单元格数据
|
||
Returns:
|
||
|
||
"""
|
||
sheets = eval(self.get_excel_init().get("sheets"))
|
||
|
||
for sheet_name in sheets:
|
||
sheet = self.wb[sheet_name]
|
||
max_row = sheet.max_row # 获取最大行
|
||
for i in range(2, max_row + 1):
|
||
sheet.cell(i, 22).value = ""
|
||
sheet.cell(i, 23).value = ""
|
||
sheet.cell(i, 24).value = ""
|
||
self.wb.save(self.file_name)
|
||
return f"清空指定 {sheets} 中的单元格成功"
|
||
|
||
@logger.log_decorator()
|
||
def get_excel_init(self):
|
||
"""
|
||
获取 excel 中 sheet 名称为 init 中的基础数据
|
||
Returns:init 表中的所有数据
|
||
"""
|
||
max_row = self.init_sheet.max_row # 获取最大行
|
||
max_column = self.init_sheet.max_column # 获取最大列
|
||
first_head = [] # 存储标题的 list
|
||
for i in range(max_column):
|
||
first_head.append(self.init_sheet.cell(1, i + 1).value)
|
||
init = {}
|
||
for k in range(2, max_row + 1):
|
||
for i, v in enumerate(first_head):
|
||
init[v] = self.init_sheet.cell(k, i + 1).value
|
||
if init.get("run").upper() == "YES":
|
||
break
|
||
return init
|
||
|
||
def get_excel_init_and_cases(self):
|
||
"""
|
||
|
||
Returns:初始化数据及测试用例组成的元组
|
||
|
||
"""
|
||
try:
|
||
self.clear_date()
|
||
test_case = self.do_excel_yield()
|
||
init_data = self.get_excel_init()
|
||
except Exception as e:
|
||
raise e
|
||
return init_data, test_case
|
||
|
||
|
||
if __name__ == '__main__':
|
||
from common.config import Config
|
||
|
||
file_n = Config.test_case
|
||
excel = DoExcel(file_n)
|
||
# excel.get_excel_init()
|
||
# excel.do_excel()
|
||
# excel.clear_date()
|
||
# excel.do_excel()
|