zabbix screen 图片以邮件形式发送

zabbix screen 图片以邮件形式发送

#! /usr/bin/env python
#coding=utf-8
# Andy_f
import time,os
import urllib
import urllib2
import cookielib
import MySQLdb
import smtplib
from email.mime.text import MIMEText

screen = "Servers"
#
save_graph_path = "/usr/share/zabbix/reports/%s"%time.strftime("%Y-%m-%d")
if not os.path.exists(save_graph_path):
    os.makedirs(save_graph_path)
# zabbix host
zabbix_host = "10.10.10.1/zabbix"
# zabbix login username
username = "****"
# zabbix login password
password = "****"
# graph width
width = 500
# graph height
height = 100
# graph Time period, s
period = 604800
# zabbix DB
dbhost = "localhost"
dbport = 3306
dbuser = "zabbix"
dbpasswd = "******"
dbname = "zabbix"
# mail
to_list = ["ygw@163.com"]
smtp_server = "10.10.10.8"
mail_user = "zabbix"
mail_pass = "xxxxx"
domain  = "163.com"
subj = 'Servers Status ' + time.strftime('%Y-%m-%d',time.localtime(time.time()))

def mysql_query(sql):
    try:
        conn = MySQLdb.connect(host=dbhost,user=dbuser,passwd=dbpasswd,port=dbport,connect_timeout=20)
        conn.select_db(dbname)
        cur = conn.cursor()
        count = cur.execute(sql)
        if count == 0:
            result = 0
        else:
            result = cur.fetchall()
        return result
        cur.close()
        conn.close()
    except MySQLdb.Error,e:
        print "mysql error:" ,e

def get_graph(zabbix_host,username,password,screen,width,height,period,save_graph_path):
    global screenid
    screenid = '27'
    global html
    html = ''
    graphid_list = [8895,8901,8903,8905,8897,8899,8926,8907,8909]
    for graphid in graphid_list:
        login_opt = urllib.urlencode({
        "name": username,
        "password": password,
        "autologin": 1,
        "enter": "Sign in"})
        get_graph_opt = urllib.urlencode({
        "graphid": graphid,
        "screenid": screenid,
        "width": width,
        "height": height,
        "period": period})
        cj = cookielib.CookieJar()
        opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cj))
        login_url = r"http://%s/index.php"%zabbix_host
        save_graph_url = r"http://%s/chart2.php"%zabbix_host
        opener.open(login_url,login_opt).read()
        data = opener.open(save_graph_url,get_graph_opt).read()
        filename = "%s/%s.%s.png"%(save_graph_path,screenid,graphid)
        html += '<img width="600" height="250" src="http://%s/%s/%s/%s.%s.png">'%(zabbix_host,save_graph_path.split("/")[len(save_graph_path.split("/"))-2],save_graph_path.split("/")[len(save_graph_path.split("/"))-1],screenid,graphid) + '<p><br></p>'
        f = open(filename,"wb")
        f.write(data)
        f.close()            

def send_mail(username,password,smtp_server,to_list,sub,content):
    print to_list
    me = "Monitoring"+"<"+username+"@"+domain +">"
    msg = MIMEText(content,_subtype="html",_charset="utf8")
    msg["Subject"] = sub
    msg["From"] = me
    msg["To"] = ";".join(to_list)
    try:
        server = smtplib.SMTP()
        server.connect(smtp_server)
     #   server.login(username,password)
        server.sendmail(me,to_list,msg.as_string())
        server.close()
        print "send mail Ok!"
    except Exception, e:
        print e

if __name__ == '__main__':
#    for screen in screens:
    get_graph(zabbix_host,username,password,screen,width,height,period,save_graph_path)
    send_mail(mail_user,mail_pass,smtp_server,to_list,subj,html)