Python通过ssh链接Mysql定时发送邮件V2

不积跬步,无以至千里;不积小流,无以成江海

之前写过一篇python自动发邮件的文章,这段时间又进行了改进,来更新一下。
主要更新内容有:
1)通过pandas连接MySQL数据库
2)多sql语句查询
3)不同收件人,发送不同内容
4)pandas生成日期列表
5)增加错误日志信息

接下来一一介绍,实际业务当中,很少有一个sql语句可以得出结果的,一般都需要一些临时表,那么我们怎么来实现呢,在python中利用pandas就比较容易了。之前利用了mysqldb的cursor,这次用一个更擅长处理数据的pandas,用这个工具实际处理起来更容易、方便一些。

# 我们使用pandas的read_sql函数,
# 将具体的sql语句传入,就可以将结果保存为DataFrame
def dbconnect_ssh_dq(ssh_host,ssh_port,ssh_password,ssh_user,db_host,db_name,db_port,db_user,db_passwd):
    with SSHTunnelForwarder(
            (ssh_host, ssh_port),
            ssh_password = ssh_password,
            #ssh_pkey=keyfile,
            ssh_username = ssh_user,
            remote_bind_address = (db_host, db_port)
    ) as server:
        db = MySQLdb.connect(
            host='127.0.0.1',
            port=server.local_bind_port,
            user=db_user,
            passwd=db_passwd,
            db=db_name,
            charset="utf8")

        df = pd.read_sql('''SELECT
date_y,
uid,
sum(money) amt
FROM
table
GROUP BY date_y DESC,uid;''', con = db)

        db.close()
        return df

2)对于多sql语句查询,也可以利用类似的方法,我这里由于将脚本都放入了txt中,就通过一个循环,将语句依次读入,生成临时变量,然后再返回

# 这段也动态生成了十几个变量
def dbconnect_ssh(ssh_host,ssh_port,ssh_password,ssh_user,db_host,db_name,db_port,db_user,db_passwd,province):
    with SSHTunnelForwarder(
            (ssh_host, ssh_port),
            ssh_password = ssh_password,
            #ssh_pkey=keyfile,
            ssh_username = ssh_user,
            remote_bind_address = (db_host, db_port)
    ) as server:
        db = MySQLdb.connect(
            host='127.0.0.1',
            port=server.local_bind_port,
            user=db_user,
            passwd=db_passwd,
            db=db_name,
            charset="utf8")

        for i in range(1, 13):
            sql = ''
            fd = open('/home/data/datacenter/sql_' + str(i) + '.txt')
            lines = fd.readlines()
            for line in lines:
                line = line.strip('\n')
                sql += line + ' '
            sql = sql.replace(':prov', province)
            print '1st',i,sql
            #locals()['sql_' + str(i)] = sql
            #print 'var sql',i,locals()['sql_' + str(i)]
            locals()['df' + str(i)] = pd.read_sql(sql, con = db)
            #print locals()['df1']
            #print 'df done',locals()['df' + str(i)]
            #df += locals()['df' + str(i)]
        db.close()

        return locals()['df1'],locals()['df2'],locals()['df3'],locals()['df4'],locals()['df5'],locals()['df6'],locals()['df7'],locals()['df8'],locals()['df9'],locals()['df10'],locals()['df11'],locals()['df12']

3)不同的收件人,发送不同的内容,我这边是通过一个Excel文件来记录不同的收件人,发送什么内容的,其实发送的内容是有规律的,就是某个人获取某个省份的数据,我只要在excel里面记录好收件人和省份的对应关系就好了,接下来就是读取excel的列作为参数传入发送邮件的函数就好了。

# 我这里写了一个函数用来返回省份、地址、收件人、抄送人等
# 接下来就可以作为参数出入发送邮件的函数
def email_to_person(filename):
    wb = xlrd.open_workbook(filename)
    ws = wb.sheet_by_name('table_message')
    prov = []
    email_addr = []
    prov_name = []
    prov_no = []
    email_cc = []
    for rownum in range(ws.nrows):
        for colnum in range(ws.ncols):
            if colnum == 0:
                prov_name.append(ws.cell(rownum,colnum).value)
            elif colnum == 1:
                email_addr.append(ws.cell(rownum,colnum).value)
            elif colnum == 2:
                prov.append(ws.cell(rownum,colnum).value)
            elif colnum == 3:
                prov_no.append(ws.cell(rownum,colnum).value)
            else:
                email_cc.append(ws.cell(rownum,colnum).value)
    return prov,email_addr,prov_no,email_cc

