Fix: Handle missing Connection-Size (#162)

Author: GuyKh

custom_components/iec/coordinator.py

@@ -642,21 +642,30 @@ async def _estimate_bill(self, contract_id, device_number, is_private_producer,
                 phase_count = None
                 connection_size = None
 
-        if not is_private_producer or not last_meter_read:
+        if is_private_producer or not last_meter_read:
             last_meter_reading = await self._get_last_meter_reading(self._bp_number, contract_id,
                                                                     device_number)
             last_meter_read = last_meter_reading.reading
             last_meter_read_date = last_meter_reading.reading_date.date()
 
             account_id = await self._get_account_id()
             connection_size = await self._get_connection_size(account_id)
-            phase_count_str = connection_size.split("X")[0] \
-                if connection_size.find("X") != -1 else "1"
-            phase_count = int(phase_count_str)
+            if connection_size:
+                phase_count_str = connection_size.split("X")[0] \
+                    if connection_size.find("X") != -1 else "1"
+                phase_count = int(phase_count_str)
 
-        distribution_tariff = await self._get_distribution_tariff(phase_count)
-        delivery_tariff = await self._get_delivery_tariff(phase_count)
-        power_size = await self._get_power_size(connection_size)
+        if connection_size:
+            power_size = await self._get_power_size(connection_size)
+        else:
+            _LOGGER.warning("Couldn't get Connection Size")
+
+        if phase_count:
+            distribution_tariff = await self._get_distribution_tariff(phase_count)
+            delivery_tariff = await self._get_delivery_tariff(phase_count)
+        else:
+            if connection_size:
+                _LOGGER.warning("Couldn't get Phase Count")
 
         return self._calculate_estimated_bill(device_number, future_consumption,
                                               last_meter_read, last_meter_read_date,

custom_components/iec/manifest.json

@@ -10,6 +10,6 @@
   "iot_class": "cloud_polling",
   "issue_tracker": "https://github.com/guykh/iec-custom-component/issues",
   "loggers": ["iec_api"],
-  "requirements": ["iec-api==0.4.2"],
-  "version": "0.0.33"
+  "requirements": ["iec-api==0.4.3"],
+  "version": "0.0.33-b2"
 }

requirements.txt

@@ -1,5 +1,5 @@
 colorlog>=6.8.2
 homeassistant==2024.2.0
-iec-api==0.4.2
+iec-api==0.4.3
 pip>=21.0
 ruff>=0.5.6