【Zabbix】5.0版本釘釘告警Python腳本及配置

Zabbix 釘釘群機器人發送信息

這里利用的是釘釘群機器人，將zabbix告警恢復信息發送至釘釘群，並@相應的管理人員。

1. 在釘釘群中創建機器人

• 安全模式選擇加簽，並復制一個 67位字符串
• 記錄webhook地址
  獲取到的這兩串字符配置到后面的zabbix_ding.conf文件中

2. Zabbix 服務器端配置

2.1 安裝Python3、相關包以及創建釘釘發送消息日志文件

# 安裝python3，腳本是基於Python3寫的
yum install -y python3

# 安裝pip，pip是python的包管理器
# 下載pip安裝腳本
curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
# 運行安裝腳本
sudo python3 get-pip.py
# 安裝puthon包
pip3 install configparser
pip3 install requests
# 創建釘釘發送日志文件路徑
touch /var/log/zabbix/zabbix_ding.log
chown zabbix.zabbix /var/log/zabbix/zabbix_ding.log

2.2 將腳本及配置文件上傳至服務器zabbix腳本目錄

1. 修改以下配置文件zabbix_ding.conf，並將文件上傳至/etc/zabbix/下

[config]
# 日志文件
log_path=/var/log/zabbix/zabbix_ding.log
#釘釘機器人 webhook 值
webhook=https://oapi.dingtalk.com/robot/send?access_token=b3e6fa0f410e7ced04c81680f036xxxx
# 安全設置 -- 加簽
secret=SEC64d20b4e9d2e2677f7aa01d2a7c2f9xxxx

2. 自定義修改，並復制腳本內容，上傳至AlertScriptsPath路徑下

# 查看zabbix的腳本默認路徑，請注意使用您自己的zabbix_server.conf
cat /etc/zabbix/zabbix_server.conf |grep AlertScriptsPath
### Option: AlertScriptsPath
# AlertScriptsPath=${datadir}/zabbix/alertscripts
AlertScriptsPath=/usr/lib/zabbix/alertscripts

#!/usr/bin/env python3
# coding:utf8
#
import configparser
import os
import time
import hmac
import hashlib
import base64
import urllib.parse
import requests
import json
import sys

config = configparser.ConfigParser()
config.read('/etc/zabbix/zabbix_ding.conf', encoding='utf-8')
log_path = config.get('config', 'log_path')
api_url = config.get('config', 'webhook')
api_secret = config.get('config', 'secret')
log_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())


# 釘釘機器人文檔說明
# https://ding-doc.dingtalk.com/doc#/serverapi2/qf2nxq
def get_timestamp_sign():
    timestamp = str(round(time.time() * 1000))
    secret = api_secret
    secret_enc = secret.encode('utf-8')
    string_to_sign = '{}\n{}'.format(timestamp, secret)
    string_to_sign_enc = string_to_sign.encode('utf-8')
    hmac_code = hmac.new(secret_enc, string_to_sign_enc, digestmod=hashlib.sha256).digest()
    sign = urllib.parse.quote_plus(base64.b64encode(hmac_code))
    return timestamp, sign

# 獲取加簽后的鏈接
def get_signed_url():
    timestamp, sign = get_timestamp_sign()
    webhook = api_url + "&timestamp=" + timestamp + "&sign=" + sign
    return webhook

# 定義消息模式
def get_webhook(mode):
    if mode == 0:  # only 關鍵字
        webhook = api_url
    elif mode == 1 or mode == 2:  # 關鍵字和加簽 或 # 關鍵字+加簽+ip
        webhook = get_signed_url()
    else:
        webhook = ""
        print("error! mode:   ", mode, "  webhook :  ", webhook)
    return webhook


def get_message(text, user_info):
    # 和類型相對應，具體可以看文檔 ：https://ding-doc.dingtalk.com/doc#/serverapi2/qf2nxq
    # 可以設置某個人的手機號，指定對象發送
    message = {
        "msgtype": "text",  # 有text, "markdown"、link、整體跳轉ActionCard 、獨立跳轉ActionCard、FeedCard類型等
        "text": {
            "content": text  # 消息內容
        },
        "at": {
            "atMobiles": [
                user_info,
            ],
            "isAtAll": False  # 是否是發送群中全體成員
        }
    }
    return message


# 消息發送日志
def log(info):
    if os.path.exists(log_path):
        log_file = open(log_path, "a+")
    else:
        log_file = open(log_path, "w+")
    log_file.write(info)


def send_ding_message(text, user_info):
    # 請求的URL，WebHook地址
    # 主要模式有 0 ： 關鍵字 1：# 關鍵字 +加簽 3：關鍵字+加簽+IP
    webhook = get_webhook(1)
    # 構建請求頭部
    header = {
        "Content-Type": "application/json",
        "Charset": "UTF-8"
    }
    # 構建請求數據
    message = get_message(text, user_info)
    # 對請求的數據進行json封裝
    message_json = json.dumps(message)
    # 發送請求
    info = requests.post(url=webhook, data=message_json, headers=header).json()
    code = info["errcode"]
    errmsg = info["errmsg"]
    if code == 0:
        log(log_time + ":消息已發送成功 返回信息:%s %s\n" % (code, errmsg))
    else:
        log(log_time + ":消息發送失敗 返回信息:%s %s\n" % (code, errmsg))
        print(log_time + ":消息發送失敗 返回信息:%s %s\n" % (code, errmsg))
        exit(3)


if __name__ == "__main__":
    text = sys.argv[3]
    user_info = sys.argv[1]
    send_ding_message(text, user_info)

# 修改腳本權限
chown zabbix.zabbix /lib/zabbix/alertscripts/zabbix_ding.py
chmod u+x /lib/zabbix/alertscripts/zabbix_ding.py

# 測試腳本內容
python3 zabbix_ding.py  user subject mesages

3. Web端配置

3.1 添加 報警媒體類型

按如下圖中信息填寫

{ALERT.SENDTO}

{ALERT.SUBJECT}

{ALERT.MESSAGE}

3.2 在用戶中配置報警媒介

這里以管理員用戶為例

3.3 配置動作

告警主機：{HOST.NAME}
告警  IP：{HOST.IP}
告警時間：{EVENT.DATE}-{EVENT.TIME}
告警等級：{TRIGGER.SEVERITY}
告警信息：{TRIGGER.NAME}:{ITEM.VALUE}
事件  ID：{EVENT.ID}

3.4 測試

我這里關閉一台主機

查看釘釘消息日志