脚本处理逻辑分析：
通过zabbix传递给脚本的message参数，筛选出报警信息的itemid;
通过itemid获取到图片并保存;
将报警信息和图片组装成html;
发送邮件。
后续脚本里面的处理细节还会在进一步分析，将脚本中涉及到的代码行进行注释分析
#!/usr/bin/env python
#coding=utf-8
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from email.mime.image import MIMEImage
import MySQLdb,smtplib,sys,os,time,re
import graph
import logging
import logging.config
import logging.handlers
import traceback
import requests
#日志输出配置格式
logger = logging.getLogger("root")
logger.setLevel(logging.DEBUG)
formatter = logging.Formatter("[%(asctime)s %(filename)s:%(lineno)s - %(funcName)15s()] %(message)s")
#日志输出位置，到指定/hskj/logs/Data_sheet.log这个文件，从系统规划上，日志统一放置到这个地方
handler = logging.handlers.RotatingFileHandler("/hskj/logs/altermail_img.log")
handler.setFormatter(formatter)
logger.addHandler(handler)
user='USER'
#zabbix用户名
password='PASSWORD'
#zabbix密码
url='http://IP/zabbix/'
#zabbix首页
period='3600'
chart2_url='http://IP/chart2.php'
#zabbix获取图片url http://IP/chart2.php
mysql_host='localhost'
mysql_user='root'
mysql_pass='MYSQLPASSWD'
mysql_db='zabbix'
#zabbix数据库相关信息
graph_path='/usr/local/zabbix/alertscripts/'
#图片保存路径
def get_itemid():
#获取itemid
a=re.findall(r"ITEM ID:\d+",sys.argv[3])
i=str(a)
itemid=re.findall(r"\d+",i)
logger.info(itemid)
return str(itemid).lstrip('[\'').rstrip('\']')
def get_graphid(itemid):
#获取graphid
conn =MySQLdb.connect(host=mysql_host,user=mysql_user,passwd=mysql_pass,db=mysql_db,connect_timeout=20)
cur=conn.cursor()
cur.execute("select graphid from graphs_items where itemid=%s;" %itemid)
result=cur.fetchone()
cur.close()
conn.close()
graphid=re.findall(r'\d+',str(result))
logger.info(graphid)
return str(graphid).lstrip('[\'').rstrip('\']')
#get grafa#
def get_graph(itemID,pName=None):
myRequests = requests.Session()
HOST='IP'
try:
"""
获取性能图，首先需要登录
通过分析，可以直接Post/Get方式登录
"""
loginUrl = "http://%s/zabbix/index.php" % HOST
loginHeaders={
"Host":HOST,
"Accept":"text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8",
}
# 构建登录所需的信息
playLoad = {
"name":"USER",
"password":'PASSWORD',
"autologin":"1",
"enter":"Sign in",
}
# 请求登录
res = myRequests.post(loginUrl,headers=loginHeaders,data=playLoad)
"""
登入状态后，在POST数据中加入itemid
"""
testUrl = "http://%s/zabbix/chart.php" % HOST
testUrlplayLoad = {
"period" :"10800",
"itemids[0]" : itemID,
"type" : "0",
"profileIdx" : "web.item.graph",
"width" : "700",
}
testGraph = myRequests.get(url=testUrl,params=testUrlplayLoad)
# 返回图片源码，直接保存到本地
IMAGEPATH = os.path.join('/usr/local/zabbix/alertscripts/imges/', pName)
f = open(IMAGEPATH,'wb')
f.write(testGraph.content)
f.close()
pName = '/usr/local/zabbix/alertscripts/imges/' + pName
return pName
except Exception as e:
print e
return False
def text_transfe_html(text):
#将message转换为html
d=text.splitlines()
html_text=''
for i in d:
i='' + i + '</br>'
html_text+=i + '\n'
return html_text
def send_mail(to_email,subject,picture_name):
#发送邮件
graph_name=get_graph(itemid,picture_name)
html_text=text_transfe_html(sys.argv[3])
smtp_host = 'MAIL_SERVER'
from_email = 'FROM'
#邮箱账户
passwd = 'PASSWD'
#邮箱密码
msg=MIMEMultipart('related')
fp=open(graph_name,'rb')
image=MIMEImage(fp.read())
fp.close()
image.add_header('Content-ID','<image1>')
msg.attach(image)
html="""
<html>
<body>
"""
html+=html_text
html+='<img src="cid:image1"></br>'
html+="""
</body>
</html>
"""
html=MIMEText(html,'html','utf8')
msg.attach(html)
msg['Subject'] = subject
msg['From'] = from_email
smtp_server=smtplib.SMTP_SSL()
smtp_server.connect(smtp_host,'465')
smtp_server.login(from_email,passwd)
smtp_server.sendmail(from_email,to_email,msg.as_string())
smtp_server.quit()
if __name__ == '__main__':
time_tag=time.strftime("%Y%m%d%H%M%S", time.localtime())
to=sys.argv[1]
subject=sys.argv[2]
subject=subject.decode('utf-8')
itemid=get_itemid()
#graphid=get_graphid(itemid)
picture_name=time_tag + ".png"
graph_name=get_graph(itemid,picture_name)
send_mail(to,subject,picture_name)
执行结果图片:
参考连接：http://www.bubuko.com/infodetail-1678696.html
参考：微信公众号： 数睿技术