Merge pull request #5 from alexandrehassan/host_exception

Add exception for un-reachable host

Author: sydragos

sagemcom_api/client.py

@@ -10,7 +10,7 @@
 from typing import Dict, List, Optional, Type
 import urllib.parse
 
-from aiohttp import ClientSession, ClientTimeout
+from aiohttp import ClientConnectionError, ClientSession, ClientTimeout
 from aiohttp.connector import TCPConnector
 import humps
 
@@ -33,6 +33,7 @@
     AccessRestrictionException,
     AuthenticationException,
     BadRequestException,
+    LoginConnectionException,
     LoginTimeoutException,
     MaximumSessionCountException,
     NonWritableParameterException,
@@ -275,6 +276,10 @@ async def login(self):
             raise LoginTimeoutException(
                 "Request timed-out. This is mainly due to using the wrong encryption method."
             ) from exception
+        except ClientConnectionError as exception:
+            raise LoginConnectionException(
+                "Unable to connect to the device. Please check the host address."
+            ) from exception
 
         data = self.__get_response(response)
 

sagemcom_api/exceptions.py

@@ -20,6 +20,12 @@ class LoginTimeoutException(Exception):
     pass
 
 
+class LoginConnectionException(Exception):
+    """Raised when a connection error is encountered during login."""
+
+    pass
+
+
 class NonWritableParameterException(Exception):
     """Raised when provided parameter is not writable."""