Add exceptions

Author: fabaff

elmax/__init__.py

@@ -5,6 +5,8 @@
 import httpx
 import json
 
+from . import exceptions
+
 from .constants import (
     BASE_URL,
     ENDPOINT_DEVICES,
@@ -52,11 +54,12 @@ async def connect(self):
             "username": self.username,
             "password": self.password,
         }
-
-        async with httpx.AsyncClient() as client:
-            response = await client.post(str(url), headers=headers, data=data)
-
-        _LOGGER.debug("Status code:", response.status_code)
+        try:
+            async with httpx.AsyncClient() as client:
+                response = await client.post(str(url), headers=headers, data=data)
+                _LOGGER.debug("Status code:", response.status_code)
+        except httpx.ConnectError:
+            raise exceptions.ElmaxConnectionError(f"Connection to {BASE_URL} failed")
 
         try:
             response_data = response.json()
@@ -80,7 +83,7 @@ def is_authenticated(self):
     async def get_devices(self):
         """Retrieve the devices."""
         if self.authorized is False:
-            self.connect()
+            await self.connect()
 
         url = URL(BASE_URL) / ENDPOINT_DEVICES
         headers["Authorization"] = self.authorization
@@ -109,7 +112,7 @@ async def list_devices(self):
     async def get_units(self, device_id, pin):
         """List all units of a devices."""
         if self.authorized is False:
-            self.connect()
+            await self.connect()
 
         url = URL(BASE_URL) / ENDPOINT_DISCOVERY / device_id / str(pin)
 

elmax/exceptions.py

@@ -0,0 +1,19 @@
+"""Exceptions for the Elmax Cloud services client."""
+
+
+class ElmaxError(Exception):
+    """General ElmxError exception occurred."""
+
+    pass
+
+
+class ElmaxConnectionError(ElmaxError):
+    """When a connection error is encountered."""
+
+    pass
+
+
+class ElmaxNoDataAvailable(ElmaxError):
+    """When no data is available."""
+
+    pass