python 第八讲 操作excle

代码如下：

import openpyxl


class OperateExcle:
    # 传入文件名称和sheet名称
    def __init__(self, file_path, sheet_name):
        self.file_path = file_path
        self.sheet = sheet_name
        self.workbook = openpyxl.load_workbook(self.file_path)
        self.sh = self.workbook[self.sheet]
        self.max_row = self.sh.max_row
        self.max_column = self.sh.max_column

    # 读取excle方法
    def read_excle(self):
        case_list = []
        for row in range(1, self.max_row + 1):
            # 获取excle第一行的数据，作为标题存在列表中
            if row == 1:
                title = []
                for column in range(1, self.max_column + 1):
                    value = self.sh.cell(row=row, column=column).value
                    title.append(value)
            else:
                # 遍历获取excle除第一行之外的数据作为内容，存在列表中
                info = []
                for column in range(1, self.max_column + 1):
                    value = self.sh.cell(row=row, column=column).value
                    info.append(value)
                # 打包标题和内容存在字典中作为一条测试用例数据
                case = dict(zip(title, info))
                case_list.append(case)
        return case_list

    # 保存Excel方法
    def save_excle(self, row, column, value):
        self.sh.cell(row, column, value)
        self.workbook.save(self.file_path)