fix: Fix Today/Yesterday readings (#228)

Author: GuyKh

custom_components/iec/__init__.py

@@ -16,10 +16,16 @@
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Set up IEC from a config entry."""
+    if DOMAIN not in hass.data:
+        hass.data[DOMAIN] = {}
 
     iec_coordinator = IecApiCoordinator(hass, entry)
-    await iec_coordinator.async_config_entry_first_refresh()
     hass.data.setdefault(DOMAIN, {})[entry.entry_id] = iec_coordinator
+    try:
+        await hass.data[DOMAIN][entry.entry_id].async_config_entry_first_refresh()
+    except Exception as err:
+        # Log the error but don't fail the setup
+        _LOGGER.error("Failed to fetch initial data: %s", err)
 
     await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
 
@@ -40,6 +46,8 @@ async def handle_debug_get_coordinator_data(call) -> None:  # noqa: ANN001 ARG00
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Unload a config entry."""
     if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
-        hass.data[DOMAIN].pop(entry.entry_id)
+        coordinator = hass.data[DOMAIN].pop(entry.entry_id, None)
+        if coordinator:
+            await coordinator.async_unload()
 
     return unload_ok