From 42480df5c128c2c2c36253f9222878ef1126d33a Mon Sep 17 00:00:00 2001
From: Tim Dithmer <dith.tim@gmail.com>
Date: Mon, 28 Oct 2024 21:07:57 +0100
Subject: [PATCH] Add initial state of huette energy consumption per week day
 script

---
 .dockerignore    |  1 +
 .gitignore       |  1 +
 Dockerfile       |  9 ++++++
 README           | 11 ++++++++
 main.py          | 72 ++++++++++++++++++++++++++++++++++++++++++++++++
 requirements.txt |  6 ++++
 6 files changed, 100 insertions(+)
 create mode 100644 .dockerignore
 create mode 100644 .gitignore
 create mode 100644 Dockerfile
 create mode 100644 README
 create mode 100644 main.py
 create mode 100644 requirements.txt

diff --git a/.dockerignore b/.dockerignore
new file mode 100644
index 0000000..f7275bb
--- /dev/null
+++ b/.dockerignore
@@ -0,0 +1 @@
+venv/
diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..f7275bb
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+venv/
diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..c82ec53
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,9 @@
+FROM python:3.12-slim
+
+COPY . /app
+
+WORKDIR /app
+
+RUN pip install -r requirements.txt
+
+ENTRYPOINT ["python3", "main.py"]
diff --git a/README b/README
new file mode 100644
index 0000000..fa7f379
--- /dev/null
+++ b/README
@@ -0,0 +1,11 @@
+# Build
+
+```
+docker build -t huette-energy-consumption-weekperday .
+```
+
+# Run
+
+```
+docker run -d -e MQTT_HOST=10.2.3.2 -e MQTT_TOPIC=c3re/huette/energy/consumption/week_per_day -t huette-energy-consumption-weekperday
+```
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..a61c1e7
--- /dev/null
+++ b/main.py
@@ -0,0 +1,72 @@
+import datetime  # Import datetime module
+import requests
+import paho.mqtt.client as mqtt
+import paho.mqtt.enums as mqtt_enums
+import os
+import json
+import time
+
+base_url = "https://prom.c3re.de/api/v1/query"
+
+
+def get_params(timestamp):
+    return {
+        "query": 'energy{location="Hütte", type="Energymeter"}',
+        "time": str(timestamp),
+    }
+
+
+def calc_prometheus_response_value_diff(response_before, response_after):
+    return int(response_after.json()["data"]["result"][0]["value"][1]) - int(
+        response_before.json()["data"]["result"][0]["value"][1]
+    )
+
+
+def get_energy_consumption_between(b, a):
+    response_before = requests.get(base_url, params=get_params(b.timestamp()))
+    response_after = requests.get(base_url, params=get_params(a.timestamp()))
+    return calc_prometheus_response_value_diff(response_before, response_after)
+
+
+def get_energy_consumption_1_day(n):
+    return get_energy_consumption_between(n - datetime.timedelta(days=1), n)
+
+
+def get_energy_consumption_today():
+    return get_energy_consumption_between(
+        datetime.datetime.now().replace(hour=0, minute=0, second=0, microsecond=0),
+        datetime.datetime.now(),
+    )
+
+
+def main():
+    host = os.getenv("MQTT_HOST")
+    port = int(os.getenv("MQTT_PORT", 1883))
+    topic = os.getenv("MQTT_TOPIC")
+
+    mqttc = mqtt.Client(mqtt_enums.CallbackAPIVersion.VERSION2)
+    mqttc.connect(host, port)
+    mqttc.loop_start()
+
+    while True:
+        consumptions = []
+        consumptions.append(get_energy_consumption_today())
+
+        n = datetime.datetime.now().replace(hour=0, minute=0, second=0, microsecond=0)
+        days_back = 1
+        while days_back <= 7:
+            consumptions.append(get_energy_consumption_1_day(n))
+
+            days_back += 1
+            n -= datetime.timedelta(days=1)
+
+        print("sending data")
+        print(consumptions)
+
+        consumptions_msg = mqttc.publish(topic, json.dumps(consumptions))
+        consumptions_msg.wait_for_publish()
+        time.sleep(15)
+
+
+if __name__ == "__main__":
+    main()
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..b4ca7cb
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,6 @@
+certifi==2024.8.30
+charset-normalizer==3.4.0
+idna==3.10
+paho-mqtt==2.1.0
+requests==2.32.3
+urllib3==2.2.3