graylog 通过 python 实现钉钉 / 微信 / webhook 告警

> 本文由 [简悦 SimpRead](http://ksria.com/simpread/) 转码， 原文地址 [www.cnblogs.com](https://www.cnblogs.com/jonnyan/p/12869452.html)

> 由于不会编写graylog的插件,我们通过python脚本来间接实现钉钉告警.
> 我们可以在代码里实现我们想要的任何支持的告警方式,也可以在逻辑里增加对告警内容的二次处理再通知.

代码内容如下:
<details>
<summary>点击查看详细内容</summary>
<pre><code>
# coding:utf-8
# python3.6
# 只支持HTTP头包含Content-Length的请求
# @创建者：jonnyan404
# #日期：2020-05-09
# 博客地址：www.mrdoc.fun

import socket
import threading
import json
import my_logging
import logging
import requests

my_logging.load_my_logging_cfg()

HTTPHEADER = b"""HTTP/1.1 200 OK
Content-Type: text/html;charset=utf-8
Connection:Keep-Alive

"""

def resolve_http_header(header):
    header = header.decode('utf-8')
    headers = {}
    for i in header.split('\r\n'):
        if ':' in i:
            k = i.split(':')[0].strip()
            v = i.split(':')[1].strip()
            headers[k] = v
    return headers

def dingding(text):
    headers = {'Content-Type': 'application/json;charset=utf-8'}
    api_url = "此处更改为你自己的钉钉api url"
    json_text= {
     "msgtype": "text",
        "at": {
            "atMobiles": [
                "你的手机号码"
            ],
            "isAtAll": False
        },
        "text": {
            "content": text
        }
    }
    requests.post(api_url,json.dumps(json_text),headers=headers).content

class React(threading.Thread):
    def __init__(self, conn, addr):
        threading.Thread.__init__(self)
        self.setDaemon(True)
        self.conn = conn
        self.addr = addr

def read_chunk(self):
        data = b''
        # 第一次接收
        try:
            self.conn.settimeout(5)
            chunk = self.conn.recv(1024)
        except Exception as e:
            logging.error(e)
            return b'', b''

header_len = chunk.index(b'\r\n\r\n') + len(b'\r\n\r\n')

# 获取请求头并转换成字典格式
        header = chunk[:header_len]
        headers = resolve_http_header(header)

# 将第一次请求获取到的数据部分保存到data
        data += chunk[header_len:]
        data = data.strip() if data else b''

# print(header)
        # print()
        # print(data)

# 在while循环中接收剩余数据，当data长度与header中的Content-Length相同时结束循环
        try:
            print('Contetn-Length:', headers['Content-Length'])
            logging.info('Contetn-Length:', headers['Content-Length'])
        except:
            print('http header error.')
            logging.error('http header error.')
            return b'', b''

while not int(headers['Content-Length']) == len(data):
            try:
                chunk = self.conn.recv(4096)
                if chunk:
                    data += chunk
                    logging.info('read %d bytes.' % len(data))
                # 如果客户端没有主动关闭连接则不再接收
                else:
                    break
            except Exception as e:
                logging.error(e)
                break
        logging.info('done')

# 数据接收完成，响应客户端请求，并关闭连接
        self.conn.send(HTTPHEADER + b'ok ')
        self.conn.close()

return header, data

def run(self):
        header, payload = self.read_chunk()
        if not header or not payload:
            return
        # 保存接收到的所有数据
        logging.info(payload)
        data=json.loads(payload.decode("utf-8"))
        msg=data['event']['message']
        logging.info(msg)
        dingding(msg)

def server():
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    address = ('0.0.0.0', 8000)
    s.bind(address)

s.listen(5)

logging.info('listening on port %d ...' % 8000)

while True:
        conn, addr = s.accept()
        React(conn, addr).start()

if __name__ == '__main__':
    server()

</code>  </pre>
</details>

---

> 注意:代码里的 my_logging 文件为我自己的日志配置文件,请注意自行更换或去除.

- 运行上方的代码
- 在告警配置处,填写http://ip:port

# flask版本

```
# coding:utf-8
# python3.7
# @创建者：jonnyan404
# #日期：2022-12-02
# 博客地址：www.mrdoc.fun

import json
import requests
from flask import Flask, request
from gevent.pywsgi import WSGIServer

def dingding(text):
    headers = {'Content-Type': 'application/json;charset=utf-8'}
    api_url = "此处更改为你自己的钉钉api url"
    json_text= {
     "msgtype": "text",
        "at": {
            "atMobiles": [
                "你的手机号码"
            ],
            "isAtAll": False
        },
        "text": {
            "content": text
        }
    }
    requests.post(api_url,json.dumps(json_text),headers=headers).content

app = Flask(__name__)

@app.route('/webhook', methods=['POST'])
def webhook():
    prometheus_data = json.loads(request.data)
    print(prometheus_data)
	msg=prometheus_data['event']['message']
	dingding(msg)
    return "ok"

if __name__ == '__main__':
    WSGIServer(('0.0.0.0', 5000), app).serve_forever()

```