Python对Excel进行操作的方法封装

Python对Excel进行操作的方法封装

class Excel:
    def __init__(self,filename,sheetname):
        '''

        :type filename: str -- excel的文件目录
        :param sheetname: str -- 选择操作的excel页名
        '''
        self.filename = filename
        try:
            self.wb = openpyxl.load_workbook(self.filename)     #生成文件对象,表示要操作的是哪个文件
        except:
            self.wb=openpyxl.Workbook()
            self.wb.save(filename)
        try:
            self.sheet = self.wb[sheetname]                #生成页对象,表示要操作的是哪个页
        except:
            self.wb.create_sheet(title=sheetname)
            self.sheet=self.wb[sheetname]
            self.wb.save(filename)

    #获取excel文件中的总行数
    def MaxRow(self):
        maxrow = self.sheet.max_row
        return maxrow

    #获取excel文件中最大的列数
    def MaxCol(self):
        maxcol=self.sheet.max_column
        return maxcol

    #获取某个单元格内容
    def read(self,row,col):
        info=self.sheet.cell(row=row, column=col).value
        try:
            info=str(info)
        except:
            pass
        return info

    #获取某一行数据
    def readline(self,row):
        maxcol=self.MaxCol()
        info=[]
        for i in range(1,maxcol+1):
            content=self.sheet.cell(row=row,column=i).value
            try:
                content=str(content)
            except:
                pass
            info.append(content)
        return info

    #获取所有数据
    def readall(self):
        maxrow=self.MaxRow()
        info=[]
        for i in range(1,maxrow+1):
            content=self.readline(i)
            info.append(content)
        return info

    def readcol(self,column):
        maxrow = self.MaxRow()
        info=[]
        for i in range(1, maxrow + 1):
            content=self.sheet.cell(row=i, column=column).value
            try:
                content = str(content)
            except:
                pass
            info.append(content)
        return info

    # 在指定单元格写数据
    def wt(self, rownum, colnum, data):
        self.sheet.cell(row=rownum, column=colnum).value = data
        self.wb.save(self.filename)


    #在指定行写一行内容,data为列表
    def wtline(self,row,data):
        num=len(data)
        for i in range(0,num):
            self.sheet.cell(row=row, column=i+1).value=data[i]
        self.wb.save(self.filename)

(转载注明出处)

你可能感兴趣的:(python,excel,封装)