From 291d36693314fff6ba9c7da853fec1d666de2bfb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jan=20Jedelsk=C3=BD?= <8687359+jjedelsky@users.noreply.github.com> Date: Mon, 17 Oct 2022 09:42:20 +0200 Subject: [PATCH] Handle ReadTimeout during wolflink setup (#78135) * Handle ReadTimeout during wolflink setup * Reorder imports Co-authored-by: Yevhenii Vaskivskyi * Reorder exceptions Co-authored-by: Yevhenii Vaskivskyi * Use RequestError instead of ConnectError, ReadTimeout, and ConnectTimeout Co-authored-by: Yevhenii Vaskivskyi --- homeassistant/components/wolflink/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/wolflink/__init__.py b/homeassistant/components/wolflink/__init__.py index 9d76c61806b..290e7337617 100644 --- a/homeassistant/components/wolflink/__init__.py +++ b/homeassistant/components/wolflink/__init__.py @@ -2,7 +2,7 @@ from datetime import timedelta import logging -from httpx import ConnectError, ConnectTimeout +from httpx import RequestError from wolf_smartset.token_auth import InvalidAuth from wolf_smartset.wolf_client import FetchFailed, ParameterReadError, WolfClient @@ -74,7 +74,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: for parameter in parameters if parameter.value_id in values } - except ConnectError as exception: + except RequestError as exception: raise UpdateFailed( f"Error communicating with API: {exception}" ) from exception @@ -134,7 +134,7 @@ async def fetch_parameters_init(client: WolfClient, gateway_id: int, device_id: """Fetch all available parameters with usage of WolfClient but handles all exceptions and results in ConfigEntryNotReady.""" try: return await fetch_parameters(client, gateway_id, device_id) - except (ConnectError, ConnectTimeout, FetchFailed) as exception: + except (FetchFailed, RequestError) as exception: raise ConfigEntryNotReady( f"Error communicating with API: {exception}" ) from exception