fix: Add timeout to wait_for_connection (#2885)

This commit is contained in:
Riccardo Balbo
2025-01-14 11:31:12 +02:00
committed by GitHub
parent 2ee7180aa0
commit bba7614012
+4 -1
View File
@@ -440,11 +440,14 @@ class NWCConnection:
logger.error("Error closing subscription: " + str(e))
return subscription
async def _wait_for_connection(self):
async def _wait_for_connection(self, timeout: int = 60 * 2):
"""
Waits until the connection is ready
"""
t = time.time()
while not self.connected:
if time.time() - t > timeout:
raise Exception("Connection timeout, cannot connect to NWC service")
if self._is_shutting_down():
raise Exception("Connection is closing")
logger.debug("Waiting for connection...")