import os

import xlrd

PATH = lambda p: os.path.abspath(

    os.path.join(os.path.dirname(__file__), p)

)

class ExcelData:

    def __init__(self, file, sheet="sheet1", title=True):

        # 判断文件存在不存在

        if os.path.isfile(PATH(file)):

            self.file = PATH(file)

            self.sheet = sheet

            self.title = title

            self.data = list()

            self.workbook = xlrd.open_workbook(self.file)

        else:

            raise FileNotFoundError("文件不存在")

    @property

    def get_data(self):

        """获取表格数据"""

        if not self.data:

            # 判断表单名称

            if type(self.sheet) not in [int, str]:

                raise Exception("表单名称类型错误")

            else:

                if type(self.sheet) == int:

                    book = self.workbook.sheet_by_index(self.sheet)

                else:

                    book = self.workbook.sheet_by_name(self.sheet)

            # 判断表格是否有表头，有则输出列表嵌套字典形式数据，否则输入列表嵌套列表形式数据

            if self.title:

                title = book.row_values(0)

                for i in range(1, book.nrows):

                    self.data.append(dict(zip(title, book.row_values(i))))   # 可参考字典章节

            else:

                for i in range(book.nrows):

                    self.data.append(book.row_values(i))

        return self.data

    @property

    def get_sheets(self):

        """获取所有表单，这个在后续会用到"""

        book = self.workbook.sheets()

        return book

infos = ExcelData("htmls/测试用例.xlsx", "登入页面", True).get_data

print(infos)

sheets = ExcelData("htmls/测试用例.xlsx").get_sheets

print(sheets)