combine scripts

Dockerfile

@@ -1,4 +1,4 @@
-FROM python:3.12-bookworm
+FROM python:3.12-slim-bookworm
 
 # Labels
 LABEL MAINTAINER="Jan-Ole Hübner <huebner@jan-ole.de>"

get_price.py

@@ -1,35 +0,0 @@
-import os
-import sys
-import tibber
-from influxdb_client import InfluxDBClient
-from influxdb_client.client.write_api import SYNCHRONOUS
-from DataPoints import Price
-import logging
-
-TOKEN = os.getenv('TOKEN', '')
-TIBBERTOKEN = os.getenv('TIBBERTOKEN', '')
-URL = os.getenv('URL', "")
-BUCKET = os.getenv('BUCKET', "tibber")
-ORG = os.getenv('ORG', "Default")
-
-# Logging configuration
-logger = logging.getLogger("TibberInflux")
-formatter = logging.Formatter(
-    '%(asctime)s - %(name)s - %(levelname)s - %(message)s')
-ch = logging.StreamHandler(sys.stdout)
-ch.setFormatter(formatter)
-logger.addHandler(ch)
-logger.setLevel(logging.INFO)
-
-__version__ = "v0.3.0"
-logger.info(__version__)
-
-client = InfluxDBClient(url=URL, token=TOKEN, org=ORG)
-write_api = client.write_api(write_options=SYNCHRONOUS)
-query_api = client.query_api()
-
-account = tibber.Account(TIBBERTOKEN)
-home = account.homes[0]
-p = Price(home.current_subscription.price_info.current).get_datapoint()
-write_api.write(record=p, bucket=BUCKET)
-logger.info(vars(p))

pulse.py

@@ -2,13 +2,12 @@ import asyncio
 import os
 import signal
 import sys
-import time
+import logging
 
 import tibber
 from influxdb_client import InfluxDBClient
 from influxdb_client.client.write_api import SYNCHRONOUS
-from DataPoints import Pulse
-import logging
+from DataPoints import Pulse, Price
 
 TOKEN = os.getenv('TOKEN', '')
 TIBBERTOKEN = os.getenv('TIBBERTOKEN', '')
@@ -18,25 +17,24 @@ ORG = os.getenv('ORG', "Default")
 
 # Logging configuration
 logger = logging.getLogger("TibberInflux")
-formatter = logging.Formatter(
-    '%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 ch = logging.StreamHandler(sys.stdout)
 ch.setFormatter(formatter)
 logger.addHandler(ch)
 logger.setLevel(logging.INFO)
 
-__version__ = "v0.3.4"
+__version__ = "v0.4.0"
 logger.info(__version__)
 
-
-logger.info("connecting to db...")
+# Initialize InfluxDB client
 client = InfluxDBClient(url=URL, token=TOKEN, org=ORG)
 write_api = client.write_api(write_options=SYNCHRONOUS)
 query_api = client.query_api()
-logger.info("connecting to tibber...")
+
+# Initialize Tibber account
 account = tibber.Account(TIBBERTOKEN)
 home = account.homes[0]
-logger.info("starting subscription...")
+
 
 def _incoming(data):
     try:
@@ -47,32 +45,60 @@ def _incoming(data):
         logger.exception("Error in _incoming():")
         raise e
 
+
 def timeout_handler(signum, frame):
     raise TypeError("Timeout occurred")
 
-@home.event("live_measurement")
+
 async def show_current_power(data):
+    signal.signal(signal.SIGALRM, timeout_handler)
     signal.alarm(8)
     _incoming(data)
 
-def stop(home):
-    return False
 
-try:
+async def periodic_task():
+    while True:
+        try:
+            # Fetch price info and write to InfluxDB
+            p = Price(home.current_subscription.price_info.current).get_datapoint()
+            write_api.write(record=p, bucket=BUCKET)
+            logger.info(vars(p))
+        except Exception as e:
+            logger.exception("Error in periodic_task():")
+        # Wait for the next run (e.g., every 60 seconds)
+        await asyncio.sleep(3)
+
+
+async def main():
+    loop = asyncio.get_event_loop()
     signal.signal(signal.SIGALRM, timeout_handler)
     signal.alarm(15)
-    home.start_live_feed(user_agent="pulse.py/0.3.4",exit_condition=stop(home),retries=2,retry_interval=3.0)
-except TypeError:
-    logger.exception("Timeout occurred while executing start_live_feed()")
-    client.close()
-    exit(1)
-except Exception:
-    logger.exception("Error in start_live_feed()")
-    client.close()
-    exit(41)
-
-
-
-
+
+    try:
+        # Start the live feed in a separate task
+        home_feed_task = loop.create_task(home.start_live_feed(
+            user_agent="pulse.py/0.4.0",
+            exit_condition=lambda: False,
+            retries=2,
+            retry_interval=3.0,
+            callback=show_current_power
+        ))
+
+        # Start the periodic task
+        periodic_task_task = loop.create_task(periodic_task())
+
+        # Wait for both tasks to complete
+        await asyncio.gather(home_feed_task, periodic_task_task)
+
+    except TypeError:
+        logger.exception("Timeout occurred while executing start_live_feed()")
+        client.close()
+        sys.exit(1)
+    except Exception:
+        logger.exception("Error in start_live_feed()")
+        client.close()
+        sys.exit(41)
 
 
+if __name__ == "__main__":
+    asyncio.run(main())