Merge pull request #34 from MangelSpec/master

Change default CannotConnect strategy to retrying instead of failure …
dvejsada · Jan 30, 2025 · 46db753 · 46db753
2 parents da10d89 + 773e7d6
commit 46db753
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/custom_components/pid_departures/__init__.py b/custom_components/pid_departures/__init__.py
@@ -14,7 +14,15 @@
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Set up Departure Board from a config entry flow."""
     hub = DepartureBoard(hass, entry.data[CONF_API_KEY], entry.data[CONF_ID], entry.data[CONF_DEP_NUM])  # type: ignore[Any]
-    await hub.async_update()
+    try:
+        await hub.async_update()
+    except CannotConnect:
+        # try again later again
+        raise ConfigEntryNotReady from None
+    except StopNotFound:
+        return False
+    except WrongApiKey:
+        return False
 
     hass.data.setdefault(DOMAIN, {})[entry.entry_id] = hub  # type: ignore[Any]