Python从数据库导出数据至Excel

基于Python3.6.4，导出Excel的文件后缀为.xls

#coding=utf-8
import pymysql
import xlwt
import importlib,sys
importlib.reload(sys)
#    __author__ = 'itechzero'
#    __date__ = '2018/02/26'
#    __Desc__ = 从数据库中导出数据到excel数据表中

def export(host,user,password,dbname,table_name,outputpath):
    conn = pymysql.connect(host,user,password,dbname,charset='utf8')
    cursor = conn.cursor()

    count = cursor.execute('select * from '+table_name)
    print(count)
    # 重置游标的位置
    cursor.scroll(0,mode='absolute')
    # 搜取所有结果
    results = cursor.fetchall()

    # 获取MYSQL里面的数据字段名称
    fields = cursor.description
    workbook = xlwt.Workbook()
    sheet = workbook.add_sheet('table_'+table_name,cell_overwrite_ok=True)

    # 写上字段信息
    for field in range(0,len(fields)):
        sheet.write(0,field,fields[field][0])

    # 获取并写入数据段信息
    row = 1
    col = 0
    for row in range(1,len(results)+1):
        for col in range(0,len(fields)):
            sheet.write(row,col,u'%s'%results[row-1][col])

    workbook.save(outputpath)


# 结果测试
if __name__ == "__main__":
    export('localhost','root','test','test','user',r'foo.xls')