apiAutoTest/tools/read_data.py

#!/usr/bin/env/python3
# -*- coding:utf-8 -*-
"""
@project: apiAutoTest
@author: zy7y
@file: read_data.py
@ide: PyCharm
@time: 2020/7/31
"""
import xlrd
from loguru import logger


class ReadData(object):
    def __init__(self, excel_path):
        self.excel_file = excel_path
        self.book = xlrd.open_workbook(self.excel_file)

    def get_data(self):
        """

        :return: data_list - pytest参数化可用的数据， title_list pytest参数化 ids关键字用到的标题数据
        """
        data_list = []
        title_list = []

        table = self.book.sheet_by_index(0)
        for norw in range(1, table.nrows):
            # 每行第4列 是否运行
            if table.cell_value(norw, 3) == '否':
                continue
            # 每行第3列， 标题单独拿出来
            title_list.append(table.cell_value(norw, 1))

            # 返回该行的所有单元格组成的数据 table.row_values(0) 0代表第1列
            case_number = table.cell_value(norw, 0)
            path = table.cell_value(norw, 2)
            is_token = table.cell_value(norw, 4)
            method = table.cell_value(norw, 5)
            # 入参关键字
            parametric_key = table.cell_value(norw, 6)
            file_var = table.cell_value(norw, 7)
            file_path = table.cell_value(norw, 8)
            # 路径参数
            parameters = table.cell_value(norw, 9)
            dependent = table.cell_value(norw, 10)
            data = table.cell_value(norw, 11)
            expect = table.cell_value(norw, 12)
            actual = table.cell_value(norw, 13)
            value = [case_number, path, is_token, method, parametric_key, file_var, file_path, parameters, dependent, data, expect, actual]
            logger.info(value)
            # 配合将每一行转换成元组存储，迎合 pytest的参数化操作，如不需要可以注释掉 value = tuple(value)
            value = tuple(value)
            data_list.append(value)
        return data_list, title_list