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 GitHub
parent 0b90ebf91e
commit 5faa9795b0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions

View File

@ -350,12 +350,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."""