python3生成Excel报告

Python生成Excel文件需要先安装第三方库xlwt或xlsxwriter

1. xlwt库

第三方库函数介绍：

创建工作簿（workbook）：

workbook = xlwt.Workbook() 

创建工作表（sheet）

sheet = workbook.add_sheet("Sheet Name") 

写单元格（cell）：

sheet.write(0, 0, 'foobar') # row, column, value

对单元格应用样式（加粗为例）：

style = xlwt.easyxf('font: bold 1')

sheet.write(0, 0, 'foobar', style)

设置列宽：

sheet.col(0).width = 256 * (len(key) + 1) 

保存：

workbook.save("foobar.xls")

代码示例：

# -*- coding: utf-8 -*-

import os

import xlwt

class XlsReport:

#    '''Excel报告接口'''

    def __init__(self,file_path):

        self.xls_workbook = None           #创建的Excel对象

        self.xls_file_path = file_path       #创建Excel的文件路径

        self.xls_worksheets = {}              #工作表的行列标记符{worksheet对象:{r:行,c:列}}

        self.xls_sheet_col_length = {}    #工作表的列宽{worksheet对象:{列1:宽度,列2:宽度...}}

        

    def xlsOpenWorkbook(self):

        '''创建一个Excel'''

        self.xls_workbook = xlwt.Workbook()

        

    def xlsAddWorksheet(self,sheet_name='sheet',r=0,c=0):

        '''

        创建一个工作表对象

        @parameter sheet_name:工作簿名称

        @parameter r:工作表的行

        @parameter c:工作表的列

        '''

        obj_worksheet = self.xls_workbook.add_sheet(sheet_name)   #创建工作表对象

        xls_worksheet = {}                                        #工作表的行列标记符

        xls_worksheet['r'] = r                                    #行标记符

        xls_worksheet['c'] = c                                    #列标记符

        self.xls_worksheets[obj_worksheet] = xls_worksheet

        self.xls_sheet_col_length[obj_worksheet] = {}

        return obj_worksheet

    

    def xlsCloseWorkbook(self,obj_worksheet):

        '''

        关闭Excel文件对象,保存Excel数据

        @parameter obj_worksheet:工作表对象

        '''

        #设置工作簿的列宽

        for c in self.xls_sheet_col_length[obj_worksheet]:

            if self.xls_sheet_col_length[obj_worksheet][c] < 10:

                obj_worksheet.col(c).width = 256*10

            elif self.xls_sheet_col_length[obj_worksheet][c] > 50:

                obj_worksheet.col(c).width = 256*50

            else:

                obj_worksheet.col(c).width = 256*(self.xls_sheet_col_length[obj_worksheet][c])

        #关闭工作表对象

        self.xls_workbook.save(self.xls_file_path)

    

    def addWorksheetTitle(self,obj_worksheet,titles=[],r=0,c=0):

        '''

        添加工作簿的标题

        @parameter obj_worksheet:工作表对象

        @parameter titles:标题

        @parameter r:工作表的行

        @parameter c:工作表的列

        '''

        #设置标题的样式

        style_title = xlwt.easyxf('pattern:pattern solid,fore_colour lime; font:height 200,bold on; align:horz center;')

        #写工作簿的标题

        for title in titles:

            obj_worksheet.write(r,c,title,style_title)

            c += 1

        #工作簿的行标记+1

        r += 1

        self.xls_worksheets[obj_worksheet]['r'] = r

    def appendWorkshetData(self,obj_worksheet,datas=[],r=None,c=None,gold=0):   

        '''

        按行追加数据

        @parameter obj_worksheet:工作表对象

        @parameter datas:标题

        @parameter r:工作表的行

        @parameter c:工作表的列

        @parameter gold:0(不变/Pass),-1(变差/Fail),1(变好)

        ''' 

        #标准

        stylebox = xlwt.easyxf('font:height 200; borders:left 1,right 1,top 1,bottom 1; align:horiz left,wrap 1')       

        #红体

        stylebox_red = xlwt.easyxf('font:height 200,color-index red; borders:left 1,right 1,top 1,bottom 1; align:horiz left,wrap 1')

        #蓝体

        stylebox_blue = xlwt.easyxf('font:height 200,color-index blue; borders:left 1,right 1,top 1,bottom 1; align:horiz left,wrap 1') 

        #写工作表的行

        if None == r:

            r = self.xls_worksheets[obj_worksheet]['r']

        else:

            r = r

        #写工作表的列        

        if None == c:

            c = self.xls_worksheets[obj_worksheet]['c']

        else:

            c = c

        #写工作表的数据

        for data in datas:

            if str != type(data):

                data = str(data)

            if 0 == gold:

                obj_worksheet.write(r,c,data,stylebox)

            elif -1 == gold:

                obj_worksheet.write(r,c,data,stylebox_red)

            elif 1 == gold:

                obj_worksheet.write(r,c,data,stylebox_blue)

            else:

                print('Info:gold was illegal.')

                obj_worksheet.write(r,c,data,stylebox)

            #工作表每列的最大字符长度

            if c not in self.xls_sheet_col_length[obj_worksheet]:

                self.xls_sheet_col_length[obj_worksheet][c] = len(data)

            else:

                if self.xls_sheet_col_length[obj_worksheet][c] < len(data):

                    self.xls_sheet_col_length[obj_worksheet][c] = len(data)

            c += 1

        #工作表的行标记+1

        r += 1

        self.xls_worksheets[obj_worksheet]['r'] = r

                        

