python 之excel文件读取封装

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)