4)pandas生成日期列表,当某一天没有数据的时候,我们不希望这一天的数据缺失,我们想这一天的数据都是为0.为了解决这个问题,我想到了弄一个日历表,这样数据和日历表关联,每日的数据都有了,如果这天没有交易,就用0去填补。其实pandas在处理时间、日期方面是很强大的,我看了一下文档,着实震惊了,连自定义的日历、工作日函数都有,太强大了。言归正传,这里弄一个七天的日期列表就好了:

# 是不是很强大啊
def datelist(beginDate, endDate):
    # '20160601' or datetime
    date_l=[x.strftime('%Y%m%d') for x in list(pd.date_range(start=beginDate, end=endDate))]
    return pd.DataFrame(date_l, columns = ['date_y'])

5)增加错误日志信息,这个需求logging和traceback模块

# 其中level用来定义日志的级别,我这边跟踪错误日志信息,就设置了error
logging.basicConfig(level = logging.ERROR,
                format = '%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',
                datefmt = '%a, %d %b %Y %H:%M:%S',
                filename = '/home/data/datacenter/mylog.log',
                filemode = 'w')

    #定义一个StreamHandler,将INFO级别或更高的日志信息打印到标准错误,并将其添加到当前的日志处理对象#
    console = logging.StreamHandler()
    console.setLevel(logging.ERROR)
    formatter = logging.Formatter('%(name)-12s: %(levelname)-8s %(message)s')
    console.setFormatter(formatter)
    logging.getLogger('').addHandler(console)

# 然后通过try except将错误信息记录在mylog.log文件
try:
    pass
except:
        s = traceback.format_exc()
        logging.error(s)

下面是完整代码:

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

import MySQLdb
from sshtunnel import SSHTunnelForwarder
import MySQLdb.cursors
import xlwt
import os

import sys
import time,datetime
from smtplib import SMTP
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from email.mime.application import MIMEApplication
import datetime
import os.path
import mimetypes
from email.MIMEBase import MIMEBase
from email import Encoders
import pandas as pd 
import numpy as np
import pickle
import xlrd,xlwt
import logging,traceback

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


def dbconnect_ssh(ssh_host,ssh_port,ssh_password,ssh_user,db_host,db_name,db_port,db_user,db_passwd,province):
    with SSHTunnelForwarder(
            (ssh_host, ssh_port),
            ssh_password = ssh_password,
            #ssh_pkey=keyfile,
            ssh_username = ssh_user,
            remote_bind_address = (db_host, db_port)
    ) as server:
        db = MySQLdb.connect(
            host='127.0.0.1',
            port=server.local_bind_port,
            user=db_user,
            passwd=db_passwd,
            db=db_name,
            charset="utf8")

        for i in range(1, 13):
            sql = ''
            fd = open('/home/data/datacenter/sql_' + str(i) + '.txt')
            lines = fd.readlines()
            for line in lines:
                line = line.strip('\n')
                sql += line + ' '
            sql = sql.replace(':prov', province)
            print '1st',i,sql
            #locals()['sql_' + str(i)] = sql
            #print 'var sql',i,locals()['sql_' + str(i)]
            locals()['df' + str(i)] = pd.read_sql(sql, con = db)
            #print locals()['df1']
            #print 'df done',locals()['df' + str(i)]
            #df += locals()['df' + str(i)]
        db.close()

        return locals()['df1'],locals()['df2'],locals()['df3'],locals()['df4'],locals()['df5'],locals()['df6'],locals()['df7'],locals()['df8'],locals()['df9'],locals()['df10'],locals()['df11'],locals()['df12']

def dbconnect_ssh_dq(ssh_host,ssh_port,ssh_password,ssh_user,db_host,db_name,db_port,db_user,db_passwd):
    with SSHTunnelForwarder(
            (ssh_host, ssh_port),
            ssh_password = ssh_password,
            #ssh_pkey=keyfile,
            ssh_username = ssh_user,
            remote_bind_address = (db_host, db_port)
    ) as server:
        db = MySQLdb.connect(
            host='127.0.0.1',
            port=server.local_bind_port,
            user=db_user,
            passwd=db_passwd,
            db=db_name,
            charset="utf8")

        df = pd.read_sql('''SELECT
date_y,
uid,
sum(money) amt
FROM
table
GROUP BY date_y DESC,uid;''', con = db)

        db.close()
        return df

