msv-clubhouse-node/backend/msv_clubhouse_backend.py

159 lines
5.5 KiB
Python

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
""" Author: Hendrik Schutter, mail@hendrikschutter.com
"""
from http.server import BaseHTTPRequestHandler, HTTPServer
import paho.mqtt.client as mqtt
from datetime import datetime
import threading
import time
import json
import config
scrape_healthy = True
startTime = datetime.now()
node_metrics = list()
mutex = threading.Lock()
request_count = 0
class RequestHandler(BaseHTTPRequestHandler):
def get_metrics(self):
global request_count
global node_metrics
global mutex
mutex.acquire()
self.send_response(200)
self.send_header("Content-type", "text/html")
self.end_headers()
self.wfile.write(bytes(config.exporter_prefix + "expoter_duration_seconds_sum " + str(int((datetime.now() - startTime).total_seconds())) + "\n", "utf-8"))
self.wfile.write(bytes(config.exporter_prefix + "exporter_request_count " + str(request_count) + "\n", "utf-8"))
self.wfile.write(bytes(config.exporter_prefix + "exporter_scrape_healthy " + str(int(scrape_healthy)) + "\n", "utf-8"))
for metric in node_metrics:
#print(metric)
self.wfile.write(bytes(config.exporter_prefix + metric + "\n", "utf-8"))
mutex.release()
def do_GET(self):
global request_count
request_count = request_count + 1
print("Request: " + self.path)
if (self.path.startswith("/metrics")):
self.get_metrics()
else:
self.send_response(200)
self.send_header("Content-type", "text/html")
self.end_headers()
self.wfile.write(bytes("<html>", "utf-8"))
self.wfile.write(bytes("<head><title>MSV Clubhouse exporter</title></head>", "utf-8"))
self.wfile.write(bytes("<body>", "utf-8"))
self.wfile.write(bytes('<h1>msv-clubhouse exporter based on data from LoRaWAN TTN node.</h1>', "utf-8"))
self.wfile.write(bytes('<p><a href="/metrics">Metrics</a></p>', "utf-8"))
self.wfile.write(bytes("</body>", "utf-8"))
self.wfile.write(bytes("</html>", "utf-8"))
def update_metrics(payload, metadata):
#print("Payload: "+ str(payload))
#print("Metadata: "+ str(metadata))
global node_metrics
global mutex
global scrape_healthy
mutex.acquire()
scrape_healthy = True
node_metrics.clear()
if "degreesC" in payload:
print("set degree: " + str(float(payload["degreesC"])))
node_metrics.append("temperature " + str(float(payload["degreesC"])))
if "pressure" in payload:
print("set pressure: " + str(float(payload["pressure"])))
node_metrics.append("pressure " + str(float(payload["pressure"])))
if "winddirection" in payload:
print("set winddirection: " + str(float(payload["winddirection"])))
node_metrics.append("winddirection " + str(float(payload["winddirection"])))
if "windspeed" in payload:
print("set windspeed: " + str(float(payload["windspeed"])))
node_metrics.append("windspeed " + str(float(payload["windspeed"])))
if "dooropen" in payload:
print("set dooropen: " + str(bool(payload["dooropen"])))
node_metrics.append("dooropen " + str(int(payload["dooropen"])))
# if "gateway_id" in metadata[0]["gateway_ids"]:
# print("set gateway_id: " + str(metadata[0]["gateway_ids"]["gateway_id"]))
# node_metrics.append("gateway_id " + str(metadata[0]["gateway_ids"]["gateway_id"]))
if "rssi" in metadata[0]:
print("set rssi: " + str(int(metadata[0]["rssi"])))
node_metrics.append("rssi " + str(int(metadata[0]["rssi"])))
if "channel_rssi" in metadata[0]:
print("set channel_rssi: " + str(int(metadata[0]["channel_rssi"])))
node_metrics.append("channel_rssi " + str(int(metadata[0]["channel_rssi"])))
if "snr" in metadata[0]:
print("set snr: " + str(float(metadata[0]["snr"])))
node_metrics.append("snr " + str(float(metadata[0]["snr"])))
#scrape_healthy = False
mutex.release()
def on_connect(mqttc, obj, flags, rc):
print("\nConnected to MQTT: rc = " + str(rc))
def on_message(mqttc, obj, msg):
#print("\nMessage: " + msg.topic + " " + str(msg.qos))
parsedJSON = json.loads(msg.payload)
#print(json.dumps(parsedJSON, indent=4))
try:
uplink_message = parsedJSON["uplink_message"];
update_metrics(uplink_message["decoded_payload"], uplink_message["rx_metadata"])
except:
print("Unable to parse uplink")
def on_subscribe(mqttc, obj, mid, granted_qos):
print("\nSubscribed to MQTT: " + str(mid) + " " + str(granted_qos))
def poll_mqtt(mqttc):
while True:
mqttc.loop(10) # seconds timeout
def main():
print("starting ...")
mqttc = mqtt.Client()
mqttc.on_connect = on_connect
mqttc.on_subscribe = on_subscribe
mqttc.on_message = on_message
mqttc.username_pw_set(config.ttn_user, config.ttn_key)
mqttc.tls_set()
mqttc.connect(config.ttn_region.lower() + ".cloud.thethings.network", 8883, 60)
mqttc.subscribe("#", 0) # all device uplinks
# run mqtt in thread forever
poll_mqtt_thread = threading.Thread(target=poll_mqtt, args=((mqttc,)))
poll_mqtt_thread.start()
webServer = HTTPServer((config.hostName, config.serverPort), RequestHandler)
print("Server started http://%s:%s" % (config.hostName, config.serverPort))
try:
webServer.serve_forever()
except KeyboardInterrupt:
pass
webServer.server_close()
print("Server stopped.")
poll_mqtt_thread.join()
if __name__ == "__main__":
main()