Add availability test to context and set availability in coordinator

Author: PimDoos

custom_components/sessy/coordinator.py

@@ -25,7 +25,7 @@
 )
 from sessypy.util import SessyLoginException, SessyNotSupportedException
 
-from typing import Callable, Optional
+from typing import Any, Callable, Optional
 
 from .const import (
     COORDINATOR_RETRIES,
@@ -268,7 +268,7 @@ def __init__(
         availability_key: str = None,
         availability_test_value: str = None
     ):
-        self.context = SessyEntityContext(data_key, transform_function)
+        self.context = SessyEntityContext(data_key, transform_function, availability_key, availability_test_value)
         super().__init__(coordinator, self.context)
         self.hass = hass
         self.config_entry = config_entry
@@ -310,7 +310,10 @@ def _handle_coordinator_update(self) -> None:
             self.async_write_ha_state()
 
     def copy_from_cache(self):
-        self.cache_value, self._attr_available = self.coordinator.data.get(self.context, None)
+        value, available = self.coordinator.data.get(self.context, tuple((None, False)))
+        self.cache_value = value
+        self._attr_available = available
+
         if self.cache_value is None:
             raise TypeError(
                 f"Key {self.data_key} has no value in coordinator {self.coordinator.name}"
@@ -319,6 +322,11 @@ def copy_from_cache(self):
     def update_from_cache(self):
         """Entity function to write the latest cache value to the proper attributes. Implemented on platform level."""
         raise NotImplementedError()
+    
+    @property
+    def available(self) -> bool:
+        """Return True if entity is available."""
+        return super().available and self._attr_available
 
 
 class SessyEntityContext:
@@ -328,7 +336,7 @@ def __init__(self, data_key: str, transform_function: Optional[Callable] = None,
         self.availability_key = availability_key
         self.availability_test_value = availability_test_value
 
-    def apply(self, data):
+    def apply(self, data) -> tuple[Any, bool]:
         value = get_nested_key(data, self.data_key)
         if self.transform_function:
             value = self.transform_function(value)