Use custom exception

custom_components/integration_blueprint/coordinator.py

@@ -5,9 +5,17 @@
 
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
-from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-
-from .api import IntegrationBlueprintApiClient
+from homeassistant.helpers.update_coordinator import (
+    DataUpdateCoordinator,
+    UpdateFailed,
+)
+from homeassistant.exceptions import ConfigEntryAuthFailed
+
+from .api import (
+    IntegrationBlueprintApiClient,
+    IntegrationBlueprintApiClientAuthenticationError,
+    IntegrationBlueprintApiClientError,
+)
 from .const import DOMAIN, LOGGER
 
 
@@ -34,5 +42,7 @@ async def _async_update_data(self):
         """Update data via library."""
         try:
             return await self.client.async_get_data()
-        except Exception as exception:
-            raise UpdateFailed() from exception
+        except IntegrationBlueprintApiClientAuthenticationError as exception:
+            raise ConfigEntryAuthFailed(exception) from exception
+        except IntegrationBlueprintApiClientError as exception:
+            raise UpdateFailed(exception) from exception