Initial commit

2026-07-07 13:29:22 +08:00
commit 3fdd9ec927
154 changed files with 56566 additions and 0 deletions
@@ -0,0 +1,47 @@
+from flask import Flask, request, jsonify
+import pymysql
+
+app = Flask(__name__)
+
+def get_db():
+    return pymysql.connect(
+        host="127.0.0.1",
+        user="root",
+        password="hp93000",
+        database="alert_mail_stat",
+        charset="utf8mb4"
+    )
+
+@app.route("/alert", methods=["POST"])
+def alert_receive():
+    data = request.get_json()
+    alerts = data.get("alerts", [])
+    insert_num = 0
+    db = get_db()
+    cur = db.cursor()
+    for alert in alerts:
+        fp = alert["fingerprint"]
+        alert_name = alert["labels"].get("alertname", "")
+        instance = alert["labels"].get("instance", "")
+        severity = alert["labels"].get("severity", "warning")
+        status = alert["status"]
+        start_at = alert.get("startsAt", "")
+        end_at = alert.get("endsAt")
+        alert_type = 2 if status == "resolved" else 1
+
+        cur.execute("SELECT id FROM alert_log WHERE mail_uid=%s", (fp,))
+        if cur.fetchone():
+            continue
+        sql = """
+        INSERT INTO alert_log(mail_uid,alert_type,alert_name,instance,severity,starts_at,ends_at,content,receive_time)
+        VALUES(%s,%s,%s,%s,%s,%s,%s,%s,NOW())
+        """
+        cur.execute(sql, (fp, alert_type, alert_name, instance, severity, start_at, end_at, str(alert)))
+        insert_num += 1
+    db.commit()
+    cur.close()
+    db.close()
+    return jsonify({"code": 200, "insert": insert_num})
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=909, debug=False)