initial commit

statuspage/__init__.py

@@ -0,0 +1,139 @@
+#!/usr/bin/env python3
+
+import json
+import logging
+from datetime import datetime, timedelta
+from os import environ
+from re import sub
+
+from flask import Flask, abort, jsonify, render_template, request
+from psycopg2.pool import ThreadedConnectionPool
+
+app = Flask(__name__)
+app.config.from_file(environ['APP_CONFIG'], json.load)
+
+if not app.debug:
+    log = logging.StreamHandler()
+    log.setLevel(logging.INFO)
+    app.logger.addHandler(log)
+
+
+pg_pool = ThreadedConnectionPool(
+    minconn=1,
+    maxconn=20,
+    user=app.config['DB_USER'],
+    password=app.config['DB_PASS'],
+    database=app.config['DB_NAME'],
+    host='localhost',
+)
+
+
+def check_freshness(cur):
+    cur.execute('select status_update_time from icinga_programstatus;')
+    if datetime.utcnow() - cur.fetchone()[0] > timedelta(minutes=5):
+        abort(503)
+
+
+def get_nodename(identifier):
+    if identifier in HOSTS:
+        return HOSTS[identifier]
+    abort(404)
+
+
+@app.route('/status.json')
+def services_as_json():
+    conn = pg_pool.getconn()
+    results = []
+
+    try:
+        cur = conn.cursor()
+
+        check_freshness(cur)
+
+        cur.execute(
+            '''
+            select
+                objs.name1,
+                objs.name2,
+                icinga_servicestatus.current_state,
+                icinga_servicestatus.scheduled_downtime_depth,
+                icinga_servicestatus.state_type,
+                icinga_servicestatus.output,
+                icinga_servicestatus.problem_has_been_acknowledged,
+                icinga_servicestatus.status_update_time,
+                icinga_servicestatus.last_state_change,
+                (
+                    select vars.varvalue
+                    from icinga_customvariables vars, icinga_services
+                    where objs.object_id = icinga_services.service_object_id
+                    and icinga_services.host_object_id = vars.object_id
+                    and vars.varname = 'pretty_name'
+                )
+            from icinga_objects objs
+            left join icinga_servicestatus on objs.object_id = icinga_servicestatus.service_object_id
+            left join icinga_servicegroup_members sgrm on objs.object_id = sgrm.service_object_id
+            where
+                objs.objecttype_id = 2 and
+                objs.is_active = 1 and
+                sgrm.servicegroup_id = %s
+            ;''',
+            (app.config['SERVICEGROUP_ID'],),
+        )
+
+        for (
+            host_name,
+            service_name,
+            state,
+            downtime_depth,
+            state_type,
+            output,
+            acked,
+            update_time,
+            last_state_change,
+            pretty_name,
+        ) in cur.fetchall():
+            if last_state_change is None:
+                last_state_change = update_time
+
+            for regex, replacement in app.config.get('NAME_REPLACEMENTS', {}).items():
+                service_name = sub(regex, replacement, service_name)
+
+            results.append(
+                {
+                    'type': 'Service',
+                    'attrs': {
+                        'acknowledgement': acked,
+                        'display_name': service_name,
+                        'downtime_depth': downtime_depth,
+                        'host_name': pretty_name,
+                        'last_check': update_time.timestamp(),
+                        'last_state_change': last_state_change.timestamp(),
+                        'state': state,
+                        'state_type': state_type,
+                        'last_check_result': {
+                            'output': output,
+                        },
+                        '__custom': {
+                            'last_check': update_time.strftime('%Y-%m-%d %H:%M:%S UTC'),
+                            'last_state_change': last_state_change.strftime(
+                                '%Y-%m-%d %H:%M:%S UTC'
+                            ),
+                        },
+                    },
+                }
+            )
+
+        cur.close()
+    finally:
+        pg_pool.putconn(conn)
+
+    return jsonify({'results': results})
+
+
+@app.route('/')
+def statuspage():
+    return render_template('statuspage.html')
+
+
+if __name__ == '__main__':
+    app.run(host='::')