jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 11:17:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Migrate to async (sensor.statistics) (#4138)

Browse Source 
* Migrate to async

* Add async_ prefix and remove stale print
This commit is contained in:
Fabian Affolter 2016-10-30 18:56:26 +01:00 committed by Pascal Vizeli
parent 9649097b32
commit b910a9917d
1 changed files with 18 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
homeassistant/components/sensor/statistics.py
View File

@ -4,18 +4,20 @@ Support for statistics for sensor values.
For more details about this platform, please refer to the documentation at For more details about this platform, please refer to the documentation at
https://home-assistant.io/components/sensor.statistics/ https://home-assistant.io/components/sensor.statistics/
""" """
import asyncio
import logging import logging
import statistics import statistics
from collections import deque from collections import deque
import voluptuous as vol import voluptuous as vol
import homeassistant.helpers.config_validation as cv
from homeassistant.components.sensor import PLATFORM_SCHEMA from homeassistant.components.sensor import PLATFORM_SCHEMA
from homeassistant.const import ( from homeassistant.const import (
CONF_NAME, CONF_ENTITY_ID, STATE_UNKNOWN, ATTR_UNIT_OF_MEASUREMENT) CONF_NAME, CONF_ENTITY_ID, STATE_UNKNOWN, ATTR_UNIT_OF_MEASUREMENT)
import homeassistant.helpers.config_validation as cv from homeassistant.core import callback
from homeassistant.helpers.entity import Entity from homeassistant.helpers.entity import Entity
from homeassistant.helpers.event import track_state_change from homeassistant.helpers.event import async_track_state_change
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -41,13 +43,16 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
}) })
def setup_platform(hass, config, add_devices, discovery_info=None): @asyncio.coroutine
"""Setup the Statistics sensor.""" def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
"""Set up the Statistics sensor."""
entity_id = config.get(CONF_ENTITY_ID) entity_id = config.get(CONF_ENTITY_ID)
name = config.get(CONF_NAME) name = config.get(CONF_NAME)
sampling_size = config.get(CONF_SAMPLING_SIZE) sampling_size = config.get(CONF_SAMPLING_SIZE)
add_devices([StatisticsSensor(hass, entity_id, name, sampling_size)]) hass.loop.create_task(async_add_devices(
[StatisticsSensor(hass, entity_id, name, sampling_size)], True))
return True
# pylint: disable=too-many-instance-attributes # pylint: disable=too-many-instance-attributes
@ -72,9 +77,10 @@ class StatisticsSensor(Entity):
self.states = deque(maxlen=self._sampling_size) self.states = deque(maxlen=self._sampling_size)
self.median = self.mean = self.variance = self.stdev = 0 self.median = self.mean = self.variance = self.stdev = 0
self.min = self.max = self.total = self.count = 0 self.min = self.max = self.total = self.count = 0
self.update()
def stats_sensor_state_listener(entity, old_state, new_state): @callback
# pylint: disable=invalid-name
def async_stats_sensor_state_listener(entity, old_state, new_state):
"""Called when the sensor changes state.""" """Called when the sensor changes state."""
self._unit_of_measurement = new_state.attributes.get( self._unit_of_measurement = new_state.attributes.get(
ATTR_UNIT_OF_MEASUREMENT) ATTR_UNIT_OF_MEASUREMENT)
@ -85,9 +91,10 @@ class StatisticsSensor(Entity):
except ValueError: except ValueError:
self.count = self.count + 1 self.count = self.count + 1
self.update_ha_state(True) hass.loop.create_task(self.async_update_ha_state(True))
track_state_change(hass, entity_id, stats_sensor_state_listener) async_track_state_change(
hass, entity_id, async_stats_sensor_state_listener)
@property @property
def name(self): def name(self):
@ -131,7 +138,8 @@ class StatisticsSensor(Entity):
"""Return the icon to use in the frontend, if any.""" """Return the icon to use in the frontend, if any."""
return ICON return ICON
def update(self): @asyncio.coroutine
def async_update(self):
"""Get the latest data and updates the states.""" """Get the latest data and updates the states."""
if not self.is_binary: if not self.is_binary:
try: try: