用python脚本导出mysql数据库查询结果到Excel表

Posted 2021-01-13

最近需要导数据的情况太多，总用跳板机上的navicat工具有点效率低，也觉得挺麻烦的(由于跳板机无法连通外网 所以导出数据文件还得通过sftp传到本机)
anyway 还是写个脚本好了。之前写过一个shell脚本做的定时导出任务，现在试试用python写下

主要用到的库有:
pymysql -- 连数据库的无需多说
os & sys -- 可能回涉及到工作目录或者外部传参
xlwt -- 写excel

下面就是代码了，总体还是挺简单的，主要遇到个字符编码的问题，改成utf-8就解决了
还有个没解决掉的问题: select的结果没有字段名，所以....文件第一行的字段我是手动加进去的 =.=

#!/usr/bin/python
# -*- coding: cp936 -*-
# security:数据库相关的敏感参数最好通过传参形式传入而非写入脚本

import pymysql
import xlwt
import os,sys

def dbConnect(dburl):
    db_user = sys.argv[1]
    db_pass = sys.argv[2]
    db_name = raw_input("要使用的数据库:")
    db_conn = pymysql.connect(dburl,db_user, db_pass, db_name)# db_name传参失败?编码问题"?"
    return db_conn

def sqlOpt(opt):
    db_conn = dbConnect("localhost")
    cur     = db.cursor()
    cur.execute(opt)
    data    = cur.fetchall()
    db_conn.close()
    return data

def write_to_excel(content):
    os.chdir("/home/appuser/export_mysql_data/")
    filename   = raw_input("请输入要保存的文件名,无需后缀:") + ‘.xls‘
    wbk        = xlwt.Workbook(encoding=‘utf-8‘)
    test       = wbk.add_sheet(‘test‘,cell_overwrite_ok=True)
    # ?如何获取列名
    fileds     = [u‘ID‘,u‘name‘,u‘sex‘,u‘birth‘,u‘department‘,u‘address‘]
    trans_data = list(content)
    # 写入列名
    for filed in range(0,len(fileds)):
        test.write(0,filed,fileds[filed])
    for row in range(1,len(trans_data)+1):
        for col in range(0,len(fileds)):
            test.write(row,col,str(trans_data[row-1][col]))
    wbk.save(filename)

def run_Task():
    sql    = raw_input("请输入sql查询语句:")
    result = sqlOpt(sql)
    write_to_excel(result)

run_Task()