python 實現zabbix自動發送報表

實現將zabbix中的screens中的圖片以郵件的方式發送到管理員郵箱

你們都知道zabbix的screens通常是定義管理員比較關注的監控指標好方便瀏覽,那麼咱們就根據管理員自定義的screens name獲取相關圖信息的.

簡介

首先咱們來看下圖片的url (右擊screens中隨便一張圖片,選複製圖片地址)

http://zabbix.xxx.com/chart2.php?graphid=524&screenid=16&width=400&height=156&updateProfile=1&profileIdx=web.screens&profileIdx2=16&period=86400&stime=20150629172712&sid=f6dd0d127bba0123&curtime=1404120808037

能夠看到上面的url後面有不少的參數,固然咱們須要的沒有那麼多,簡化後的url以下

http://zabbix.xxx.com/chart2.php?graphid=524&screenid=16&width=400&height=156&period=86400

簡化後咱們只須要5個參數

graphid     # 圖片的ID,對應的值爲某一個graph的惟一標識
screenid    # screenID,對應的值爲某一個screen的惟一標識
width       # 圖片的寬度,對應的值爲圖片的寬度(可根據須要本身定義)
height      # 圖片的高度,對應的值爲圖片的高度(可根據須要本身定義)
period      # 圖片展現數據的時鐘週期 單位爲 秒 (86400 =1天)

經過上面的信息咱們知道要經過zabbix獲取screen中的圖片須要的參數下面咱們須要作的就是經過screen name到數據庫中查詢相關參數便可

一、根據管理員自定義的screens name在數據庫中查詢此screen的ID (screens 表)

二、根據所查到的screenID查詢此screen中所包含的resourceid (resourceid==graphid)(screens_items 表)

三、經過一、2兩步查到的參數獲取圖片並保存至zabbix服務器上的web目錄並生成html代碼

四、經過smtplib模塊把html代碼發送到管理員郵箱

大概的流程是這樣的,代碼以下,代碼中所涉及的參數變量根據本身的環境作出相應的配置便可

cat zabbix_send_report.py

#! /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

screens = ["xxx","xxx"]
#
save_graph_path = "/var/www/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 = "zabbix.xxx.com"
# zabbix login username
username = "admin"
# zabbix login password
password = "zabbix"
# graph width
width = 600
# graph height
height = 100
# graph Time period, s
period = 86400
# zabbix DB
dbhost = "xxx.xxx.xxx.xxx"
dbport = 3306
dbuser = "zabbix"
dbpasswd = "xxxxx"
dbname = "zabbix"
# mail
to_list = ["xxx@xxx.com","aaa@xxx.com"]
smtp_server = "smtp.163.com"
mail_user = "xxxx"
mail_pass = "xxxxx"
domain  = "163.com"

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):
    screenid_list = []
    global html
    html = ''
    for i in mysql_query("select screenid from screens where name='%s'"%(screen)):
                for screenid in i:
                    graphid_list = []
                    for c in mysql_query("select resourceid from screens_items where screenid='%s'"%(int(screenid))):
                        for d in c:
                            graphid_list.append(int(d))
                    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)
                        f = open(filename,"wb")
                        f.write(data)
                        f.close()


def send_mail(username,password,smtp_server,to_list,sub,content):
    print to_list
    me = "運維"+"<"+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,"test email",html)

設置crontab 天天早上上班前執行一次

crontab -e
45 08 * * * python /root/zabbix_send_report.py

上效果圖