evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update time triggers to use async_track_state_change_event (#39338)

Browse source 
This was one of the ones missed in the
async_track_state_change to
async_track_state_change_event conversion.
This commit is contained in:
J. Nick Koston 2020-08-28 09:32:05 -05:00 • committed by GitHub
parent f8704a2dfc
commit 2109444ba5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

14
homeassistant/components/homeassistant/triggers/time.py
View file

@ -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