Python腳本：實現對象集合List導入到excel表格，支持mysql，postergrsql，MongoDB

import xlwt
import os
import datetime



#驗證export_filed中的字段是否在對象字段中
def checkField(obj_list,filed_dict):
    flag = True
    msg = ''
    if obj_list == []:
        return False,'導出內容為空'
    else:
        for i in range(len(obj_list)):
            for field in filed_dict:
                try:
                    obj_list[i][field]
                except:
                    flag = False
                    if len(msg) <=100:
                        msg +='第{0}行未找到{1}字段'.format(i+1,field)
                    else:
                        msg += '....'
                        break
        return flag, msg



#1.存放導出表格的目錄
#2.table_name:導出的表格名字,不帶后綴名，默認為.xls
#3. obj_list:需要導出對象集合 ,通過sql語句查詢獲得的對象集合
#4.export_filed：導出的字段和中文的對應關系，字段必須時對象里面包含的,
# 如{'name':'名字','age':'年齡'}，不可為空，只有在這里的字段才會被導出
def tableExportToXlsx(export_dir,table_name,obj_list,filed_dict):

    #1.創建表格
    workbook = xlwt.Workbook()
    #創建excel的一個sheet
    sheet = workbook.add_sheet(table_name,cell_overwrite_ok=True)

    #2.驗證export_filed中的字段是否在對象字段中
    flag,msg = checkField(obj_list,filed_dict)
    if not flag:
        return {'msg':msg}
    else:#通過驗證
        attrs_list = []
        #3.在表格第一行寫入表格的中文字段，即是表頭
        #將表格字段提取出來轉化成數組
        table_index = [k for k,v in filed_dict.items()]
        for i in range(len(table_index)):
            #寫入中文字段
            sheet.write(0,i,filed_dict[table_index[i]])

        #4.對象導入表格
        for row in range(len(obj_list)):
            for col in range(0,len(table_index)):
                sheet.write(row+1,col,obj_list[row][table_index[col]])

        #5.存儲文件
        if not os.path.exists(export_dir):
            os.makedirs(export_dir)
        #加上導出時間
        export_time = datetime.datetime.now().strftime('%Y%m%d%H%M%S')
        file_name = '{0}-{1}.xls'.format(table_name,export_time)
        save_path = os.path.join(export_dir,file_name)
        print('文件路徑為' + save_path)
        try:
            workbook.save(save_path)
        except:
            print(export_dir+'目錄不存在')

        #6.檢查是否導出成功
        if os.path.isfile(save_path):
            print('導出成功,文件為'+save_path)
            return {'msg':'ok','file_name':file_name}
        else:
            msg = '存儲文件發生異常，檢查{0}目錄是否存在'.format(export_dir)
            print(msg)
            return  {'msg':msg}