def send_mail(config):
    print 'Sending Mail...'
    message = MIMEMultipart()
    message["Accept-Charset"] = "ISO-8859-1,utf-8"
    message['From'] = '[email protected]'
    #print 'to',config['to']
    message['To'] =','.join(config['to'])
    #print 'message_to',message['To']
    message['CC'] = ','.join(config['cc'])
    message['Subject'] = config['subject']
    message['Date'] = time.ctime(time.time())
    #message['Reply-To'] = '[email protected]'
    message['X-Priority'] = '3'
    message['X-MSMail-Priority'] = 'Normal'
    if config['text']:
        text = config['text']
        message.attach(text)
    smtp = SMTP(config['server'], config['port'])

    username = '[email protected]'
    smtp.login(username, 'password')

    smtp.sendmail(username, config['to'] + config['cc'], message.as_string())
    print 'Send Mail OK'
    print time.strftime("%Y%m%d %H:%M:%S", time.localtime())
    #print 'to',message['To']
    #print 'cc',message['CC']
    smtp.close()
    time.sleep(1)

def send_mail_to_test(context,curr_time,email,prov_n,email_c):
    #print '$$$$$',email
    send_mail({
        #'to': ["[email protected]"],
        #'cc': ['[email protected]'],
        'to': email.encode('utf8').split(','),
        'cc': email_c.encode('utf8').split(','),
        'server': 'smtp.qiye.163.com',
        'port': 25,
        'subject': '【上海数据部】' + prov_n + '日交易汇报'+curr_time,
        'username': '[email protected]',
        'password': 'password',
        'text': context}
    )

def message_from_excel(filename,email,prov_n,email_c):
    #print 'html', email
    wb = xlrd.open_workbook(filename)
    #ws = wb.sheet_by_name('table_message')
    title = [u'日交易数据',u'交易明细']
    html_content = "" + "" + "" + "

Dear all,

邮件为" + prov_n + "日交易数据,已更新至" + datetime.datetime.now().strftime('%Y-%m-%d') +",请查阅,谢谢!

" content = "" for s in range(2): content += "" ws = wb.sheets()[s] for rownum in range(ws.nrows): content += "" for colnum in range(ws.ncols): #print int(ws.cell(rownum,colnum).value) content += "".format(ws.cell(rownum,colnum).value) content += "" content += "
" + prov_n + title[s] + "
{0}
" content += "
" html_body = "" + content + '

说明:

1、say something!

2、如有需求变更,请联系[email protected]

' + "

" html_content += html_body #html_content += '

以上数据为系统自动发送,如有疑问请联系[email protected]


