From 69e11599c2194187e1c172ebe6b7b51945a7f306 Mon Sep 17 00:00:00 2001
From: pks <pks@pks.rocks>
Date: Sun, 2 May 2021 21:56:23 +0200
Subject: v2

---
 weather-logger-receiver | 112 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 112 insertions(+)
 create mode 100755 weather-logger-receiver

(limited to 'weather-logger-receiver')

diff --git a/weather-logger-receiver b/weather-logger-receiver
new file mode 100755
index 0000000..389a9de
--- /dev/null
+++ b/weather-logger-receiver
@@ -0,0 +1,112 @@
+#!/usr/bin/env python3
+
+import logging
+import os
+import signal
+import sqlite3
+import sys
+import time
+import yaml
+
+from RFXtrx import PySerialTransport
+
+
+def setup_database(path="weather.db"):
+    db_connection = sqlite3.connect(path)
+    db_cursor = db_connection.cursor()
+    try:
+        db_cursor.execute("CREATE TABLE weather( \
+                                          id INTEGER PRIMARY KEY, \
+                                          timestamp DATETIME, \
+                                          handle TEXT, \
+                                          temperature FLOAT, \
+                                          humidity FLOAT);")
+    except sqlite3.OperationalError:
+        pass
+
+    return db_connection
+
+
+def setup_serial(device="/dev/ttyUSB0"):
+    serial_connection = PySerialTransport(device)
+    serial_connection.reset()
+
+    return serial_connection
+
+
+def load_known_devices(location, path="known-devices.yaml"):
+    known_devices = None
+    with open(path, "r") as f:
+        known_devices = yaml.safe_load(f.read())
+
+    return known_devices[location]
+
+
+def add_data_to_db(data, handle, db_connection):
+    db_cursor = db_connection.cursor()
+    db_cursor.execute(f"INSERT INTO weather \
+                        (timestamp, handle, temperature, humidity) \
+                        VALUES (  {int(time.time())}, \
+                                \"{handle}\", \
+                                  {data.values['Temperature']}, \
+                                  {data.values['Humidity']})")
+    db_connection.commit()
+
+
+def loop(serial_connection, db_connection, logger):
+    while True:
+        logger.info("(Re-)Starting receiver loop")
+
+        data = serial_connection.receive_blocking()
+        
+        if data is None:
+            logger.info("Received no data")
+            continue
+
+        logger.info(data)
+
+        try:
+            if data.device.id_string not in known_devices:
+                logger.info(f"Unknown device with id {device_id} and type {device_type}, ignoring")
+                continue
+        except:
+            logger.info("Device ID not found, likely no SensorEvent, ignoring")
+            continue
+
+        handle = known_devices[data.device.id_string]["handle"]
+        
+        add_data_to_db(data, handle, db_connection)
+
+
+if __name__ == "__main__":
+    logger = logging.getLogger("weather-logger-receiver")
+    logging.basicConfig(format='%(asctime)-2s %(levelname)-8s %(message)s',
+                        datefmt='%Y-%m-%dT%H:%M:%S',
+                        level=logging.INFO)
+
+    assert len(sys.argv) == 2
+    location = sys.argv[1]
+    known_devices = load_known_devices(location=location)
+    logger.info(f"Loaded known devices for location {location}")
+
+    db_connection = setup_database()
+    logger.info("Set up database")
+
+    serial_connection = setup_serial()
+    logger.info("Set up serial connection")
+
+    def shutdown(*args):
+        logger.info("Shutting down")
+        serial_connection.close()
+        db_connection.close()
+        sys.exit(0)
+
+    signal.signal(signal.SIGINT, shutdown)
+    signal.signal(signal.SIGTERM, shutdown)
+
+    try:
+        loop(serial_connection, db_connection, logger)
+    except Exception as e:
+        logger.error(f"Exception: {e}")
+    finally:
+        shutdown(serial_connection, db_connection, logger)
-- 
cgit v1.2.3