From b78f2870649fe7d6776f1177dfced7078cd1064e Mon Sep 17 00:00:00 2001
From: Thomas Jensen <thomas@uctrl.net>
Date: Fri, 19 Jul 2019 01:03:22 +0200
Subject: [PATCH] initial commit

---
 .gitignore                   |  1 +
 alert-api.py                 | 92 ++++++++++++++++++++++++++++++++++++
 alertmanager/__init__.py     |  1 +
 alertmanager/alertmanager.py | 35 ++++++++++++++
 4 files changed, 129 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 alert-api.py
 create mode 100644 alertmanager/__init__.py
 create mode 100644 alertmanager/alertmanager.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..61f2dc9
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+**/__pycache__/
diff --git a/alert-api.py b/alert-api.py
new file mode 100644
index 0000000..c26f767
--- /dev/null
+++ b/alert-api.py
@@ -0,0 +1,92 @@
+import RPi.GPIO as GPIO
+from collections import Counter, defaultdict
+from alertmanager import Alertmanager
+import time, json
+
+GPIO.setmode(GPIO.BCM)   # set board mode to Broadcom
+GPIO.setwarnings(False)  # don't show warnings
+
+al = Alertmanager("http://prometheus.lan.uctrl.net:9093")
+
+colors = {
+        'red': 1,
+        'orange': 2,
+        'green': 3,
+        'blue': 4,
+        'clear': 5
+        }
+
+types = {
+        'alert': {
+            'color': 'red',
+            'prev': 0
+            },
+        'warn': {
+            'color': 'orange',
+            'prev': 0
+            }
+        }
+
+boot_seq = ['clear', 'blue', 'green', 'orange', 'red']
+
+for color, io in colors.items():
+    GPIO.setup(io, GPIO.OUT)
+
+
+def setColor(color, state):
+    io = colors[color]
+
+    if GPIO.input(io) is not int(state):
+        print('Setting color', color.upper(), str(state))
+        GPIO.output(io, state)
+
+
+def bootSequence():
+    for color in boot_seq:
+        setColor(color, True)
+        time.sleep(0.2)
+        setColor(color, False)
+
+
+bootSequence()
+
+while True:
+    d = defaultdict(list)
+    start = time.time()
+
+    for alert in al.alerts():
+        if alert["labels"]["alertname"] == "DeadMansSwitch":
+            continue
+
+        for t in types:
+            if alert["labels"]["severity"] == t:
+                d[t].append(alert["status"]["state"])
+        
+    for t in types:
+        print(t, json.dumps(Counter(d[t]), indent=4, sort_keys=True))
+
+    for x in range(0, 10):
+        for t, v in types.items():
+            if Counter(d[t])['active'] > 0:
+                setColor(v['color'], True)
+
+        time.sleep(.5)
+
+        for t, v in types.items():
+            if v['prev'] < Counter(d[t])['active']:
+                setColor(v['color'], False)
+
+        time.sleep(.5)
+
+    setColor('green', False)
+
+    for t, v in types.items():
+        if Counter(d[t])['active'] == 0:
+            setColor(v['color'], False)
+        if v['prev'] > Counter(d[t])['active']:
+            setColor('green', True)
+
+        v['prev'] = Counter(d[t])['active']
+
+    print(time.time() - start)
+    print('---')
diff --git a/alertmanager/__init__.py b/alertmanager/__init__.py
new file mode 100644
index 0000000..9a2ffd8
--- /dev/null
+++ b/alertmanager/__init__.py
@@ -0,0 +1 @@
+from .alertmanager import *
diff --git a/alertmanager/alertmanager.py b/alertmanager/alertmanager.py
new file mode 100644
index 0000000..2b12a18
--- /dev/null
+++ b/alertmanager/alertmanager.py
@@ -0,0 +1,35 @@
+import urllib.request, urllib.parse, json
+
+
+class Alertmanager:
+    def __init__(self, host):
+        self.host = host
+
+
+    def alerts(self):
+        with urllib.request.urlopen(self.host + "/api/v2/alerts") as url:
+            data = json.loads(url.read().decode())
+
+            return data
+
+    def fire(self):
+        url = self.host + "/api/v2/alerts"
+        data_fields = [{
+            "startsAt": "2019-07-15T22:07:01.535Z",
+            #"endsAt": "2019-07-15T22:07:01.535Z",
+            "annotations": {
+                "summary": "string",
+                },
+            "labels": {
+                'alertname': 'string',
+                'instance': 'string',
+                'job': 'string',
+                'severity': 'string'
+                },
+            "generatorURL": "http://string"
+            }]
+
+        request = urllib.request.Request(url, data=json.dumps(data_fields).encode('utf8'), method='POST', headers={"Content-Type": "application/json", "accept": "application/json"})
+        data = urllib.request.urlopen(request).getcode()
+
+        return data