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}