Compare commits

...

3 Commits

Author SHA1 Message Date
98f7cd6d17 Update nix build 2026-03-25 08:15:52 +01:00
b88fcb2f04 Cache last value for each topic and only insert into DB on change
Reduces database size
2020-12-21 22:22:27 +01:00
adf4e564d9 Add translation table to convert yes/no/on/off strings to numbers 2020-12-13 19:11:28 +01:00
2 changed files with 31 additions and 8 deletions

View File

@@ -1,10 +1,12 @@
{ python3 }:
with python3.pkgs;
buildPythonPackage rec {
name = "mqtt-config";
name = "mqtt-tools";
src = ./.;
propagatedBuildInputs = [ pyyaml paho-mqtt ];
buildInputs = [];
buildInputs = [ ];
doCheck = false;
shellHook = "";
pyproject = true;
build-system = [ setuptools ];
}

View File

@@ -7,6 +7,7 @@ import paho.mqtt.client
import sqlite3
import datetime
MAX_VALUE_CACHE_SIZE = 1024
def init_db(path, topics):
db = sqlite3.connect(path)
@@ -31,17 +32,37 @@ def main():
def on_disconnect(client, userdata, rc):
client.reconnect()
current_values = {}
def on_message(client, userdata, msg):
try:
c = db.cursor()
table = msg.topic.replace("/", "_")
ts = datetime.datetime.now()
textual = {
b"on": 1.,
b"off": 0.,
b"yes": 1.,
b"no": 0.
}
if msg.payload.lower() in textual:
value = textual[msg.payload.lower()]
else:
value = float(msg.payload)
c.execute(f'CREATE TABLE IF NOT EXISTS {table} '
print(f"{table}: {value} ")
if current_values.get(table, None) != value:
current_values[table] = value
if len(current_values) > MAX_VALUE_CACHE_SIZE:
first_entry = current_values.keys().next()
del current_values[first_entry]
c.execute(f'CREATE TABLE IF NOT EXISTS "{table}" '
'(timestamp timestamp, value real)')
c.execute(f'INSERT INTO {table} VALUES (?, ?)', (ts, value))
c.execute(f'SELECT * from {table}')
c.execute(f'INSERT INTO "{table}" VALUES (?, ?)', (ts, value))
db.commit()
except Exception as e:
print(f'Error writing to database: {e}')