FIX error when production value is None

custom_components/solar_optimizer/coordinator.py

@@ -104,6 +104,12 @@ async def _async_update_data(self):
 
         # Add a power_consumption and power_production
         power_production = get_safe_float(self.hass, self._power_production_entity_id)
+        if not power_production:
+            _LOGGER.warning(
+                "Power production is not valued. Solar Optimizer will be disabled"
+            )
+            return None
+
         if not self._smooth_production:
             calculated_data["power_production"] = power_production
         else:

custom_components/solar_optimizer/sensor.py

@@ -54,8 +54,10 @@ def __init__(self, coordinator, hass, idx):
     @callback
     def _handle_coordinator_update(self) -> None:
         """Handle updated data from the coordinator."""
-        self._attr_native_value = self.coordinator.data[self.idx]
-        self.async_write_ha_state()
+        value = self.coordinator.data.get(self.idx)
+        if value:
+            self._attr_native_value = value
+            self.async_write_ha_state()
 
     @property
     def device_info(self):

custom_components/solar_optimizer/switch.py

@@ -148,7 +148,8 @@ def _handle_coordinator_update(self) -> None:
             return
         device: ManagedDevice = self.coordinator.data.get(self.idx)
         if not device:
-            _LOGGER.warning("No device %s found ...", self.idx)
+            # it is possible to not have device in coordinator update (if device is not enabled)
+            _LOGGER.debug("No device %s found ...", self.idx)
             return
 
         self._attr_is_on = device.is_active