fix: Add timeout to wait_for_connection (#2885)

This commit is contained in:
Riccardo Balbo 2025-01-14 10:31:12 +01:00 committed by GitHub
parent 2ee7180aa0
commit bba7614012
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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...")