Add unsupported host exception (#339)

iMicknl · Aug 9, 2024 · 9fda038 · 9fda038
1 parent 0071101
commit 9fda038
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/sagemcom_api/client.py b/sagemcom_api/client.py
@@ -51,6 +51,7 @@
     UnauthorizedException,
     UnknownException,
     UnknownPathException,
+    UnsupportedHostException,
 )
 from .models import Device, DeviceInfo, PortMapping
 
@@ -203,6 +204,10 @@ async def __post(self, url, data):
                 result = await response.text()
                 raise BadRequestException(result)
 
+            if response.status == 404:
+                result = await response.text()
+                raise UnsupportedHostException(result)
+
             if response.status != 200:
                 result = await response.text()
                 raise UnknownException(result)

diff --git a/sagemcom_api/exceptions.py b/sagemcom_api/exceptions.py
@@ -18,6 +18,10 @@ class UnknownException(BaseSagemcomException):
     """Unknown exception."""
 
 
+class UnsupportedHostException(BaseSagemcomException):
+    """Raised when API is not available on given host."""
+
+
 # Exceptions provided by SagemCom API
 class AccessRestrictionException(BaseSagemcomException):
     """Raised when current user has access restrictions."""