initial public

3 files changed, 107 insertions, 0 deletions

diff --git a/src/home/mqtt/message/__init__.py b/src/home/mqtt/message/__init__.py
new file mode 100644
index 0000000..2a2221b
--- /dev/null
+++ b/src/home/mqtt/message/__init__.py
@@ -0,0 +1,2 @@
+from .inverter import Status, Generation
+from .sensors import Temperature
diff --git a/src/home/mqtt/message/inverter.py b/src/home/mqtt/message/inverter.py
new file mode 100644
index 0000000..2df17e5
--- /dev/null
+++ b/src/home/mqtt/message/inverter.py
@@ -0,0 +1,86 @@
+import struct
+
+from typing import Tuple
+
+
+class Status:
+    # 46 bytes
+    format = 'IHHHHHHBHHHHHBHHHHHHHH'
+
+    def pack(self, time: int, data: dict) -> bytes:
+        bits = 0
+        bits |= (data['mppt1_charger_status'] & 0x3)
+        bits |= (data['mppt2_charger_status'] & 0x3) << 2
+        bits |= (data['battery_power_direction'] & 0x3) << 4
+        bits |= (data['dc_ac_power_direction'] & 0x3) << 6
+        bits |= (data['line_power_direction'] & 0x3) << 8
+        bits |= (data['load_connected'] & 0x1) << 10
+
+        return struct.pack(
+            self.format,
+            time,
+            int(data['grid_voltage'] * 10),
+            int(data['grid_freq'] * 10),
+            int(data['ac_output_voltage'] * 10),
+            int(data['ac_output_freq'] * 10),
+            data['ac_output_apparent_power'],
+            data['ac_output_active_power'],
+            data['output_load_percent'],
+            int(data['battery_voltage'] * 10),
+            int(data['battery_voltage_scc'] * 10),
+            int(data['battery_voltage_scc2'] * 10),
+            data['battery_discharging_current'],
+            data['battery_charging_current'],
+            data['battery_capacity'],
+            data['inverter_heat_sink_temp'],
+            data['mppt1_charger_temp'],
+            data['mppt2_charger_temp'],
+            data['pv1_input_power'],
+            data['pv2_input_power'],
+            int(data['pv1_input_voltage'] * 10),
+            int(data['pv2_input_voltage'] * 10),
+            bits
+        )
+
+    def unpack(self, buf: bytes) -> Tuple[int, dict]:
+        data = struct.unpack(self.format, buf)
+        return data[0], {
+            'grid_voltage': data[1] / 10,
+            'grid_freq': data[2] / 10,
+            'ac_output_voltage': data[3] / 10,
+            'ac_output_freq': data[4] / 10,
+            'ac_output_apparent_power': data[5],
+            'ac_output_active_power': data[6],
+            'output_load_percent': data[7],
+            'battery_voltage': data[8] / 10,
+            'battery_voltage_scc': data[9] / 10,
+            'battery_voltage_scc2': data[10] / 10,
+            'battery_discharging_current': data[11],
+            'battery_charging_current': data[12],
+            'battery_capacity': data[13],
+            'inverter_heat_sink_temp': data[14],
+            'mppt1_charger_temp': data[15],
+            'mppt2_charger_temp': data[16],
+            'pv1_input_power': data[17],
+            'pv2_input_power': data[18],
+            'pv1_input_voltage': data[19] / 10,
+            'pv2_input_voltage': data[20] / 10,
+            'mppt1_charger_status': data[21] & 0x03,
+            'mppt2_charger_status': (data[21] >> 2) & 0x03,
+            'battery_power_direction': (data[21] >> 4) & 0x03,
+            'dc_ac_power_direction': (data[21] >> 6) & 0x03,
+            'line_power_direction': (data[21] >> 8) & 0x03,
+            'load_connected': (data[21] >> 10) & 0x01,
+        }
+
+
+class Generation:
+    # 8 bytes
+    format = 'II'
+
+    def pack(self, time: int, wh: int) -> bytes:
+        return struct.pack(self.format, int(time), wh)
+
+    def unpack(self, buf: bytes) -> tuple:
+        data = struct.unpack(self.format, buf)
+        return tuple(data)
diff --git a/src/home/mqtt/message/sensors.py b/src/home/mqtt/message/sensors.py
new file mode 100644
index 0000000..ee522f0
--- /dev/null
+++ b/src/home/mqtt/message/sensors.py
@@ -0,0 +1,19 @@
+import struct
+
+from typing import Tuple
+
+
+class Temperature:
+    format = 'IhH'
+
+    def pack(self, time: int, temp: float, rh: float) -> bytes:
+        return struct.pack(
+            self.format,
+            time,
+            int(temp*100),
+            int(rh*100)
+        )
+
+    def unpack(self, buf: bytes) -> Tuple[int, float, float]:
+        data = struct.unpack(self.format, buf)
+        return data[0], data[1]/100, data[2]/100