Handle PACKAGE_NOT_PAID_FOR API error gracefully

PyViCare/PyViCareAbstractOAuthManager.py

@@ -9,6 +9,7 @@
 from PyViCare.PyViCareUtils import (PyViCareCommandError,
                                     PyViCareDeviceCommunicationError,
                                     PyViCareInternalServerError,
+                                    PyViCareNotPaidForError,
                                     PyViCareRateLimitError)
 
 logger = logging.getLogger('ViCare')
@@ -48,6 +49,7 @@ def get(self, url: str) -> Any:
             self.__handle_expired_token(response)
             self.__handle_rate_limit(response)
             self.__handle_device_communication_error(response)
+            self.__handle_not_paid_for(response)
             self.__handle_server_error(response)
             return response
         except TokenExpiredError:
@@ -72,6 +74,10 @@ def __handle_device_communication_error(self, response):
         if ("errorType" in response and response["errorType"] == "DEVICE_COMMUNICATION_ERROR"):
             raise PyViCareDeviceCommunicationError(response)
 
+    def __handle_not_paid_for(self, response):
+        if ("errorType" in response and response["errorType"] == "PACKAGE_NOT_PAID_FOR"):
+            raise PyViCareNotPaidForError(response)
+
     def __handle_server_error(self, response):
         if ("statusCode" in response and response["statusCode"] >= 500):
             raise PyViCareInternalServerError(response)

PyViCare/PyViCareCachedService.py

@@ -8,6 +8,8 @@
 from PyViCare.PyViCareUtils import (PyViCareDeviceCommunicationError,
                                     PyViCareInvalidDataError,
                                     PyViCareInternalServerError,
+                                    PyViCareNotPaidForError,
+                                    PyViCareNotSupportedFeatureError,
                                     ViCareTimer)
 
 logger = logging.getLogger('ViCare')
@@ -44,6 +46,11 @@ def __get_or_update_cache(self):
 
                 try:
                     data = self.fetch_all_features()
+                except PyViCareNotPaidForError as e:
+                    logger.error("Viessmann API denied access (PACKAGE_NOT_PAID_FOR). Features unavailable: %s", e)
+                    if self.__cache is not None:
+                        return self.__cache
+                    raise PyViCareNotSupportedFeatureError("PACKAGE_NOT_PAID_FOR")
                 except (PyViCareDeviceCommunicationError, PyViCareInternalServerError) as e:
                     if self.__cache is not None:
                         logger.warning("API error, returning stale cache: %s", e)

PyViCare/PyViCareUtils.py

@@ -105,6 +105,17 @@ class PyViCareInvalidDataError(Exception):
     pass
 
 
+class PyViCareNotPaidForError(Exception):
+    def __init__(self, response):
+        extended_payload = response.get("extendedPayload", {})
+        monetization = extended_payload.get("monetization", "Unknown")
+
+        msg = f"Feature not available: {monetization}"
+
+        super().__init__(self, msg)
+        self.message = msg
+
+
 class PyViCareDeviceCommunicationError(Exception):
     def __init__(self, response):
         extended_payload = response.get("extendedPayload", {})