neohub should use queue as well

1 files changed, 8 insertions, 3 deletions

diff --git a/neohub2pgsql.py b/neohub2pgsql.py
index 646fa60..452f5a9 100755
--- a/neohub2pgsql.py
+++ b/neohub2pgsql.py
@@ -1,15 +1,19 @@
 #!/usr/bin/python3
 
 import os
+import pickle
 import asyncio
+from datetime import datetime
+from litequeue import SQLQueue
 import neohubapi.neohub as neohub
 
+
 from todb import todb
 
 neohub_ip = os.environ['el_neohub_ip']
 neohub_port = os.environ['el_neohub_port']
 
-values = []
+q = SQLQueue("litequeue.db", maxsize=None)
 
 async def run():
     # Legacy connection
@@ -20,8 +24,9 @@ async def run():
     hub_data, devices = await hub.get_live_data()
     for device in devices['thermostats']:
         print(f"Temperature in zone {device.name}: {device}")
-        values.append((device.time, device.device_id, device.away, device.heat_mode, device.heat_on, device.current_floor_temperature, device.target_temperature, device.temperature))
+        sql = "INSERT INTO neohub (timestamp, time, device_id, away, heat_mode, heat_on, current_floor_temperature, target_temperature, temperature) VALUES(%s,%s,%s,%s,%s,%s,%s,%s,%s)"
+        values = (datetime.utcnow(), device.time, device.device_id, device.away, device.heat_mode, device.heat_on, device.current_floor_temperature, device.target_temperature, device.temperature)
+        q.put(pickle.dumps([sql, values]))
 
 asyncio.run(run())
 
-todb("INSERT INTO neohub (time, device_id, away, heat_mode, heat_on, current_floor_temperature, target_temperature, temperature) VALUES(%s,%s,%s,%s,%s,%s,%s,%s)", values)