python 配置文件控制读取excel的数据

# -*- coding: utf-8 -*-
# @Time     :2021/3/24  22:08
# @Author   :wang-kai
# @tel      :15313929271
# @File    :fx_excel.py
# 读取excel所有行信息并且储存在一个大列表中
from openpyxl import load_workbook
from 配置文件 import ReadConfig


class Do_excel:
    def __init__(self, filename, sheet):
        self.filename = filename
        self.sheet = sheet
        self.wb = load_workbook(self.filename)
        self.sheet = self.wb[self.sheet]

    def read_excel(self, button):  # 读取excel

        test_data = []
        if button == 'all':
            for i in range(2, self.sheet.max_row+1):
                row_data = {'CasdId': self.sheet.cell(i, 1).value, 'Module': self.sheet.cell(i, 2).value,
                            'Title': self.sheet.cell(i, 3).value, 'Method': self.sheet.cell(i, 4).value,
                            'Params': self.sheet.cell(i, 5).value, 'sql': self.sheet.cell(i, 6).value,
                            'ExpectedResule': self.sheet.cell(i, 7).value}
                test_data.append(row_data)
            self.wb.close()
        else:
            for i in button:
                row_data = {'CasdId': self.sheet.cell(i, 1).value, 'Module': self.sheet.cell(i, 2).value,
                            'Title': self.sheet.cell(i, 3).value, 'Method': self.sheet.cell(i, 4).value,
                            'Params': self.sheet.cell(i, 5).value, 'sql': self.sheet.cell(i, 6).value,
                            'ExpectedResule': self.sheet.cell(i, 7).value}
                test_data.append(row_data)
            self.wb.close()
        return test_data

    def write_excel(self, row, column, value):  # 将测试结果写回
        wb = load_workbook(self.filename)
        sheet = wb[self.sheet]
        sheet.cell(row, column).value = value
        wb.save(self.filename)
        wb.close()


if __name__ == '__main__':
    button = ReadConfig('peizhi.conf').get_data('test_02', 'test')
    t = Do_excel('test_api.xlsx', 'login')
    res = t.read_excel(button)
    print(res)