localapi: add timeout exception

Signed-off-by: Álvaro Fernández Rojas <[email protected]>
Noltari · Aug 28, 2023 · 444e58d · 444e58d
1 parent 1cb5a08
commit 444e58d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/aioqsw/exceptions.py b/aioqsw/exceptions.py
@@ -10,6 +10,10 @@ class APIError(QswError):
     """Exception raised when API fails."""
 
 
+class APITimeout(QswError):
+    """Exception raised when API timeouts."""
+
+
 class InternalServerError(APIError):
     """Exception raised when API returns internal server error."""
 

diff --git a/aioqsw/localapi.py b/aioqsw/localapi.py
@@ -53,6 +53,7 @@
 )
 from .exceptions import (
     APIError,
+    APITimeout,
     InternalServerError,
     InvalidHost,
     InvalidResponse,
@@ -117,6 +118,8 @@ async def http_request_bytes(
             )
         except ClientError as err:
             raise InvalidHost(err) from err
+        except asyncio.TimeoutError as err:
+            raise APITimeout(err) from err
 
         resp_bytes = await resp.read()
 
@@ -144,6 +147,8 @@ async def http_request(
             )
         except ClientError as err:
             raise InvalidHost(err) from err
+        except asyncio.TimeoutError as err:
+            raise APITimeout(err) from err
 
         resp_json = None
         try: