initial public

1 files changed, 77 insertions, 0 deletions

diff --git a/src/inverter_mqtt_receiver.py b/src/inverter_mqtt_receiver.py
new file mode 100755
index 0000000..dd61d9a
--- /dev/null
+++ b/src/inverter_mqtt_receiver.py
@@ -0,0 +1,77 @@
+#!/usr/bin/env python3
+import paho.mqtt.client as mqtt
+import re
+import logging
+
+from home.mqtt import MQTTBase
+from home.mqtt.message import Status, Generation
+from home.database import InverterDatabase
+from home.config import config
+
+logger = logging.getLogger(__name__)
+
+
+class MQTTReceiver(MQTTBase):
+    def __init__(self):
+        super().__init__(clean_session=False)
+        self.database = InverterDatabase()
+
+    def on_connect(self, client: mqtt.Client, userdata, flags, rc):
+        super().on_connect(client, userdata, flags, rc)
+        logger.info("subscribing to home/#")
+        client.subscribe('home/#', qos=1)
+
+    def on_message(self, client: mqtt.Client, userdata, msg):
+        try:
+            match = re.match(r'home/(\d+)/(status|gen)', msg.topic)
+            if not match:
+                return
+
+            home_id, what = int(match.group(1)), match.group(2)
+            if what == 'gen':
+                packer = Generation()
+                client_time, watts = packer.unpack(msg.payload)
+                self.database.add_generation(home_id, client_time, watts)
+
+            elif what == 'status':
+                packer = Status()
+                client_time, data = packer.unpack(msg.payload)
+                self.database.add_status(home_id,
+                                         client_time,
+                                         grid_voltage=int(data['grid_voltage']*10),
+                                         grid_freq=int(data['grid_freq'] * 10),
+                                         ac_output_voltage=int(data['ac_output_voltage'] * 10),
+                                         ac_output_freq=int(data['ac_output_freq'] * 10),
+                                         ac_output_apparent_power=data['ac_output_apparent_power'],
+                                         ac_output_active_power=data['ac_output_active_power'],
+                                         output_load_percent=data['output_load_percent'],
+                                         battery_voltage=int(data['battery_voltage'] * 10),
+                                         battery_voltage_scc=int(data['battery_voltage_scc'] * 10),
+                                         battery_voltage_scc2=int(data['battery_voltage_scc2'] * 10),
+                                         battery_discharging_current=data['battery_discharging_current'],
+                                         battery_charging_current=data['battery_charging_current'],
+                                         battery_capacity=data['battery_capacity'],
+                                         inverter_heat_sink_temp=data['inverter_heat_sink_temp'],
+                                         mppt1_charger_temp=data['mppt1_charger_temp'],
+                                         mppt2_charger_temp=data['mppt2_charger_temp'],
+                                         pv1_input_power=data['pv1_input_power'],
+                                         pv2_input_power=data['pv2_input_power'],
+                                         pv1_input_voltage=int(data['pv1_input_voltage'] * 10),
+                                         pv2_input_voltage=int(data['pv2_input_voltage'] * 10),
+                                         mppt1_charger_status=data['mppt1_charger_status'],
+                                         mppt2_charger_status=data['mppt2_charger_status'],
+                                         battery_power_direction=data['battery_power_direction'],
+                                         dc_ac_power_direction=data['dc_ac_power_direction'],
+                                         line_power_direction=data['line_power_direction'],
+                                         load_connected=data['load_connected'])
+
+        except Exception as e:
+            logger.exception(str(e))
+
+
+if __name__ == '__main__':
+    config.load('inverter_mqtt_receiver')
+
+    server = MQTTReceiver()
+    server.connect_and_loop()
+