通過Python實現生成excel並郵件傳送的功能

阿新 • • 發佈：2021-08-15

1.需求描述

因業務方需求，在一個月內，需要每天從資料庫中收集、匯聚相關資料，形成3份excel，然後發給相關人。工作雖然簡單，但是還是需要花費大家的時間和經歷，所以，通過一份簡單的python程式碼來實現此功能。

2.執行檔案

3.程式碼實現

Python 的執行環境：Python 2.7.5

3.1 郵件配置檔案

郵件配置檔案smtp_conig.py，可參閱https://www.cnblogs.com/xuliuzai/p/14156559.html的傳送郵件的模組smtp_config.py,在此不再贅述。

3.2 資料庫配置檔案

資料庫配置檔案為db_weixindb_conn.py和db_qqdb_conn.py，因為是連線不同例項上的兩個資料庫，所以，我們將配置資訊放在了兩個檔案中。其具體的配置可參照

https://www.cnblogs.com/xuliuzai/p/14721413.html中的《db_conn.py》檔案。

3.3 功能程式碼

主要執行檔案report_excel_mail_.py的程式碼如下

#!/usr/bin/python
# -*- coding: UTF-8 -*-

import sys
reload(sys)
sys.setdefaultencoding( "utf-8" )
import db_weixindb_conn
import db_qqdb_conn
import os
import time
import smtp_config
from email.mime.text import 
 MIMEText
from email.header import Header
from email.mime.image import MIMEImage
from email.mime.multipart import MIMEMultipart
from email.mime.application import MIMEApplication
##引入生成excel庫
import string
import xlsxwriter
import datetime

def send_mail(mail_msg):
    # 呼叫send_mail函式
    mail_body = """ 

    <style type="text/css">
    </style>
    <html><font color=#0000CC>Dear All,<br><br>&nbsp;&nbsp;&nbsp;本郵件為QQ Order 維護部分測試郵件。主要用來測試生成excel，並且以郵件的形式傳送的功能,請互理。</html>
     """
    mail_body = mail_body + mail_msg
    message = MIMEText(mail_body, 'html', 'utf-8')
    msg = MIMEMultipart()
    msg.attach(message)
    msg.attach(xlsxpart_publish)
    msg.attach(xlsxpart)
    msg.attach(xlsxpart_slow)

    subject = 'BBChat日報'
    msg['Subject'] = Header(subject, 'utf-8')
    smtp_config.send_mail(msg)
    return

###查詢的時間範圍
### TypeError: strptime() argument 0 must be str, not <class 'datetime.datetime'>
e_time = (datetime.datetime.now() + datetime.timedelta(-1)).strftime('%Y-%m-%d')
s_time = (datetime.datetime.now() + datetime.timedelta(-8)).strftime('%Y-%m-%d')

# 定義時間標誌變數
sheet_time = datetime.datetime.now()
sheet_mark = sheet_time.strftime('%Y-%m-%d')
book_mark = sheet_time.strftime('%Y%m%d')

# 定義輸出excel檔名
workbook_publish = xlsxwriter.Workbook('BBChat_App_publish_'+book_mark+'.xlsx')
# 定義sheet的名字
worksheet_publish = workbook_publish.add_worksheet(sheet_mark)
# 定義sheet中title的字型format
bold_publish = workbook_publish.add_format({'bold': True})

# 定義輸出excel檔名
workbook_slow = xlsxwriter.Workbook('BBChat_App_slow_'+book_mark+'.xlsx')
# 定義sheet的名字
worksheet_slow = workbook_slow.add_worksheet(sheet_mark)
# 定義sheet中title的字型format
bold_slow = workbook_slow.add_format({'bold': True})

# 定義輸出excel檔名
workbook = xlsxwriter.Workbook('BBChat_App_'+book_mark+'.xlsx')
# 定義sheet的名字
worksheet = workbook.add_worksheet(sheet_mark)
# 定義sheet中title的字型format
bold = workbook.add_format({'bold': True})

#定義郵件體變數
mail_msg = ""

# 獲取資料庫連線
db = db_weixindb_conn.db
db_qqdb = db_qqdb_conn.db
# 使用cursor()方法獲取操作遊標
cursor = db_qqdb.cursor()
# SQL 查詢語句
# 備份日報
sql_apppublish_report = " select q_id,q_title,q_app_id,q_status,q_r_type,q_creator,q_create_time from qq_release " \
                     " where q_create_time>= %s and q_create_time< %s  " \
                     " and status != 0; "
