Fix websocket connection bug/errant logic in Ambient PWS (#34217)

This commit is contained in:
Aaron Bach 2020-04-14 21:30:15 -06:00 committed by Paulus Schoutsen
parent ca48148150
commit 5f97937ba0

View file

@ -347,12 +347,17 @@ class AmbientStation:
async def _attempt_connect(self):
"""Attempt to connect to the socket (retrying later on fail)."""
try:
async def connect(timestamp=None):
"""Connect."""
await self.client.websocket.connect()
try:
await connect()
except WebsocketError as err:
_LOGGER.error("Error with the websocket connection: %s", err)
self._ws_reconnect_delay = min(2 * self._ws_reconnect_delay, 480)
async_call_later(self._hass, self._ws_reconnect_delay, self.ws_connect)
async_call_later(self._hass, self._ws_reconnect_delay, connect)
async def ws_connect(self):
"""Register handlers and connect to the websocket."""