Fix Home Assistant reboot

AlexxIT · Jun 1, 2020 · dac5fda · dac5fda
1 parent 565a7c0
commit dac5fda
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/custom_components/yandex_station/yandex_glagol.py b/custom_components/yandex_station/yandex_glagol.py
@@ -89,6 +89,15 @@ async def _connect(self, session: ClientSession):
             _LOGGER.error(f"Station connect error [{e.errno}] {e}")
             await asyncio.sleep(30)
 
+        except (asyncio.CancelledError, RuntimeError) as e:
+            if isinstance(e, RuntimeError):
+                assert e.args[0] == 'Session is closed', e.args
+
+            _LOGGER.debug(f"Останавливаем локальное подключение: {e}")
+            if not self.ws.closed:
+                await self.ws.close()
+            return
+
         except:
             _LOGGER.exception(f"Station connect")
             await asyncio.sleep(30)