使用AlertManager为Zabbix做报警分组和沉默

cat  /usr/lib/zabbix/alertscripts/zabbix-to-am.py
#!/usr/bin/env python
# -*- coding: utf-8 -*-


import os
import sys
import json
import requests
import logging.handlers
from datetime import datetime
import time


def str_time(s):
    dt = datetime.strptime(s, '%Y.%m.%d %H:%M:%S')
    tm = time.mktime(dt.timetuple())
    ut = datetime.utcfromtimestamp(tm).isoformat() + 'Z'
    return ut


if len(sys.argv) < 2:
    print("Require Arguments 1 subject")
    sys.exit()

# 设置日志
log_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'logs')
log_file = os.path.join(log_path, os.path.basename(__file__).split('.')[0] + '.log')

if not os.path.exists(log_path):
    os.mkdir(log_path)

logger = logging.getLogger()
handler = logging.handlers.RotatingFileHandler(log_file, maxBytes=(100 * 1024 * 1024 * 8), backupCount=5,
                                               encoding='utf-8')
formatter = logging.Formatter('[%(asctime)s] %(levelname)s [%(funcName)s: %(filename)s, %(lineno)d] %(message)s',
                              "%Y-%m-%d %H:%M:%S")
handler.setFormatter(formatter)
logger.addHandler(handler)
logger.setLevel(logging.INFO)

subject = sys.argv[1]

'''
Zabbix 发送的消息体

{TRIGGER.SEVERITY}
{HOST.NAME}
{HOST.IP}
{TRIGGER.HOSTGROUP.NAME}
{ITEM.NAME}
{EVENT.DATE} {EVENT.TIME}
{TRIGGER.NAME}
{ITEM.NAME}:{ITEM.VALUE}
firing/resolved
{EVENT.RECOVERY.DATE} {EVENT.RECOVERY.TIME}

---
subject = """
Average\r
putao-zabbix\r
172.20.9.24\r
Zabbix servers\r
Agent ping\r
2019.07.22 09:43:30\r
Zabbix agent on zabbix is unreachable for 5 minutes\r
Agent ping:Up (1)\r
resolved\r
2019.07.22 10:44:07
"""
'''

logging.info('[start].....')
logging.info('[source]: ' + subject)

data = {"labels": {}}

# 解析zabbix发送的消息体
labels = subject.split("\r\n")
data["labels"]["severity"] = labels[0].strip()
data["labels"]["name"] = labels[1].strip()
data["labels"]["instance"] = labels[2].strip()
data["labels"]["group"] = labels[3].strip()
data["labels"]["alertname"] = labels[4].strip()
data["labels"]["service"] = "zabbix"

data["annotations"] = {}
data["annotations"]["summary"] = labels[6].strip()
data["annotations"]["description"] = labels[7].strip()
data['status'] = labels[8].strip()

data['startsAt'] = str_time(labels[5].strip())

if data['status'] == 'resolved':
    data['endsAt'] = str_time(labels[9].strip())

data['generatorURL'] = "http://172.20.9.24/zabbix.php?action=dashboard.view&ddreset=1"

post_data = json.dumps([data])

# 向alertmanager发送请求
am_url = "http://172.20.6.7:30080/api/v2/alerts"
headers = {"Content-Type": "application/json", "Host": "alertmanager.monitoring.k8s.local"}

logging.info('[DATA] %s' % post_data)

r = requests.post(am_url, data=post_data, headers=headers)

logging.info('[return] %s' % r)
logging.info('[return] %s' % r.text)
logging.info('[end].....')