zabbix实现邮件报警--发送对应graph图片

  
  


   
   
#!/usr/bin/python 

   
   
# 

   
   
# When: 2012/12/17 

   
   
# Who: czlinux@163.com 

   
   
# 

   
   
 

   
   
import os,sys,smtplib,MySQLdb 

   
   
from email.MIMEMultipart import MIMEMultipart 

   
   
from email.MIMEText import MIMEText 

   
   
from email.MIMEImage import MIMEImage 

   
   
 

   
   
######  Variable declaration 

   
   
hostname=os.popen('echo $HOSTNAME').read() 

   
   
 

   
   
mysql_server="127.0.0.1" 

   
   
mysql_user="scripts" 

   
   
mysql_pass="135246" 

   
   
db_name="zabbix" 

   
   
 

   
   
zabbix_url="XXXXXXXXXX" 

   
   
zabbix_user="XXXXX" 

   
   
zabbix_pass="XXXXXX" 

   
   
 

   
   
cookie="/tmp/cookie" 

   
   
image_path="/tmp/zabbix-graph/" 

   
   
Stime=os.popen('date +%Y%m%d%H%M%S').read().rstrip() 

   
   
Period=3600 

   
   
Width=1222 

   
   
 

   
   
mail_server="127.0.0.1" 

   
   
mail_port=25 

   
   
mail_from="root"+"@"+"%s" % hostname.rstrip() 

   
   
 

   
   
tag = 0 

   
   
 

   
   
###### get Host 

   
   
def _getHost(Subject): 

   
   
        Host = Subject.split(" ")[0] 

   
   
        return Host 

   
   
 

   
   
###### get Graphname 

   
   
def _getgraphname(Subject): 

   
   
        tmp1 = Subject.split(" ") 

   
   
        tmp2 = tmp1[2:-1] 

   
   
        Graphname = " ".join(tmp2) 

   
   
        return Graphname 

   
   
 

   
   
###### get graphid by Graphname and Host 

   
   
def _getgraphid(Graphname,Host): 

   
   
        db = MySQLdb.connect(mysql_server,mysql_user,mysql_pass,db_name) 

   
   
        cursor = db.cursor() 

   
   
        sql = "select graphs.graphid from hosts,items,graphs,graphs_items where hosts.host= '%s' and graphs.name = '%s' and hosts.hostid=items.hostid and items.itemid=graphs_items.itemid and graphs.graphid=graphs_items.graphid;" % (Host,Graphname) 

   
   
        cursor.execute(sql) 

   
   
        results = cursor.fetchall() 

   
   
        db.close() 

   
   
        for id in results: 

   
   
                graphid = int(id[0]) 

   
   
        return graphid 

   
   
 

   
   
###### get graph using by curl 

   
   
def _getgraph(graphid): 

   
   
        os.popen("""curl -c '%s' -b '%s' -d "request=&name='%s'&password='%s'&autologin=1&enter=Sign+in" '%s'/index.php""" % (cookie,cookie,zabbix_user,zabbix_pass,zabbix_url)) 

   
   
        os.popen("""curl -b '%s' -F "graphid=%d" -F "period=%d" -F "stime='%s'" -F "width=%d" '%s'/chart2.php > '%s''%s''.png'""" % (cookie,graphid,Period,Stime,Width,zabbix_url,image_path,Stime)) 

   
   
        image_name = '%s.png' % Stime 

   
   
        return image_name 

   
   
 

   
   
###### Creating email content 

   
   
def _content(Subject,body,mail_to,*img): 

   
   
        msgRoot = MIMEMultipart() 

   
   
        msgRoot['From'] = mail_from 

   
   
        msgRoot['To'] = mail_to 

   
   
        msgRoot['Subject'] = Subject 

   
   
 

   
   
        msgText = MIMEText(body) 

   
   
        msgRoot.attach(msgText) 

   
   
 

   
   
        if tag == 0: 

   
   
                image_path = img[0] 

   
   
                image_name = img[1] 

   
   
                f = open('%s%s' % (image_path,image_name),'rb') 

   
   
                msgimg = MIMEImage(f.read()) 

   
   
                f.close() 

   
   
                msgimg.add_header('Content-Disposition', 'attachment', filename = image_name) 

   
   
                msgRoot.attach(msgimg) 

   
   
        return msgRoot 

   
   
 

   
   
######  definition sendmail function 

   
   
def send_mail(mail_server,mail_port,mail_from,mail_to,content): 

   
   
        mail=smtplib.SMTP(mail_server,mail_port) 

   
   
        mail.sendmail(mail_from,mail_to,content) 

   
   
        mail.quit() 

   
   
 

   
   
###### definition main function 

   
   
def main(mail_to,Subject,body): 

   
   
        Host = _getHost(Subject) 

   
   
        if tag == 0: 

   
   
                Graphname = _getgraphname(Subject) 

   
   
                graphid = _getgraphid(Graphname,Host) 

   
   
                image_name = _getgraph(graphid) 

   
   
                msgRoot = _content(Subject,body,mail_to,image_path,image_name) 

   
   
        else: 

   
   
                msgRoot = _content(Subject,body,mail_to) 

   
   
        send_mail(mail_server,mail_port,mail_from,mail_to,msgRoot.as_string()) 

   
   
 

   
   
if __name__ == "__main__": 

   
   
        if "Zabbix agent" in Subject: 

   
   
                tag = 1 

   
   
        main(sys.argv[1],sys.argv[2],sys.argv[3])