Fix zeroconf thread in asyncio loop #503

AlexxIT · May 3, 2024 · 2b8790c · 2b8790c
1 parent 186dfcb
commit 2b8790c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/custom_components/yandex_station/__init__.py b/custom_components/yandex_station/__init__.py
@@ -201,8 +201,10 @@ async def found_local_speaker(info: dict):
 
     zeroconf = await utils.get_zeroconf_singleton(hass)
 
-    listener = YandexIOListener(hass.loop)
-    listener.start(found_local_speaker, zeroconf)
+    listener = YandexIOListener(
+        lambda info: hass.create_task(found_local_speaker(info))
+    )
+    listener.start(zeroconf)
 
     hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, listener.stop)
 

diff --git a/custom_components/yandex_station/core/yandex_glagol.py b/custom_components/yandex_station/core/yandex_glagol.py
@@ -254,14 +254,13 @@ def debug_msg(self, data: dict):
 
 
 class YandexIOListener:
-    add_handlerer = None
+    add_handler = None
     browser = None
 
-    def __init__(self, loop):
-        self.loop = loop
+    def __init__(self, add_handler: Callable):
+        self.add_handler = add_handler
 
-    def start(self, handlerer: Callable, zeroconf: Zeroconf):
-        self.add_handlerer = handlerer
+    def start(self, zeroconf: Zeroconf):
         self.browser = ServiceBrowser(
             zeroconf, "_yandexio._tcp.local.", handlers=[self._zeroconf_handler]
         )
@@ -287,15 +286,14 @@ def _zeroconf_handler(
                 for k, v in info.properties.items()
             }
 
-            coro = self.add_handlerer(
+            self.add_handler(
                 {
                     "device_id": properties["deviceId"],
                     "platform": properties["platform"],
                     "host": str(ipaddress.ip_address(info.addresses[0])),
                     "port": info.port,
                 }
             )
-            self.loop.create_task(coro)
 
         except Exception as e:
             _LOGGER.debug("Can't get zeroconf info", exc_info=e)