Add connection timeout to API requests (#238)

sblibs · May 24, 2024 · 612d0ed · 612d0ed
1 parent 5cc84c2
commit 612d0ed
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/switchbot/devices/lock.py b/switchbot/devices/lock.py
@@ -92,7 +92,12 @@ async def api_request(
         headers: dict = None,
     ) -> dict:
         url = f"https://{subdomain}.{SWITCHBOT_APP_API_BASE_URL}/{path}"
-        async with session.post(url, json=data, headers=headers) as result:
+        async with session.post(
+            url,
+            json=data,
+            headers=headers,
+            timeout=aiohttp.ClientTimeout(total=10),
+        ) as result:
             if result.status > 299:
                 raise SwitchbotApiError(
                     f"Unexpected status code returned by SwitchBot API: {result.status}"