python封装成类_python中操作excel数据 封装成一个类

本文用python中openpyxl库，封装成excel数据的读写方法

from openpyxl import load_workbook

from openpyxl.worksheet.worksheet import Worksheet

class ExcelHandler():

def __init__(self,file):

self.file=file

def open_sheet(self,name) ->Worksheet :

#表示此函数的返回值，是一个这样的类型，函数注解

wb=load_workbook(self.file)#通过属性传递

sheet=wb[name]

return sheet

def header(self,sheet_name):

'''获取表头'''

sheet=self.open_sheet(sheet_name)

headers=[]

for i in sheet[1]:

headers.append(i.value)

return headers

def read(self,sheet_name):

'''读取所有数据'''

sheet=self.open_sheet(sheet_name)

rows=list(sheet.rows)#得到所有的数据，包括表头

data=[]#存取所有行的数据

for row in rows[1:]:

row_data=[]#存取一行的数据

for cell in row:#取出单元格

row_data.append(cell.value)#把单元格的值存在一行的数据中

#列表转字典，要和header zip

data_dict=dict(zip(self.header(sheet_name),row_data))

data.append(data_dict)#把一行的值存在放所有行的数据中

print(data)

return data

@staticmethod

def write(file,sheet_name,row,column,data):

wb=load_workbook(file)

sheet = wb[sheet_name]#获取表单

cell = sheet.cell(row,column)#获取单个单元格

cell.value = data#写入值

wb.save(file)#保存

wb.close()

if __name__ == '__main__':

excel=ExcelHandler(r'D:\cases.xlsx')

sheet=excel.read('Sheet1')