import pandas as pd
import os


class ExcelUnit:

    def read_path(self):
        # dirname获取文件当前目录路径 字符串分割获取到项目的路径 abspath设置绝对路径
        return os.path.abspath((os.path.dirname(__file__)).split("common")[0])

    def read_excel(self, sheet_name):
        # 读取测试数据文件 通过传参的方式设置每次需要读取的工作簿
        df = pd.read_excel(self.read_path() + r"\testData\test.xlsx", sheet_name=sheet_name)
        # 获取总行总列数，通过索引的方式获取总行
        sum_row = df.shape[0]
        # 创建一个存储每行数据的列表
        sava_list = []
        # 遍历每一行数据
        for lst in range(sum_row):
            # 创建一个临时存储数据的列表
            temp_list = []
            # 遍历每列的数据
            for vls in df.loc[lst].values:
                # 将遍历的数据添加到列表中
                temp_list.append(vls)
            # 将每次行的数据加入到列表中
            sava_list.append(temp_list)
        return sava_list