Handle connection errors more gracefully

Fixes #41
jjlawren · Jan 10, 2023 · 4775c9c · 4775c9c
1 parent 2455c8c
commit 4775c9c
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/custom_components/sonos_cloud/__init__.py b/custom_components/sonos_cloud/__init__.py
@@ -72,7 +72,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     hass.data[DOMAIN][SESSION] = session
 
     url = "https://api.ws.sonos.com/control/api/v1/households"
-    result = await session.async_request("get", url)
+    try:
+        result = await session.async_request("get", url)
+    except OSError as exc:
+        _LOGGER.error("Connection error requesting households: %s", exc)
+        raise ConfigEntryNotReady from exc
+
     if result.status >= 400:
         body = await result.text()
         _LOGGER.error(
@@ -87,7 +92,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
 
     async def async_get_available_players(household: str) -> list[dict]:
         url = f"https://api.ws.sonos.com/control/api/v1/households/{household}/groups"
-        result = await session.async_request("get", url)
+        try:
+            result = await session.async_request("get", url)
+        except OSError as exc:
+            _LOGGER.error("Connection error requesting players: %s", exc)
+            raise ConfigEntryNotReady from exc
         if result.status >= 400:
             body = await result.text()
             _LOGGER.error(