Python 2.7_初试连接Mysql查询数据导出到excel_20161216

由于每天到公司都需要先执行一遍检测操作,观察数据是否导入完整,今天想到能否自动连接Mysql执行SQL并导出数据,每天到公司直接查看excel文件即可

时间紧,代码初次试验,边摸索边学习吧。

xlsxwriter 模块及excel格式设置的format()方法参考这里http://xlsxwriter.readthedocs.io/format.html

#coding:utf-8
import time
import datetime
import os
import MySQLdb
import xlsxwriter
#因为数据只更新到昨日 因此命名以昨日所在的年月日命名 同时今天执行时候删除昨日的文件
starttime=time.time()
yestoday= datetime.date.today()-datetime.timedelta(days=1)
beforeyestoday= datetime.date.today()-datetime.timedelta(days=2)
oldfile=str(beforeyestoday)+'.xlsx'
newfile=str(yestoday)+'.xlsx'
#删除昨日执行文件
if os.path.isfile(oldfile):
    os.remove(oldfile)
else:
    pass
#当天如果多次测试 将当天文件也删除 测试通过 条件判断可删除
if os.path.isfile(newfile):
    os.remove(newfile)
else:
    pass
#连接mysql 返回数据集对象
def getdata():
    conn = MySQLdb.connect(host='服务器IP地址', user='root', passwd='密码', db='local_db', port=3306, charset='utf8')
    cursor = conn.cursor()
    test_sql = '''
        SELECT DATE(订单日期) AS 订单日期,SUM(金额) AS 总计 ,SUM(IF(城市="北京",金额,NULL)) AS 北京,SUM(IF(城市="成都",金额,NULL)) AS 成都
       ,SUM(IF(城市="杭州",金额,NULL)) AS 杭州,SUM(IF(城市="济南",金额,NULL)) AS 济南,SUM(IF(城市="长春",金额,NULL)) AS 长春
       ,SUM(IF(城市="青岛",金额,NULL)) AS 青岛,SUM(IF(城市="南京",金额,NULL)) AS 南京,SUM(IF(城市="郑州",金额,NULL)) AS 郑州
       ,SUM(IF(城市="西安",金额,NULL)) AS 西安
        FROM a003_order
        WHERE 金额>0 AND 订单日期>=DATE_ADD(CURRENT_DATE,INTERVAL -15 DAY) AND 订单日期<CURRENT_DATE
        GROUP BY DATE(订单日期)
    '''
    count=cursor.execute(test_sql)
    data=cursor.fetchall()
    fields = cursor.description
    cursor.close()
    conn.close()
    #将数据集的表头和数据集记录传入一个字典 并返回 以便下面函数调用
    dict={}
    dict['fields']=fields
    dict['data']=data
    return dict
#获取数据集后取出记录写入excel
def write_excel():
    #调用getdata()函数 将getdata()函数返回的数据集赋值给变量fields 和 data 对data进行取出数据记录操作
    dict=getdata()
    #获取表头
    fields=dict['fields']
    #获取数据集
    data = dict['data']
    workbook = xlsxwriter.Workbook(newfile)
    worksheet = workbook.add_worksheet('data')
    #表头格式
    format1 = workbook.add_format({'bold': True, 'font_color': 'red','font_size':11,'align':'left','font_name':u'宋体'})
    #日期A列区域
    format2 = workbook.add_format({ 'font_color': 'black', 'font_size': 9,'align':'left','num_format':'yyyy-mm-dd','font_name':u'宋体'})
    #除日期A列外数值区域
    format3 = workbook.add_format({'font_color': 'black', 'font_size': 9, 'align': 'left','font_name':u'宋体'})
    #A列列宽设置能更好的显示
    worksheet.set_column("A:A", 9)
    #插入第一行表头标题
    for i in range(0, len(fields)):
        field = fields[i][0]
        worksheet.write(0, i, field.decode('utf-8'),format1)
    #插入15行第一列 订单日期 日期取字符型
    for j in range(0, 15):
        worksheet.write(j + 1, 0, str(data[j][0]).decode('utf-8'),format2)
    #插入15行第2-10列 数字取整数
    for x in range(0, 15):
        for y in range(1, 11):
            db = int(data[x][y])
            worksheet.write(x+1 , y, db,format3)
    workbook.close()
    return "写入成功"
print write_excel()
endtime=time.time()
print "用时"+str(endtime-starttime)+"s"

　执行导出到excel是下图 数据已处理 明天执行时候会先删除今天执行生成的文件