Catch charger not found error

cyberjunky · Aug 12, 2024 · 4662e89 · 4662e89
1 parent 1bf2b93
commit 4662e89
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/custom_components/shell_recharge/__init__.py b/custom_components/shell_recharge/__init__.py
@@ -5,6 +5,8 @@
 from asyncio.exceptions import CancelledError
 
 import shellrecharge
+from shellrecharge import LocationEmptyError
+
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import Platform
 from homeassistant.core import HomeAssistant
@@ -70,6 +72,11 @@ async def _async_update_data(self) -> shellrecharge.Location | None:
         data = None
         try:
             data = await self.api.location_by_id(self.serial_number)
+        except LocationEmptyError:
+            _LOGGER.error(
+                "Error occurred while fetching data for charger(s) %s, not found, or serial is invalid",
+                self.serial_number,
+            )
         except CancelledError:
             _LOGGER.error(
                 "CancelledError occurred while fetching data for charger(s) %s",