51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
|
"""Class to hold all sensor accessories."""
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from homeassistant.const import STATE_UNKNOWN
|
||
|
|
from homeassistant.helpers.event import async_track_state_change
|
||
|
|
|
||
|
|
from . import TYPES
|
||
|
|
from .accessories import HomeAccessory
|
||
|
|
from .const import (
|
||
|
|
SERVICES_TEMPERATURE_SENSOR, CHAR_CURRENT_TEMPERATURE)
|
||
|
|
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
@TYPES.register('TemperatureSensor')
|
||
|
|
class TemperatureSensor(HomeAccessory):
|
||
|
|
"""Generate a TemperatureSensor accessory for a temperature sensor.
|
||
|
|
|
||
|
|
Sensor entity must return either temperature in °C or STATE_UNKNOWN.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self, hass, entity_id, display_name):
|
||
|
|
"""Initialize a TemperatureSensor accessory object."""
|
||
|
|
super().__init__(display_name)
|
||
|
|
self.set_category(self.ALL_CATEGORIES.SENSOR)
|
||
|
|
self.set_accessory_info(entity_id)
|
||
|
|
self.add_preload_service(SERVICES_TEMPERATURE_SENSOR)
|
||
|
|
|
||
|
|
self._hass = hass
|
||
|
|
self._entity_id = entity_id
|
||
|
|
|
||
|
|
self.service_temp = self.get_service(SERVICES_TEMPERATURE_SENSOR)
|
||
|
|
self.char_temp = self.service_temp. \
|
||
|
|
get_characteristic(CHAR_CURRENT_TEMPERATURE)
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
"""Method called be object after driver is started."""
|
||
|
|
state = self._hass.states.get(self._entity_id)
|
||
|
|
self.update_temperature(new_state=state)
|
||
|
|
|
||
|
|
async_track_state_change(
|
||
|
|
self._hass, self._entity_id, self.update_temperature)
|
||
|
|
|
||
|
|
def update_temperature(self, entity_id=None, old_state=None,
|
||
|
|
new_state=None):
|
||
|
|
"""Update temperature after state changed."""
|
||
|
|
temperature = new_state.state
|
||
|
|
if temperature != STATE_UNKNOWN:
|
||
|
|
self.char_temp.set_value(float(temperature))
|