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()