services/power-mqtt: Rewrite to Python

bar.yaml

@@ -1,5 +1,5 @@
 ---
-- hosts: bar
+- hosts: bank-pi.bitlair.nl
   vars:
     raspi_rotate_display: "2"
   roles:

roles/bank/tasks/inflatinator.yaml

@@ -0,0 +1,12 @@
+---
+- name: Install dependencies
+  apt:
+    name: [ links, python3-pyquery ]
+    state: present
+
+- name: Clone revbank-inflatinator source
+  git:
+    repo: https://github.com/bitlair/revbank-inflatinator.git
+    version: main
+    dest: /opt/revbank-inflatinator
+    accept_hostkey: yes

roles/bank/tasks/main.yaml

@@ -4,3 +4,6 @@
 
 - tags: [ bank, bank_revbank ]
   import_tasks: revbank.yaml
+
+- tags: [ bank, bank_inflatinator ]
+  import_tasks: inflatinator.yaml

roles/services/tasks/power_mqtt.yaml

@@ -1,7 +1,12 @@
+---
+- name: Install dependencies
+  apt:
+    name: [python3-paho-mqtt, python3-requests]
+
 - name: Install power-mqtt
   template:
-    src: power-mqtt.sh
-    dest: /var/lib/power-mqtt.sh
+    src: power-mqtt.py
+    dest: /var/lib/power-mqtt.py
     owner: root
     group: root
     mode: 0755

roles/services/templates/power-mqtt.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+
+import paho.mqtt.client as mqtt
+import paho.mqtt.subscribe as subscribe
+from time import sleep
+import requests
+
+
+sdm630_host = '100.64.0.187'
+mqtt_host = '{{ mqtt_internal_host }}'
+
+
+client = mqtt.Client()
+client.connect(mqtt_host, 1883, 60)
+
+while True:
+    sleep(1)
+    try:
+        data = requests.get(f'http://{sdm630_host}/api/v1/data').json()
+        client.publish('bitlair/power/total_kwh', data['total_power_import_kwh'])
+        client.publish('bitlair/power/All/all_now', data['active_power_w'])
+        client.publish('bitlair/power/L1/all_now', data['active_power_l1_w'])
+        client.publish('bitlair/power/L2/all_now', data['active_power_l2_w'])
+        client.publish('bitlair/power/L3/all_now', data['active_power_l3_w'])
+    except Exception as err:
+        print(err)
+        continue

roles/services/templates/power-mqtt.service

@@ -6,11 +6,10 @@ After=network.target
 
 [Service]
 Type=simple
-ExecStart=/bin/bash /var/lib/power-mqtt.sh
+ExecStart=/var/lib/power-mqtt.py
 Restart=on-failure
 RestartSec=10s
 DynamicUser=true
 
 [Install]
 WantedBy=multi-user.target
-

roles/services/templates/power-mqtt.sh

@@ -1,19 +0,0 @@
-#!/bin/bash
-
-# Managed by Ansible
-
-set -euo pipefail
-
-h="{{ mqtt_internal_host }}"
-
-while true; do
-    j=$(curl -s http://100.64.0.187/api/v1/data)
-
-    mqtt-simple -h $h -p bitlair/power/total_kwh -m $(echo "$j" | jq .total_power_import_kwh)
-    mqtt-simple -h $h -p bitlair/power/All/now_w -m $(echo "$j" | jq .active_power_w)
-    mqtt-simple -h $h -p bitlair/power/L1/now_w -m $(echo "$j" | jq .active_power_l1_w)
-    mqtt-simple -h $h -p bitlair/power/L2/now_w -m $(echo "$j" | jq .active_power_l2_w)
-    mqtt-simple -h $h -p bitlair/power/L3/now_w -m $(echo "$j" | jq .active_power_l3_w)
-
-    sleep 1
-done