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

Switch async_track_state_change to the faster async_track_state_change_event part 4 (#37863)

Browse source 
* Switch async_track_state_change to the faster async_track_state_change_event part 4

Calling async_track_state_change_event directly is faster than async_track_state_change (see #37251) since async_track_state_change is a wrapper around async_track_state_change_event now

* pylint
This commit is contained in:
J. Nick Koston 2020-07-14 19:25:48 -10:00 • committed by GitHub
parent aed98a830f
commit b430496b13
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 50 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

18
homeassistant/components/statistics/sensor.py
View file

@ -17,8 +17,12 @@ from homeassistant.const import (
STATE_UNKNOWN,
)
from homeassistant.core import callback
from homeassistant.helpers import config_validation as cv, event
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity import Entity
from homeassistant.helpers.event import (
async_track_point_in_utc_time,
async_track_state_change_event,
)
from homeassistant.util import dt as dt_util
_LOGGER = logging.getLogger(__name__)
@ -101,8 +105,12 @@ class StatisticsSensor(Entity):
"""Register callbacks."""
@callback
def async_stats_sensor_state_listener(entity, old_state, new_state):
def async_stats_sensor_state_listener(event):
"""Handle the sensor state changes."""
new_state = event.data.get("new_state")
if new_state is None:
return
self._unit_of_measurement = new_state.attributes.get(
ATTR_UNIT_OF_MEASUREMENT
)
@ -116,8 +124,8 @@ class StatisticsSensor(Entity):
"""Add listener and get recorded state."""
_LOGGER.debug("Startup for %s", self.entity_id)
event.async_track_state_change(
self.hass, self._entity_id, async_stats_sensor_state_listener
async_track_state_change_event(
self.hass, [self._entity_id], async_stats_sensor_state_listener
)
if "recorder" in self.hass.config.components:
@ -292,7 +300,7 @@ class StatisticsSensor(Entity):
self.async_schedule_update_ha_state(True)
self._update_listener = None
self._update_listener = event.async_track_point_in_utc_time(
self._update_listener = async_track_point_in_utc_time(
self.hass, _scheduled_update, next_to_purge_timestamp
)