interface: handle asyncio.TimeoutError on aiohttp requests

Signed-off-by: Álvaro Fernández Rojas <noltari@gmail.com>
Noltari · Feb 21, 2024 · 3ed2511 · 3ed2511
1 parent 4f06286
commit 3ed2511
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/aemet_opendata/interface.py b/aemet_opendata/interface.py
@@ -138,6 +138,8 @@ async def api_call(self, cmd: str, fetch_data: bool = False) -> dict[str, Any]:
                 timeout=API_TIMEOUT,
                 headers=self.headers,
             )
+        except asyncio.TimeoutError as err:
+            raise AemetTimeout(err) from err
         except ClientError as err:
             raise AemetError(err) from err
 
@@ -185,6 +187,8 @@ async def api_data(self, url: str) -> dict[str, Any]:
                 url,
                 timeout=API_TIMEOUT,
             )
+        except asyncio.TimeoutError as err:
+            raise AemetTimeout(err) from err
         except ClientError as err:
             raise AemetError(err) from err