diff options
| -rwxr-xr-x | neohub2pgsql.py | 11 |
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) |