接口自动化数据存储与读取（2）

使用内置函数读取

import re


class RecordFirstLinePlace:
    CordPlace = {
   } # 记录每一行title在列的位置，使用索引取值时会使用其中记录的数据


class TableTreating:
    def __init__(self, filename):
        self.filename = filename
        self.filepath = filename
        self.wb = load_workbook(self.filepath)
        self.test_data = []

    def read_excel(self, sheet_name):
        work_sheet_data = self.wb[sheet_name].rows
        work_sheet_title = [title_value.value for title_value in next(work_sheet_data)]
        title_place = [i for i in range(1, len(work_sheet_title) + 1)]
        RecordFirstLinePlace.CordPlace[sheet_name] = dict(list(zip(work_sheet_title, title_place)))
        return work_sheet_data, work_sheet_title

    def get_all_data(self, config_var):
        sheet_names = self.wb.sheetnames + ['others']
        model = config_var.get('mode').get(self.filename)
        if model.get('others') == 'skip':
            sheet_name_dict = model
        else:
            sheet_name_dict = {
   sheet_name: 'all' for sheet_name