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

add device class to sensors (#24373)

Browse Source
This commit is contained in:
David F. Mulcahey 2019-06-08 02:19:52 -04:00 committed by Paulus Schoutsen
parent f77514c6f2
commit 787bd75587
1 changed files with 24 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
homeassistant/components/zha/sensor.py
View File

@ -2,7 +2,10 @@
import logging import logging
from homeassistant.core import callback from homeassistant.core import callback
from homeassistant.components.sensor import DOMAIN from homeassistant.components.sensor import (
DOMAIN, DEVICE_CLASS_HUMIDITY, DEVICE_CLASS_ILLUMINANCE,
DEVICE_CLASS_TEMPERATURE, DEVICE_CLASS_PRESSURE, DEVICE_CLASS_POWER
)
from homeassistant.const import ( from homeassistant.const import (
TEMP_CELSIUS, POWER_WATT, ATTR_UNIT_OF_MEASUREMENT TEMP_CELSIUS, POWER_WATT, ATTR_UNIT_OF_MEASUREMENT
) )
@ -11,13 +14,12 @@ from .core.const import (
DATA_ZHA, DATA_ZHA_DISPATCHERS, ZHA_DISCOVERY_NEW, HUMIDITY, TEMPERATURE, DATA_ZHA, DATA_ZHA_DISPATCHERS, ZHA_DISCOVERY_NEW, HUMIDITY, TEMPERATURE,
ILLUMINANCE, PRESSURE, METERING, ELECTRICAL_MEASUREMENT, ILLUMINANCE, PRESSURE, METERING, ELECTRICAL_MEASUREMENT,
GENERIC, SENSOR_TYPE, ATTRIBUTE_CHANNEL, ELECTRICAL_MEASUREMENT_CHANNEL, GENERIC, SENSOR_TYPE, ATTRIBUTE_CHANNEL, ELECTRICAL_MEASUREMENT_CHANNEL,
SIGNAL_ATTR_UPDATED, SIGNAL_STATE_ATTR) SIGNAL_ATTR_UPDATED, SIGNAL_STATE_ATTR, UNKNOWN)
from .entity import ZhaEntity from .entity import ZhaEntity
PARALLEL_UPDATES = 5 PARALLEL_UPDATES = 5
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
# Formatter functions # Formatter functions
def pass_through_formatter(value): def pass_through_formatter(value):
"""No op update function.""" """No op update function."""
@ -91,6 +93,16 @@ FORCE_UPDATE_REGISTRY = {
ELECTRICAL_MEASUREMENT: False ELECTRICAL_MEASUREMENT: False
} }
DEVICE_CLASS_REGISTRY = {
UNKNOWN: None,
HUMIDITY: DEVICE_CLASS_HUMIDITY,
TEMPERATURE: DEVICE_CLASS_TEMPERATURE,
PRESSURE: DEVICE_CLASS_PRESSURE,
ILLUMINANCE: DEVICE_CLASS_ILLUMINANCE,
METERING: DEVICE_CLASS_POWER,
ELECTRICAL_MEASUREMENT: DEVICE_CLASS_POWER
}
async def async_setup_platform(hass, config, async_add_entities, async def async_setup_platform(hass, config, async_add_entities,
discovery_info=None): discovery_info=None):
@ -155,6 +167,10 @@ class Sensor(ZhaEntity):
self._channel = self.cluster_channels.get( self._channel = self.cluster_channels.get(
CHANNEL_REGISTRY.get(self._sensor_type, ATTRIBUTE_CHANNEL) CHANNEL_REGISTRY.get(self._sensor_type, ATTRIBUTE_CHANNEL)
) )
self._device_class = DEVICE_CLASS_REGISTRY.get(
self._sensor_type,
None
)
async def async_added_to_hass(self): async def async_added_to_hass(self):
"""Run when about to be added to hass.""" """Run when about to be added to hass."""
@ -165,6 +181,11 @@ class Sensor(ZhaEntity):
self._channel, SIGNAL_STATE_ATTR, self._channel, SIGNAL_STATE_ATTR,
self.async_update_state_attribute) self.async_update_state_attribute)
@property
def device_class(self) -> str:
"""Return device class from component DEVICE_CLASSES."""
return self._device_class
@property @property
def unit_of_measurement(self): def unit_of_measurement(self):
"""Return the unit of measurement of this entity.""" """Return the unit of measurement of this entity."""