#self test           

if __name__ == '__main__':

    #文件的后缀名为xls

    file_path = os.path.join(os.getcwd(),'test.xls')

    xls = XlsReport(file_path)

    #创建Excel对象

    xls.xlsOpenWorkbook()

    #添加工作对象

    sheet_names = ['sheet1','sheet2','sheet3']

    for sheet_name in sheet_names:

        sheet = xls.xlsAddWorksheet(sheet_name)

        #Excel的标题

        xls.addWorksheetTitle(sheet,['测试用例编号','测试用例名称','测试结果','备注'])

        #Excel的数据

        xls.appendWorkshetData(sheet,[1001,'test1','Pass',''],gold=0)

        xls.appendWorkshetData(sheet,[1002,'test2','Fail','失败'],gold=-1)

        xls.appendWorkshetData(sheet,[1003,'test3','Pass','调优'],gold=1)

        xls.xlsCloseWorkbook(sheet)

运行结果：



2. xlsxwriter库

代码示例：

# -*- coding: utf-8 -*-

import os

import xlsxwriter

class XlsAPI:

#    '''Excel报告接口'''

    def __init__(self,file_path):

        self.xls_workbook = None          #创建的Excel对象

        self.xls_file_path = file_path    #创建Excel的文件路径

        self.xls_worksheets = {}          #工作簿的行列标记符{worksheet对象:{r:行,c:列}}

        self.xls_sheet_col_length = {}    #工作簿的列宽{worksheet对象:{列1:宽度,列2:宽度...}}

        

    def xlsOpenWorkbook(self):

        '''创建一个Excel'''

        self.xls_workbook = xlsxwriter.Workbook(self.xls_file_path)

        

    def xlsAddWorksheet(self,sheet_name='sheet',r=0,c=0):

        '''

        创建一个工作簿对象

        @parameter sheet_name:工作簿名称

        @parameter r:工作簿的行

        @parameter c:工作簿的列

        '''

        obj_worksheet = self.xls_workbook.add_worksheet(sheet_name)   #创建工作簿对象

        xls_worksheet = {}                                            #工作簿的行列标记符

        xls_worksheet['r'] = r                                        #行标记符

        xls_worksheet['c'] = c                                        #列标记符

        self.xls_worksheets[obj_worksheet] = xls_worksheet

        self.xls_sheet_col_length[obj_worksheet] = {}

        return obj_worksheet

    

    def xlsCloseWorkbook(self):

        '''

        关闭Excel文件对象,保存Excel数据

        '''

        #关闭工作簿对象

        self.xls_workbook.close()

    

    def addWorksheetTitle(self,obj_worksheet,titles=[],r=0,c=0):

        '''

        添加工作簿的标题

        @parameter obj_worksheet:工作簿对象

        @parameter titles:标题

        @parameter r:工作簿的行

        @parameter c:工作簿的列

        '''

        #设置标题的样式

        style_title = self.xls_workbook.add_format({'bold':True})

        style_title.set_pattern(1)    #填充颜色

        style_title.set_bg_color('green')

        #写工作簿的标题

        for title in titles:

            obj_worksheet.write(r,c,title,style_title)

            c += 1

        #工作簿的行标记+1

        r += 1

        self.xls_worksheets[obj_worksheet]['r'] = r

    def appendWorkshetData(self,obj_worksheet,datas=[],r=None,c=None,gold=0):   

        '''

        按行追加数据

        @parameter obj_worksheet:工作簿对象

        @parameter datas:标题

        @parameter r:工作簿的行

        @parameter c:工作簿的列

        @parameter gold:0(不变/Pass),-1(变差/Fail),1(变好)

        ''' 

        #标准

        stylebox = self.xls_workbook.add_format()

        #红体

        stylebox_red = self.xls_workbook.add_format({'font_color':'red'})

        #蓝体

        stylebox_blue = self.xls_workbook.add_format({'font_color':'blue'})

        #写工作簿的行

        if None == r:

            r = self.xls_worksheets[obj_worksheet]['r']

        else:

            r = r

        #写工作簿的列        

        if None == c:

            c = self.xls_worksheets[obj_worksheet]['c']

        else:

            c = c

        #写工作簿的数据

        for data in datas:

            if str != type(data):

                data = str(data)

            if 0 == gold:

                obj_worksheet.write(r,c,data,stylebox)

            elif -1 == gold:

                obj_worksheet.write(r,c,data,stylebox_red)

            elif 1 == gold:

                obj_worksheet.write(r,c,data,stylebox_blue)

            else:

                print('Info:gold was illegal.')

                obj_worksheet.write(r,c,data,stylebox)

            #工作簿每列的最大字符长度

            if c not in self.xls_sheet_col_length[obj_worksheet]:

                self.xls_sheet_col_length[obj_worksheet][c] = len(data)

            else:

                if self.xls_sheet_col_length[obj_worksheet][c] < len(data):

                    self.xls_sheet_col_length[obj_worksheet][c] = len(data)

            c += 1

        #工作簿的行标记+1

        r += 1

        self.xls_worksheets[obj_worksheet]['r'] = r

                        

    def setWorksheetWidth(self,obj_worksheet):

        '''

        设置Excel表宽度

        @parameter obj_worksheet:工作簿对象

        '''

        list_col = ['A','B','C','D','E','F','G']

        #设置工作簿的列宽

        for c in self.xls_sheet_col_length[obj_worksheet]:

            width_string = ''.join([list_col[c],':',list_col[c]])

            if self.xls_sheet_col_length[obj_worksheet][c] < 10:

                obj_worksheet.set_column(width_string,10)

            elif self.xls_sheet_col_length[obj_worksheet][c] > 50:

                obj_worksheet.set_column(width_string,10)

            else:

                obj_worksheet.set_column(width_string,self.xls_sheet_col_length[obj_worksheet][c])

#self test           

if __name__ == '__main__':

    #文件的后缀名为xls

    file_path = os.path.join(os.getcwd(),'test.xlsx')

    xls = XlsAPI(file_path)

    #创建Excel对象

    xls.xlsOpenWorkbook()

    #添加工作对象

    sheet_names = ['sheet1','sheet2','sheet3']

    for sheet_name in sheet_names:

        sheet = xls.xlsAddWorksheet(sheet_name)

        #Excel的标题

        xls.addWorksheetTitle(sheet,['测试用例编号','测试用例名称','测试结果','备注'])

        #Excel的数据

        xls.appendWorkshetData(sheet,[1001,'test1','Pass',''],gold=0)

        xls.appendWorkshetData(sheet,[1002,'test2','Fail','失败'],gold=-1)

        xls.appendWorkshetData(sheet,[1003,'test3','Pass','调优'],gold=1)

        xls.setWorksheetWidth(sheet)

    xls.xlsCloseWorkbook()