Python + Zabbix 实现微信告警信息 + 性能监控图

8,578次阅读
没有评论

思路

1、通过zabbix传递给脚本的message参数,筛选出报警信息的itemid;

2、通过获取的itemid,在数据库中查找对应的grpahid;

3、拉取对应graphid的图片并保存;

4、使用微信API“上传临时素材”接口,将告警时保存的性能监控图上传,以获取“media_id”

5、使用获取到的“media_id为参数”,调用发送消息的API地址

代码实现

get_img

def get_itemid(self):
    # 获取itemid
    pattern = re.compile('ITEM.*?:(\d+)', re.S)
    itemid = re.findall(pattern, sys.argv[3])
    itemid = str(itemid).lstrip('[\'').rstrip('\']')
    print(itemid)
    return itemid


def get_graphid(self):
    itemid = self.get_itemid()
    conn = pymysql.connect(host=self.host, user=self.username, password=self.password, port=3306, db='zabbix')
    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))
    graphid = str(graphid).lstrip('[\'').rstrip('\']')
    print(graphid)
    return graphid


def get_img(self):
    graphid = self.get_graphid()
    # 拉取图片
    os.system(
        'curl -s -L -c /usr/lib/zabbix/alertscripts/cookie.txt --user-agent Mozilla/4.0 -d "reauest=&name=%s&password=%s&autologin=1&enter=Sign+in" %s' % (
            self.zabbix_user, self.zabbix_pwd, self.url))
    os.system(
        'curl -s -c /usr/lib/zabbix/alertscripts/cookie.txt -b /usr/lib/zabbix/alertscripts/cookie.txt --user-agent Mozilla/4.0 -F "graphid=%s" -F "period=%s" -F "width=900" %s > /usr/lib/zabbix/img/zabbix_%s.png' % (
            graphid, self.time, self.chart2_url, str(slerttime)))
    img_name = '/usr/lib/zabbix/img/' + 'zabbix_' + str(self.alerttime) + '.png'
    print(img_name)
    return img_name

gettoken

def gettoken(corpid, corpsecret):
    gettoken_url = 'https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=' + corpid + '&corpsecret=' + corpsecret
    print(gettoken_url)
    try:
        token_file = urllib.request.urlopen(gettoken_url)
    except urllib.request.HTTPError as e:
        print(e.code)
        print(e.read().decode("utf8"))
        sys.exit()
    token_data = token_file.read().decode('utf-8')
    token_json = json.loads(token_data)
    token_json.keys()
    token = token_json['access_token']
    return token

senddata

def senddata(self, access_token, user, subject, content):
    send_url = 'https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=' + access_token
    send_values = {
        "touser": "wangxiao",
        "msgtype": "textcard",
        "agentid": "1000002",
        "textcard": {
            "title": "监控告警",
            "description": subject + '\n' + content,
            "url": "URL",
            "btntxt": "更多"
        },
        "safe": "0"
    }
    send_data = simplejson.dumps(send_values, ensure_ascii=False).encode('utf-8')
    # print(send_data)
    send_request = urllib.request.Request(send_url, send_data)
    # print(send_request)
    response = json.loads(urllib.request.urlopen(send_request).read())
    print(str(response))

 

sendimg

def sendimg(access_token, user, subject, content):
    send_url = 'https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=' + access_token
    tm = datetime.datetime.now()
    time = tm.strftime("%Y-%m-%d-%H%M")
    savedir = '/data/img'
    imgname = str(savedir) + '/' + 'zabbix_' + str(time) + '.jpg'
    url = 'https://qyapi.weixin.qq.com/cgi-bin/media/upload?access_token=' + access_token + '&type=image'
    files = {'file': open(imgname, 'rb')}
    response = requests.post(url, files=files)
    pattern = re.compile('media_id":"(.*?)",', re.S)
    mediaid = re.findall(pattern, response.text)[-1]
    send_values = {
            "touser": "wangxiao",
            "msgtype": "image",
            "agentid": "1000002",
            "image": {
                "media_id": "%s" % mediaid
            }
    }
    send_data = simplejson.dumps(send_values, ensure_ascii=False).encode('utf-8')
    print(send_data)
    send_request = urllib.request.Request(send_url,send_data)
    print(send_request)
    response = json.loads(urllib.request.urlopen(send_request).read())
    print(str(response))

全部代码

# -*- coding:utf-8 -*-
# !/usr/local/python3/bin/python3
import json
import urllib.request
import simplejson
import re, sys, os, requests, datetime, time
import pymysql