try:
    # 執行SQL語句
    cursor.execute(sql_apppublish_report,(s_time,e_time))
    # 獲取所有記錄列表
    results = cursor.fetchall()
    # 查詢資料結果和欄位名字 賦值給兩個變數
    ##result = cur.fetchall() ##前面已經賦值
    fields = cursor.description # get column name

    ## 將結果寫入excel中
    ## 定義title的座標：row=0，col=0~欄位總數 也就是excel的第一行：0，0  ~ 0，len(fields),通過fields[field][0] 獲取欄位名
    for field in range(0,len(fields)):
      worksheet_publish.write(0,field,fields[field][0],bold_publish)

    for row in range(1,len(results)+1):
      for col in range(0,len(fields)):
        worksheet_publish.write(row,col,u'%s' % results[row-1][col])

    workbook_publish.close()

    ##新增郵件附件,xlsx型別的附件
    xlsxFile_publish = '/data/scripts/' + 'BBChat_App_publish_'+book_mark+'.xlsx'
    xlsxpart_publish = MIMEApplication(open(xlsxFile_publish, 'rb').read())
    xlsxpart_publish.add_header('Content-Disposition', 'attachment', filename=xlsxFile_publish)
    ##attachfile = xlsxpart
    ##傳送郵件
    #send_mail(mail_msg)

except  Exception as e:
    print str(Exception)
    print str(e)
# 關閉遊標
cursor.close()
# 關閉資料庫連線
db_qqdb.close()

####呼叫儲存生成資料
# 使用cursor()方法獲取操作遊標
cursor = db.cursor()
try:
    cursor.execute('CALL usp_collect_appBBChat()')
except  Exception as e:
    print str(Exception)
    print str(e)
cursor.close()
####資料生成

#### 生成第二個檔案
# 使用cursor()方法獲取操作遊標
cursor = db.cursor()
# SQL 查詢語句
# 備份日報
sql_app_report = " select t1.topic,t1.cnt_1,t2.cnt_2 from " \
                     " ( select topic,count(*) as cnt_1 from qqapp_alertlog_2 group by topic ) t1 " \
                     " join " \
                     " ( select topic,count(*) as cnt_2 from ( select distinct topic,api_id from qqapp_alertlog_2 ) t group by topic ) t2 on t1.topic = t2.topic " \
                     " order by cnt_1 desc;  "

try:
    # 執行SQL語句
    cursor.execute(sql_app_report)
    # 獲取所有記錄列表
    results = cursor.fetchall()
    # 查詢資料結果和欄位名字 賦值給兩個變數
    ##result = cur.fetchall() ##前面已經賦值
    fields = cursor.description # get column name

    ## 將結果寫入excel中
    ## 定義title的座標：row=0，col=0~欄位總數 也就是excel的第一行：0，0  ~ 0，len(fields),通過fields[field][0] 獲取欄位名
    for field in range(0,len(fields)):
      worksheet.write(0,field,fields[field][0],bold)

    for row in range(1,len(results)+1):
      for col in range(0,len(fields)):
        worksheet.write(row,col,u'%s' % results[row-1][col])

    workbook.close()

    ##新增郵件附件,xlsx型別的附件
    xlsxFile = '/data/scripts/' + 'BBChat_App_'+book_mark+'.xlsx'
    xlsxpart = MIMEApplication(open(xlsxFile, 'rb').read())
    xlsxpart.add_header('Content-Disposition', 'attachment', filename=xlsxFile)
##### 生成第二個檔案 End ###
    ##send_mail(mail_msg)

except  Exception as e:
    print str(Exception)
    print str(e)
# 關閉遊標
cursor.close()
# 關閉資料庫連線
#db.close()

#### 生成第三個檔案
# 使用cursor()方法獲取操作遊標
cursor = db.cursor()
# SQL 查詢語句
# 備份日報
sql_app_slow = " select id, core_api, post_time,topic,app_id, api_id,method,cnt, taken_avg,fail_cnt,fail_per " \
                     " from qqapp_alertlog_2  "

try:
    # 執行SQL語句
    cursor.execute(sql_app_slow)
    # 獲取所有記錄列表
    results = cursor.fetchall()
    # 查詢資料結果和欄位名字 賦值給兩個變數
    ##result = cur.fetchall() ##前面已經賦值
    fields = cursor.description # get column name

    ## 將結果寫入excel中
    ## 定義title的座標：row=0，col=0~欄位總數 也就是excel的第一行：0，0  ~ 0，len(fields),通過fields[field][0] 獲取欄位名
    for field in range(0,len(fields)):
      worksheet_slow.write(0,field,fields[field][0],bold_slow)

    for row in range(1,len(results)+1):
      for col in range(0,len(fields)):
        worksheet_slow.write(row,col,u'%s' % results[row-1][col])

    workbook_slow.close()

    ##新增郵件附件,xlsx型別的附件
    xlsxFile_slow = '/data/scripts/' + 'BBChat_App_slow_'+book_mark+'.xlsx'
    xlsxpart_slow = MIMEApplication(open(xlsxFile_slow, 'rb').read())
    xlsxpart_slow.add_header('Content-Disposition', 'attachment', filename=xlsxFile_slow)
##### 生成第二個檔案 End ###
    send_mail(mail_msg)

except  Exception as e:
    print str(Exception)
    print str(e)
# 關閉遊標
cursor.close()
# 關閉資料庫連線
db.close()