From 2109444ba53d238a90edf2833b30ef340cc4fa25 Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Fri, 28 Aug 2020 09:32:05 -0500 Subject: [PATCH] Update time triggers to use async_track_state_change_event (#39338) This was one of the ones missed in the async_track_state_change to async_track_state_change_event conversion. --- .../components/homeassistant/triggers/time.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/homeassistant/triggers/time.py b/homeassistant/components/homeassistant/triggers/time.py index 76acaf89c6c..01eb1a9b85f 100644 --- a/homeassistant/components/homeassistant/triggers/time.py +++ b/homeassistant/components/homeassistant/triggers/time.py @@ -9,7 +9,7 @@ from homeassistant.core import callback from homeassistant.helpers import config_validation as cv from homeassistant.helpers.event import ( async_track_point_in_time, - async_track_state_change, + async_track_state_change_event, async_track_time_change, ) import homeassistant.util.dt as dt_util @@ -43,7 +43,13 @@ async def async_attach_trigger(hass, config, action, automation_info): hass.async_run_job(action, {"trigger": {"platform": "time", "now": now}}) @callback - def update_entity_trigger(entity_id, old_state=None, new_state=None): + def update_entity_trigger_event(event): + """update_entity_trigger from the event.""" + return update_entity_trigger(event.data["entity_id"], event.data["new_state"]) + + @callback + def update_entity_trigger(entity_id, new_state=None): + """Update the entity trigger for the entity_id.""" # If a listener was already set up for entity, remove it. remove = entities.get(entity_id) if remove: @@ -111,7 +117,9 @@ async def async_attach_trigger(hass, config, action, automation_info): # Track state changes of any entities. removes.append( - async_track_state_change(hass, list(entities), update_entity_trigger) + async_track_state_change_event( + hass, list(entities), update_entity_trigger_event + ) ) @callback