class zabbix_img(object):
    def __init__(self, url, host, username, password, time):
        # 指定zabbix获取图片的地址
        self.chart2_url = 'http://172.20.2.11/zabbix/chart2.php'
        # zabbix登录账户
        self.zabbix_user = 'Admin'
        # zabbix登录密码
        self.zabbix_pwd = 'zabbix'
        self.url = url
        self.host = host
        self.username = username
        self.password = password
        self.time = time
        self.alerttime = datetime.datetime.now().strftime("%Y-%m-%d-%H%M%S")

    def get_itemid(self):
        # 获取itemid
        pattern = re.compile('ITEM.*?(\d+)', re.S)
        itemid = re.findall(pattern, sys.argv[3])
        itemid = str(itemid).lstrip('[\'').rstrip('\']')
        print(itemid)
        return itemid

    def get_graphid(self):
        itemid = self.get_itemid()
        conn = pymysql.connect(host=self.host, user=self.username, password=self.password, port=3306, db='zabbix')
        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))
        graphid = str(graphid).lstrip('[\'').rstrip('\']')
        print(graphid)
        return graphid

    def get_img(self):
        graphid = self.get_graphid()
        # 拉取图片
        os.system(
            'curl -s -L -c /usr/lib/zabbix/alertscripts/cookie.txt --user-agent Mozilla/4.0 -d "reauest=&name=%s&password=%s&autologin=1&enter=Sign+in" %s' % (
            self.zabbix_user, self.zabbix_pwd, self.url))
        os.system(
            'curl -s -c /usr/lib/zabbix/alertscripts/cookie.txt -b /usr/lib/zabbix/alertscripts/cookie.txt --user-agent Mozilla/4.0 -F "graphid=%s" -F "period=%s" -F "width=900" %s > /usr/lib/zabbix/img/zabbix_%s.png' % (
            graphid, self.time, self.chart2_url, str(slerttime)))
        img_name = '/usr/lib/zabbix/img/' + 'zabbix_' + str(self.alerttime) + '.png'
        print(img_name)
        return img_name

    def gettoken(self, corpid, corpsecret):
        gettoken_url = 'https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=' + corpid + '&corpsecret=' + corpsecret
        print(gettoken_url)
        try:
            token_file = urllib.request.urlopen(gettoken_url)
        except urllib.request.HTTPError as e:
            print(e.code)
            print(e.read().decode("utf8"))
            sys.exit()
        token_data = token_file.read().decode('utf-8')
        token_json = json.loads(token_data)
        token_json.keys()
        token = token_json['access_token']
        return token

    def senddata(self, access_token, user, subject, content):
        send_url = 'https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=' + access_token
        send_values = {
            "touser": "wangxiao",
            "msgtype": "textcard",
            "agentid": "1000002",
            "textcard": {
                "title": "监控告警",
                "description": subject + '\n' + content,
                "url": "URL",
                "btntxt": "更多"
            },
            "safe": "0"
        }
        send_data = simplejson.dumps(send_values, ensure_ascii=False).encode('utf-8')
        # print(send_data)
        send_request = urllib.request.Request(send_url, send_data)
        # print(send_request)
        response = json.loads(urllib.request.urlopen(send_request).read())
        print(str(response))

    def sendimg(self, access_token, user, subject, content):
        send_url = 'https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=' + access_token
        imgname = '/usr/lib/zabbix/img/' + 'zabbix_' + str(self.alerttime) + '.png'
        url = 'https://qyapi.weixin.qq.com/cgi-bin/media/upload?access_token=' + access_token + '&type=image'
        files = {'file': open(imgname, 'rb')}
        response = requests.post(url, files=files)
        pattern = re.compile('media_id":"(.*?)",', re.S)
        mediaid = re.findall(pattern, response.text)[-1]
        send_values = {
            "touser": "wangxiao",
            "msgtype": "image",
            "agentid": "1000002",
            "image": {
                "media_id": "%s" % mediaid
            }
        }
        send_data = simplejson.dumps(send_values, ensure_ascii=False).encode('utf-8')
        print(send_data)
        send_request = urllib.request.Request(send_url, send_data)
        print(send_request)
        response = json.loads(urllib.request.urlopen(send_request).read())
        print(str(response))


if __name__ == '__main__':
    url = 'http://172.20.2.11/zabbix/index.php'
    host = '172.20.2.11'
    username = 'zabbix'
    password = 'zabbix'
    time = '600'
    zabbixOper = zabbix_img(url, host, username, password, time)
    zabbixOper.get_img()
    user = sys.argv[1]
    subject = sys.argv[2]
    content = sys.argv[3]
    corpid = 'wxf7141'
    corpsecret = 'U0DaihOsLvSc'
    accesstoken = zabbixOper.gettoken(corpid, corpsecret)
    zabbixOper.senddata(accesstoken, user, subject, content)
    zabbixOper.sendimg(accesstoken, user, subject, content)

注意事项:

1、务必将脚本放至zabbix执行脚本的目录

2、注意相关文件及目录的权限(cookie文件cookie.txt,脚本文件wechat.py,图片存储目录img)

运行结果

禁ping模拟告警

Python

接受告警

Python

Python

Honest1y
版权声明:本站原创文章,由Honest1y于2018年08月26日发表,共计10838字。
转载提示:除特殊说明外本站文章皆由CC-4.0协议发布,转载请注明出处。
评论(没有评论)