Improve waiting for transport/protocol

StevenLooman · Jan 24, 2025 · 25972b6 · 25972b6
1 parent ee50f79
commit 25972b6
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/async_upnp_client/server.py b/async_upnp_client/server.py
@@ -859,19 +859,15 @@ async def async_stop(self) -> None:
 
     async def async_wait_for_transport_protocol(self) -> None:
         """Wait for the protocol to become available."""
-        loop = 0
-        max_loops = 5
-        while loop < max_loops:
-            await asyncio.sleep(0.1)
+        for _ in range(0, 5):
             if (
                 self._transport is not None
                 and self._transport.get_protocol() is not None
             ):
                 break
 
-            loop += 1
-
-        if loop > max_loops:
+            await asyncio.sleep(0.1)
+        else:
             raise UpnpError("Failed to get protocol")
 
     def _announce_next(self) -> None: