fix: address race condition and consider transition time when using d…

…ebounce (#167)

README.md

@@ -39,11 +39,11 @@ Some attributes such as light brightness will be rounded off. Therefore, to asse
 You can set up Stateful Scenes to restore the state of the entities when you want to turn off a scene. This can also be configured per Stateful Scene by going to the device page.  Some complex scenes might not be able to restore the state of all the entities and may benefit from configuring an opposing 'off' scene as described below.
 
 ### Transition time
-Furthermore, you can specify the default transition time for applying scenes. This will gradually change the lights of a scene to the specified state. It does need to be supported by your lights.
+Furthermore, you can specify the default transition time for applying scenes. This will gradually change the lights of a scene to the specified state. Transition time does need to be supported by your lights.
 
 ### Debounce time
 
-After activating a scene by turning on a stateful scene switch, entities may need some time to achieve their desired states. When first turned on, the scene state switch will be assumed to be 'on'; the debounce time setting controls how long this integration will wait after observing a member entity state update event before reevaluating the entity state to determine if the scene is still active. If you're having issues with scenes immediately deactivating/reactivating, consider increasing this debounce time.
+After activating a scene by turning on a stateful scene switch, entities may need some time to achieve their desired states after the transition time elapses. When first turned on, the scene state switch will be assumed to be 'on'; the debounce time setting controls how long this integration will wait after observing a member entity state update event before reevaluating the entity state to determine if the scene is still active. If you're having issues with scenes immediately deactivating/reactivating, consider increasing this debounce time.
 
 This setting is measured in seconds, but sub-second values (e.g '0.1' for 100ms delay) can be provided such that the delay is not perceptible to humans viewing a dashboard, for example.
 
@@ -61,13 +61,13 @@ Note that while all entity states are supported only some entity attributes are
 ## Scene configurations
 For each scene you can specify:
 
-- The debounce time
+- The debounce time which is applied after the transition time has elapsed
 - Whether to ignore stateful scene changes when the underlying scene is unavailable
 - Specify an opposing 'off' scene that is activated when the stateful scene is deactivated
     (when Restore is off)
 - Restore the previous state on deactivation by changing the variables on the scene's device page.
 - The scene tolerance for the stateful scene to be active
-- The individual transition time 
+- The individual transition time
 
 ## External Scenes
 > Note this is an EXPERIMENTAL feature and may not work correctly for your setup. I have tested it with scenes configured in Zigbee2MQTT which works, but I do not have access to a Hue hub which therefore may not work correctly. If you are experiencing issues, please let me know or open a pull request with the improvements. 

custom_components/stateful_scenes/StatefulScenes.py

@@ -5,6 +5,7 @@
 from typing import Any
 
 from homeassistant.core import Event, EventStateChangedData, HomeAssistant
+from homeassistant.helpers.event import async_call_later
 from homeassistant.helpers.template import area_id, area_name
 
 from .const import (
@@ -39,6 +40,75 @@ def get_entity_id_from_id(hass: HomeAssistant, id: str) -> str:
     return None
 
 
+class SceneEvaluationTimer:
+    """Manages an HA scheduled cancellable timer for transition followed by debounce."""
+
+    def __init__(
+        self,
+        hass: HomeAssistant,
+        transition_time: float = 0.0,
+        debounce_time: float = 0.0,
+    ) -> None:
+        """Initialize with no active timer."""
+        self._cancel_callback = None
+        self._transition_time = transition_time
+        self._debounce_time = debounce_time
+        self._hass = hass
+
+    async def async_start(self, callback) -> None:
+        """Start a new timer if we have a duration."""
+        await self._hass.async_add_executor_job(self.cancel_if_active)
+        if self.transition_time > 0 and self._hass is not None:
+            _LOGGER.debug(
+                "Starting scene evaluation timer for %s seconds",
+                self.transition_time + self.debounce_time,
+            )
+
+            self._cancel_callback = async_call_later(
+                self._hass,
+                self.transition_time + self.debounce_time,
+                callback,
+            )
+
+    @property
+    def transition_time(self) -> float:
+        """Get the timer duration."""
+        return self._transition_time
+
+    def set_transition_time(self, time: float) -> None:
+        """Set the timer duration."""
+        self._transition_time = time or 0.0
+
+    @property
+    def debounce_time(self) -> float:
+        """Get the timer duration."""
+        return self._debounce_time
+
+    def set_debounce_time(self, time: float) -> None:
+        """Set the timer duration."""
+        self._debounce_time = time or 0.0
+
+    def set(self, cancel_callback) -> None:
+        """Store new timer's cancel callback."""
+        self._cancel_callback = cancel_callback
+
+    def cancel_if_active(self) -> None:
+        """Cancel current timer if active."""
+        if self._cancel_callback:
+            _LOGGER.debug("Cancelling active scene evaluation timer")
+            self._cancel_callback()
+            self._cancel_callback = None
+
+    def is_active(self) -> bool:
+        """Return whether there is an active scene evaluation timer."""
+        return self._cancel_callback is not None
+
+    async def async_clear(self) -> None:
+        """Clear timer state without cancelling."""
+        _LOGGER.debug("Clearing scene evaluation timer state")
+        self._cancel_callback = None
+
+
 class Scene:
     """State scene class."""
 
@@ -53,13 +123,15 @@ def __init__(self, hass: HomeAssistant, scene_conf: dict) -> None:
         self.learn = scene_conf[CONF_SCENE_LEARN]
         self.entities = scene_conf[CONF_SCENE_ENTITIES]
         self.icon = scene_conf[CONF_SCENE_ICON]
-        self._is_on = None
+        self._is_on = False
         self._transition_time: float = 0.0
         self._restore_on_deactivate = True
-        self._debounce_time: float = 0
+        self._debounce_time: float = 0.0
         self._ignore_unavailable = False
         self._off_scene_entity_id = None
-
+        self._scene_evaluation_timer = SceneEvaluationTimer(
+            hass, self._transition_time, self._debounce_time
+        )
         self.callback = None
         self.callback_funcs = {}
         self.schedule_update = None
@@ -117,6 +189,13 @@ def turn_on(self):
         for entity_id in self.entities:
             self.store_entity_state(entity_id)
 
+        asyncio.run_coroutine_threadsafe(
+            self._scene_evaluation_timer.async_start(
+                self.async_timer_evaluate_scene_state
+            ),
+            self.hass.loop,
+        ).result()
+
         self.hass.services.call(
             domain="scene",
             service="turn_on",
@@ -142,13 +221,20 @@ def turn_off(self):
             return
 
         if self._off_scene_entity_id:
+            self._scene_evaluation_timer.cancel_if_active()
             self.hass.services.call(
                 domain="scene",
                 service="turn_on",
                 target={"entity_id": self._off_scene_entity_id},
                 service_data={"transition": self._transition_time},
             )
         elif self.restore_on_deactivate:
+            asyncio.run_coroutine_threadsafe(
+                self._scene_evaluation_timer.async_start(
+                    self.async_timer_evaluate_scene_state
+                ),
+                self.hass.loop,
+            ).result()
             self.restore()
         else:
             self.hass.services.call(
@@ -176,15 +262,17 @@ def transition_time(self) -> float:
     def set_transition_time(self, transition_time):
         """Set the transition time."""
         self._transition_time = transition_time
+        self._scene_evaluation_timer.set_transition_time(transition_time)
 
     @property
     def debounce_time(self) -> float:
         """Get the debounce time."""
-        return self._debounce_time
+        return self._scene_evaluation_timer.transition_time
 
     def set_debounce_time(self, debounce_time: float):
         """Set the debounce time."""
         self._debounce_time = debounce_time or 0.0
+        self._scene_evaluation_timer.set_debounce_time(debounce_time)
 
     @property
     def restore_on_deactivate(self) -> bool:
@@ -221,16 +309,30 @@ def unregister_callback(self):
             self.callback()
             self.callback = None
 
-    async def update_callback(self, event: Event[EventStateChangedData]):
+    def update_callback(self, event: Event[EventStateChangedData]):
         """Update the scene when a tracked entity changes state."""
         entity_id = event.data.get("entity_id")
         new_state = event.data.get("new_state")
         old_state = event.data.get("old_state")
 
         self.store_entity_state(entity_id, old_state)
         if self.is_interesting_update(old_state, new_state):
-            await asyncio.sleep(self.debounce_time)
-            self.schedule_update(True)
+            if not self._scene_evaluation_timer.is_active():
+                asyncio.run_coroutine_threadsafe(
+                    self.async_evaluate_scene_state(), self.hass.loop
+                ).result()
+
+    async def async_evaluate_scene_state(self):
+        """Evaluate scene state immediately."""
+        await self.hass.async_add_executor_job(self.check_all_states)
+        if self.schedule_update:
+            await self.hass.async_add_executor_job(self.schedule_update, True)
+
+    async def async_timer_evaluate_scene_state(self, _now):
+        """Handle Callback from HA after expiration of SceneEvaluationTimer."""
+        await self._scene_evaluation_timer.async_clear()
+        _LOGGER.debug("SceneEvaluationTimer triggered eval callback: %s", self.name)
+        await self.async_evaluate_scene_state()
 
     def is_interesting_update(self, old_state, new_state):
         """Check if the state change is interesting."""
@@ -255,7 +357,7 @@ def is_interesting_update(self, old_state, new_state):
     def check_state(self, entity_id, new_state):
         """Check if entity's current state matches the scene's defined state."""
         if new_state is None:
-            _LOGGER.warning(f"Entity not found: {entity_id}")
+            _LOGGER.warning("Entity not found: %s", entity_id)
             return False
 
         if self.ignore_unavailable and new_state.state == "unavailable":
@@ -320,8 +422,8 @@ def check_all_states(self):
 
         if not states:
             self._is_on = False
-
-        self._is_on = all(states)
+        else:
+            self._is_on = all(states)
 
     def store_entity_state(self, entity_id, state=None):
         """Store the state of an entity.

custom_components/stateful_scenes/switch.py

@@ -152,11 +152,13 @@ def turn_on(self, **kwargs) -> None:
         """
         self._scene.turn_on()
         self._is_on = self._scene.is_on
+        self.schedule_update_ha_state()
 
     def turn_off(self, **kwargs) -> None:
         """Instruct the light to turn off."""
         self._scene.turn_off()
         self._is_on = self._scene.is_on
+        self.schedule_update_ha_state()
 
     async def async_added_to_hass(self) -> None:
         """Validate and set the actual scene state on restart."""