python excel文件写入

# coding=utf8
import openpyxl
import sys
import os

base_path = os.getcwd()
sys.path.append(base_path)


class HandExcel:
    def __init__(self, excel_path=None):
        dir_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
        file_name = os.path.join(dir_path, "config", "data.xlsx")
        if excel_path is None:
            self.excel_path = file_name
        else:
            self.excel_path = excel_path
        self.open_excel = openpyxl.load_workbook(self.excel_path)
    def load_excel(self):
        '''
        加载excel
        '''
        dir_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
        # open_excel = openpyxl.load_workbook(dir_path + r"\config\data.xlsx")
        open_excel = openpyxl.load_workbook(self.excel_path)
        return open_excel

    def get_sheet_data(self, index=None):
        '''
        加载所有sheet的内容
        '''
        sheet_name = self.open_excel.sheetnames
        if index == None:
            index = 0
        data = self.load_excel()[sheet_name[index]]
        return data

    def get_cell_value(self, row, cols):
        '''
        获取某一个单元格内容
        '''
        data = self.get_sheet_data().cell(row=row, column=cols).value
        return data

    def get_rows(self):
        '''
        获取行数
        '''
        row = self.get_sheet_data().max_row
        return row

    def get_rows_value(self, row):
        '''
        获取某一行的内容
        '''
        row_list = []
        for i in self.get_sheet_data()[row]:
            row_list.append(i.value)
        return row_list

    def excel_write_data(self, row, cols, value):
        '''
        写入数据
        '''
        wb = self.load_excel()
        wr = wb.active
        wr.cell(row, cols, value)
        wb.save(self.excel_path)

    def get_columns_value(self, key=None):
        '''
        获取某一列得数据
        '''
        columns_list = []
        if key == None:
            key = 'A'
        columns_list_data = self.get_sheet_data()[key]
        for i in columns_list_data:
            columns_list.append(i.value)
        return columns_list

    def get_rows_number(self, case_id):
        '''
        获取行号
        '''
        num = 1
        cols_data = self.get_columns_value()
        # print('cols_data', cols_data)
        for col_data in cols_data:
            if case_id == col_data:
                return num
            num = num + 1
        return num

    def get_excel_data(self):
        '''
        获取excel里面所有的数据
        '''
        data_list = []
        for i in range(self.get_rows()):
            data_list.append(self.get_rows_value(i + 2))

        return data_list


# handle = HandExcel()
#
if __name__ == "__main__":
    handle = HandExcel()
    ##print(handle.get_rows_number('imooc_001'))
    data = handle.get_excel_data()
    for row in data:
        rows = row[0]
        if rows is None:
            pass
        else:
            j = handle.get_rows_number(rows)
            # ['test_001', 'I a person', None], ['test_002', '', None]
            if row[1] == 'I am person':
                handle.excel_write_data(j,3, False)
            else:
                handle.excel_write_data(j,3, True)