From 08857b6bc6cb94a996b54ba376b8d3946de9ed28 Mon Sep 17 00:00:00 2001
From: vuko <vuko@hackerspace.pl>
Date: Fri, 12 Oct 2018 16:26:09 +0200
Subject: [PATCH] adding files

---
 init.sh   |  5 ++++
 meters.py | 86 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 run.sh    |  4 +++
 3 files changed, 95 insertions(+)
 create mode 100755 init.sh
 create mode 100644 meters.py
 create mode 100755 run.sh

diff --git a/init.sh b/init.sh
new file mode 100755
index 0000000..54f777c
--- /dev/null
+++ b/init.sh
@@ -0,0 +1,5 @@
+#!/bin/bash
+cd $(dirname "$0")
+python3 -m venv ./venv
+. ./venv/bin/activate
+pip3 install flask paho-mqtt
diff --git a/meters.py b/meters.py
new file mode 100644
index 0000000..b16d3c6
--- /dev/null
+++ b/meters.py
@@ -0,0 +1,86 @@
+import paho.mqtt.client as mqtt
+import threading
+import re
+import datetime
+import copy
+
+con = mqtt.Client()
+con.connect('10.8.1.16')
+
+lock = threading.Lock()
+measurements = {}
+
+con.subscribe('iot/+/+/energy')
+
+class Measurements:
+    class Meter:
+        def __init__(self, _id):
+            self.id = _id
+            self.energy = []
+
+        def add_energy_measurement(energy):
+            self.energy.append((datetime.datetime.now(), energy))
+            self.energy = self.energy[-10:]
+
+        def get_metrics():
+            if len(self.energy) > 0:
+                last = self.energy[-1]
+                s = ('energy{{node="esp8266-{mid:s}"}}  {energy:f} '
+                    '{epoch:d}\n').format(mid=self.id, energy=last[1],
+                    epoch=int(last[0].timestamp() * 1000))
+                return s
+            else:
+                return ''
+    def __init__(self):
+        self._meters = []
+        self._lock = threading.Lock()
+
+    def _get_meter_by_id(self, _id):
+        if _id not in self._meters:
+            self._meters[_id] = Meter(_id)
+        return self._meters[_id]
+
+    def add_energy_measurement(meter_id, energy):
+        with self._lock:
+            meter = self._get_meter_by_id(meter_id)
+            meter.energy.append((datetime.datetime.now(), energy))
+
+def on_message(client, userdata, message):
+    m = re.match('iot/(?P<id>[^/]+)/[^/]+/energy', message.topic)
+    if m is not None:
+        try:
+            energy = int(re.match('^(?P<joules>[0-9]*) J', message.payload.decode('ASCII')).group('joules'))
+        except AttributeError:
+            pass
+        else: 
+            if message.retain == 0:
+                mid = m.group('id')
+                with lock:
+                    if mid not in measurements:
+                        measurements[mid] = []
+                    ms = measurements[mid]
+                    ms.append((datetime.datetime.now(), energy))
+                    measurements[mid] = ms[-10:]
+    
+con.on_message = on_message
+con.loop_start()
+
+import flask
+app = flask.Flask(__name__)
+
+@app.route('/metrics')
+def metrics():
+    r = ""
+    r += "# HELP energy energy in joules\n"
+    r += "# TYPE energy counter\n"
+    with lock:
+        m = copy.deepcopy(measurements)
+    for mid in m:
+        ms = [m[mid][-1]]
+        for s in ms:
+            r += 'energy{{node="esp8266-{mid:s}"}} {energy:f} {epoch:d}\n'.format(mid=mid, energy=s[1], epoch=int(s[0].timestamp() * 1000) )
+    return flask.Response(r, mimetype='text/plain; version=0.0.4')
+
+app.run('0.0.0.0')
+
+
diff --git a/run.sh b/run.sh
new file mode 100755
index 0000000..2c60985
--- /dev/null
+++ b/run.sh
@@ -0,0 +1,4 @@
+#!/bin/bash
+cd $(dirname "$0")
+. ./venv/bin/activate
+python3 meters.py