home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-24 15:01:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

use correct exception

Browse Source
This commit is contained in:
Thomas Protzner
2024-08-12 17:14:19 +00:00
parent 052aa3e203
commit b10ce49222
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/components/husqvarna_automower/coordinator.py
View File

@@ -8,6 +8,7 @@ from aioautomower.exceptions import (
ApiException, ApiException,
AuthException, AuthException,
HusqvarnaWSServerHandshakeError, HusqvarnaWSServerHandshakeError,
TimeoutException,
) )
from aioautomower.model import MowerAttributes from aioautomower.model import MowerAttributes
from aioautomower.session import AutomowerSession from aioautomower.session import AutomowerSession
@@ -73,11 +74,16 @@ class AutomowerDataUpdateCoordinator(DataUpdateCoordinator[dict[str, MowerAttrib
await automower_client.auth.websocket_connect() await automower_client.auth.websocket_connect()
await automower_client.start_listening() # Ensure we catch errors here too await automower_client.start_listening() # Ensure we catch errors here too
self.reconnect_time = DEFAULT_RECONNECT_TIME # Reset reconnect time after successful connection self.reconnect_time = DEFAULT_RECONNECT_TIME # Reset reconnect time after successful connection
except (HusqvarnaWSServerHandshakeError, Exception) as err: except HusqvarnaWSServerHandshakeError as err:
_LOGGER.debug( _LOGGER.debug(
"Failed to connect to websocket. Trying to reconnect: %s", "Failed to connect to websocket. Trying to reconnect: %s",
err, err,
) )
except TimeoutException as err:
_LOGGER.debug(
"Failed to listen to websocket. Trying to reconnect: %s",
err,
)
if not hass.is_stopping: if not hass.is_stopping:
await asyncio.sleep(self.reconnect_time) await asyncio.sleep(self.reconnect_time)
self.reconnect_time = min(self.reconnect_time * 2, MAX_WS_RECONNECT_TIME) self.reconnect_time = min(self.reconnect_time * 2, MAX_WS_RECONNECT_TIME)