Add/rework exceptions

Author: CoMPaTech

airos/airos8.py

@@ -8,7 +8,12 @@
 
 import aiohttp
 
-from .exceptions import ConnectionFailedError, DataMissingError
+from .exceptions import (
+    ConnectionAuthenticationError,
+    ConnectionError,
+    ConnectionSetupError,
+    DataMissingError,
+)
 
 logger = logging.getLogger(__name__)
 
@@ -93,7 +98,7 @@ async def login(self) -> bool:
             ) as response:
                 if not response.cookies:
                     logger.exception("Empty cookies after login, bailing out.")
-                    raise DataMissingError from None
+                    raise ConnectionSetupError from None
                 else:
                     for _, morsel in response.cookies.items():
                         # If the AIROS_ cookie was parsed but isn't automatically added to the jar, add it manually
@@ -146,15 +151,15 @@ async def login(self) -> bool:
                     logger.exception(
                         "COOKIE JAR IS EMPTY after login POST. This is a major issue."
                     )
-                    raise DataMissingError from None
+                    raise ConnectionSetupError from None
                 for cookie in self.session.cookie_jar:
                     if cookie.key.startswith("AIROS_"):
                         airos_cookie_found = True
                     if cookie.key == "ok":
                         ok_cookie_found = True
 
                 if not airos_cookie_found and not ok_cookie_found:
-                    raise DataMissingError from None
+                    raise ConnectionSetupError from None
 
                 response_text = await response.text()
 
@@ -170,16 +175,16 @@ async def login(self) -> bool:
                 else:
                     log = f"Login failed with status {response.status}. Full Response: {response.text}"
                     logger.error(log)
-                    raise ConnectionFailedError from None
+                    raise ConnectionAuthenticationError from None
         except aiohttp.ClientError as err:
             logger.exception("Error during login")
-            raise ConnectionFailedError from err
+            raise ConnectionError from err
 
     async def status(self) -> dict:
         """Retrieve status from the device."""
         if not self.connected:
             logger.error("Not connected, login first")
-            raise ConnectionFailedError from None
+            raise ConnectionError from None
 
         # --- Step 2: Verify authenticated access by fetching status.cgi ---
         authenticated_get_headers = {**self._common_headers}
@@ -205,4 +210,4 @@ async def status(self) -> dict:
                     logger.error(log)
         except aiohttp.ClientError as err:
             logger.exception("Error during authenticated status.cgi call")
-            raise ConnectionFailedError from err
+            raise ConnectionError from err

airos/exceptions.py

@@ -5,9 +5,17 @@ class AirOSException(Exception):
     """Base error class for this AirOS library."""
 
 
-class ConnectionFailedError(AirOSException):
+class ConnectionError(AirOSException):
     """Raised when unable to connect."""
 
 
+class ConnectionSetupError(AirOSException):
+    """Raised when unable to prepare authentication."""
+
+
+class ConnectionAuthenticationError(AirOSException):
+    """Raised when unable to authenticate."""
+
+
 class DataMissingError(AirOSException):
     """Raised when expected data is missing."""

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name            = "airos"
-version         = "0.0.7"
+version         = "0.0.8"
 license         = "MIT"
 description     = "Ubiquity airOS module(s) for Python 3."
 readme          = "README.md"