Replace asyncio.wait_for with asyncio.timeout

bdraco · bdraco · commit f0630f548955 · 2023-07-20T21:36:19.000-05:00
Fallback to using async_timeout on older python asyncio.wait_for has some underlying problems that are only fixed in cpython 3.12. See python/cpython#98518
diff --git a/kasa/protocol.py b/kasa/protocol.py
@@ -165,7 +165,7 @@ async def _query(self, request: str, retry_count: int, timeout: int) -> Dict:
                 assert self.reader is not None
                 assert self.writer is not None
                 async with asyncio_timeout(timeout):
-                    await self._execute_query(request)
+                    return await self._execute_query(request)
             except Exception as ex:
                 await self.close()
                 if retry >= retry_count:
