Python mysql數據 讀取時間參數 for循環寫入Excel文件

最近在利用Python 實現自動化表報時,有個功能是mysql的業務時間是讀取模板文件的時間參數,需要用到for循環功能,基本思路是:

1.自動創建一個輸出文件的文件夾

2.根據模板文件創建一個新的excel文件到新創建的文件夾中

3.每次寫入時返回sheet的最大行數max_row,下次寫入時從最大行的下一行開始繼續寫入

4.每次讀取必須爲同一個文件

代碼如下:

#! /usr/bin/env python
# -*- coding:utf-8 -*-

import MySQLdb
from openpyxl import load_workbook
import sys
import time
import os

reload(sys)
sys.setdefaultencoding('utf8')


# 打開數據庫連接
db = MySQLdb.connect(host="localhost", user="zimu", passwd="zimu", db="xxx", port=0000,charset='utf8')

template_file_demo = r"D:\path\demo.xlsx"
# makedirs 創建文件時如果路徑不存在會創建這個路徑
output_path = r"D:\output\demo"+"_"+  time.strftime("%Y%m%d", time.localtime()) +"_" + str(int(time.time()))+"\\"
os.makedirs(output_path)
#創建文件到新創建的文件夾中
book_demo = load_workbook(template_file_demo)
book_demo.save(output_path + "demo" +"_"+time.strftime("%Y%m%d", time.localtime())+".xlsx")
#讀取指定文件夾下的文件
demo_file = output_path+"demo"+"_"+time.strftime("%Y%m%d", time.localtime())+".xlsx"

def savedata(start_time,end_time):
	demosql = '''select * from demo where start_date<='%s' and end_date>='%s''''%(start_time,end_time)
    cursor = db.cursor()
    cursor.execute(demosql)
    demodata = cursor.fetchall()

    demo_book = load_workbook(demo_file)
    demosheet = demo_book['demo']
    row_t = demosheet.max_row

    i = 0
    while i < len(demodata):
        for j in range(0, 8):
            demosheet.cell(row_t + i + 1, j + 1).value = demodata[i][j]
        i += 1
        demo_book.save(output_path+"demo"+"_"+time.strftime("%Y%m%d", time.localtime())+".xlsx")

book_template = load_workbook(template_file_demo)
timet = book_template['時間配置']
for t in range(2, timet.max_row + 1):  # 讀取配置表中的時間
    savedata(timet.cell(t, 1).value, timet.cell(t, 2).value)







5.模板文件的時間參數設置如下:

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章