' html_content += "" context = MIMEText(html_content,_subtype='html',_charset='utf-8') #print html_content send_mail_to_test(context,curr_time,email,prov_n,email_c) def email_to_person(filename): wb = xlrd.open_workbook(filename) ws = wb.sheet_by_name('table_message') prov = [] email_addr = [] prov_name = [] prov_no = [] email_cc = [] for rownum in range(ws.nrows): for colnum in range(ws.ncols): if colnum == 0: prov_name.append(ws.cell(rownum,colnum).value) elif colnum == 1: email_addr.append(ws.cell(rownum,colnum).value) elif colnum == 2: prov.append(ws.cell(rownum,colnum).value) elif colnum == 3: prov_no.append(ws.cell(rownum,colnum).value) else: email_cc.append(ws.cell(rownum,colnum).value) return prov,email_addr,prov_no,email_cc def datelist(beginDate, endDate): # '20160601' or datetime date_l=[x.strftime('%Y%m%d') for x in list(pd.date_range(start=beginDate, end=endDate))] return pd.DataFrame(date_l, columns = ['date_y']) if __name__ == '__main__': curr_time = time.strftime("%Y%m%d %H:%M:%S", time.localtime()) print curr_time ssh_host = "2.2.22.2" ssh_port = 22222 ssh_password = "1111111111111111" ssh_user = "name" db_host = "5.5.5.55" db_name = "name" db_port = 3306 db_user = "user_name" db_passwd = "333333333333333333" db_host_dq = "7.7.7.7" db_name_dq = "db_name" db_port_dq = 3306 db_user_dq = "user_name" db_passwd_dq = "0000000000000000000" endDate = datetime.datetime.now() delta = datetime.timedelta(days = 1) endDate = endDate - delta delta = datetime.timedelta(days = 6) beginDate = endDate - delta endDate = endDate.strftime('%Y%m%d') beginDate = beginDate.strftime('%Y%m%d') print beginDate,endDate df_date = datelist(beginDate, endDate) print df_date Excelfile = '/home/data/datacenter/r.xls' email_prov_file = '/home/data/datacenter/email_prov.xls' logging.basicConfig(level = logging.ERROR, format = '%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s', datefmt = '%a, %d %b %Y %H:%M:%S', filename = '/home/data/datacenter/mylog.log', filemode = 'w') #定义一个StreamHandler,将INFO级别或更高的日志信息打印到标准错误,并将其添加到当前的日志处理对象# console = logging.StreamHandler() console.setLevel(logging.ERROR) formatter = logging.Formatter('%(name)-12s: %(levelname)-8s %(message)s') console.setFormatter(formatter) logging.getLogger('').addHandler(console) prov = [] email_addr = [] prov_n = [] email_cc = [] prov, email_addr, prov_n,email_cc = email_to_person(email_prov_file) print prov_n try: for j in range(int(sys.argv[1]), int(sys.argv[2])): df1, df2, df3, df4, df5, df_user, df7, df8, df9, df10, df11, df12 = dbconnect_ssh(ssh_host,ssh_port,ssh_password,ssh_user,db_host,db_name,db_port,db_user,db_passwd,prov[j]) print df1 df_dq = dbconnect_ssh_dq(ssh_host,ssh_port,ssh_password,ssh_user,db_host_dq,db_name_dq,db_port_dq,db_user_dq,db_passwd_dq) #print df_dq df_dq = pd.merge(df_dq, df_user, how = 'inner', on = 'uid') df_dq = df_dq['amt'].groupby(df_dq['date_y']).sum().reset_index() print df_dq df_dq.columns = ['date_y', 'amt'] print df_dq.columns df1 = pd.concat([df1, df_dq]) print df1 df1 = df1['amt'].groupby(df1['date_y']).sum().reset_index() df1 = df1.sort_values(by = ['date_y'], ascending = False) print df1 df = pd.merge(df1, df2, how = 'outer', on = 'date_y') df = pd.merge(df, df3, how = 'outer', on = 'date_y') #print df.head() #df['total_1'] = df.apply(lambda x: x.sum(), axis=1) #df.head() df = pd.merge(df, df4, how = 'outer', on = 'date_y') df = pd.merge(df, df5, how = 'outer', on = 'date_y') df = df.fillna(0) print df #df.iloc[5,3] = 0 #print df cols = list(df) #print cols df = df.ix[:, cols] df = df.sort_index(by = u'日期', ascending = False) df = df.set_index(u'日期') #print df.head() #print df[df.index.duplicated()] df_detail = pd.merge(df7, df8, how = 'outer', on = 'date_y') df_detail = pd.merge(df_detail, df_dq, how = 'outer', on = 'date_y') df_detail = pd.merge(df_detail, df9, how = 'outer', on = 'date_y') df_detail = pd.merge(df_detail, df10, how = 'outer', on = 'date_y') df_detail = pd.merge(df_detail, df11, how = 'outer', on = 'date_y') df_detail = pd.merge(df_detail, df12, how = 'outer', on = 'date_y') df_detail = pd.merge(df_date, df_detail, how = 'outer', on = 'date_y') df_detail = df_detail.fillna(0) df_detail = df_detail.sort_index(by='date_y', ascending = False) out = pd.ExcelWriter(Excelfile) #df3.to_excel(out,sheet_name = 'df3') df = df[df.columns].astype('int') df.to_excel(out,sheet_name = 'df') df_detail.to_excel(out, sheet_name = 'df_detail') out.save() #print 'prov_n ',j,prov_n[j] message_from_excel(Excelfile,email_addr[j],prov_n[j],email_cc[j]) except: s = traceback.format_exc() logging.error(s)

你可能感兴趣的:(Python通过ssh链接Mysql定时发送邮件V2)