fix: Handle timeout errors separately from "general" ClientError

aspxstats/__init__.py

@@ -1,4 +1,4 @@
-from .exceptions import Error, ClientError, InvalidResponseError, NotFoundError, InvalidParameterError
+from .exceptions import Error, ClientError, TimeoutError, InvalidResponseError, NotFoundError, InvalidParameterError
 from .types import ResponseValidationMode
 
 """
@@ -14,6 +14,7 @@
     'ResponseValidationMode',
     'Error',
     'ClientError',
+    'TimeoutError',
     'InvalidResponseError',
     'NotFoundError',
     'InvalidParameterError'

aspxstats/async_client.py

@@ -1,11 +1,12 @@
+import asyncio
 from enum import Enum
 from typing import Dict, Optional, Union
 from urllib.parse import urljoin
 
 import aiohttp as aiohttp
 
 from .client import AspxClient
-from .exceptions import ClientError
+from .exceptions import ClientError, TimeoutError
 from .types import ResponseValidationMode
 
 
@@ -46,5 +47,7 @@ async def get_aspx_data(self, endpoint: str, params: Optional[Dict[str, Optional
                 return await response.text(errors='replace')
             else:
                 raise ClientError(f'Failed to fetch ASPX data (HTTP/{response.status})')
+        except asyncio.TimeoutError:
+            raise TimeoutError('Timed out trying to fetch ASPX data')
         except aiohttp.ClientError as e:
             raise ClientError(f'Failed to fetch ASPX data: {e}')

aspxstats/client.py

@@ -5,7 +5,7 @@
 
 import requests as requests
 
-from .exceptions import ClientError, InvalidResponseError, Error
+from .exceptions import ClientError, InvalidResponseError, Error, TimeoutError
 from .types import LineType, Dataset, ParseTarget, ResponseValidationMode
 
 
@@ -57,6 +57,8 @@ def get_aspx_data(self, endpoint: str, params: Optional[Dict[str, Optional[Union
                 return response.text
             else:
                 raise ClientError(f'Failed to fetch ASPX data (HTTP/{response.status_code})')
+        except requests.Timeout:
+            raise TimeoutError('Timed out trying to fetch ASPX data')
         except requests.RequestException as e:
             raise ClientError(f'Failed to fetch ASPX data: {e}')
 

aspxstats/exceptions.py

@@ -6,6 +6,10 @@ class ClientError(Error):
     pass
 
 
+class TimeoutError(Error):
+    pass
+
+
 class InvalidResponseError(Error):
     pass