sensor.py

Go to the documentation of this file.

"""KIRA interface to receive UDP packets from an IR-IP bridge."""
 
from __future__ import annotations
 
import logging
 
from homeassistant.components.sensor import SensorEntity
from homeassistant.const import CONF_DEVICE, CONF_NAME, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 
from . import CONF_SENSOR, DOMAIN
 
_LOGGER = logging.getLogger(__name__)
 
 
def setup_platform(
    hass: HomeAssistant,
    config: ConfigType,
    add_entities: AddEntitiesCallback,
    discovery_info: DiscoveryInfoType | None = None,
) -> None:
    """Set up a Kira sensor."""
    if discovery_info is not None:
        name = discovery_info.get(CONF_NAME)
        device = discovery_info.get(CONF_DEVICE)
        kira = hass.data[DOMAIN][CONF_SENSOR][name]
 
        add_entities([KiraReceiver(device, kira)])
 
 
class KiraReceiver(SensorEntity):
    """Implementation of a Kira Receiver."""
 
    _attr_force_update = True  # repeated states have meaning in Kira
    _attr_icon = "mdi:remote"
    _attr_should_poll = False
 
    def __init__(self, name, kira):
        """Initialize the sensor."""
        self._attr_name_attr_name = name
        self._attr_extra_state_attributes_attr_extra_state_attributes = {CONF_DEVICE: STATE_UNKNOWN}
 
        kira.registerCallback(self._update_callback_update_callback)
 
    def _update_callback(self, code):
        code_name, device = code
        _LOGGER.debug("Kira Code: %s", code_name)
        self._attr_native_value_attr_native_value = code_name
        self._attr_extra_state_attributes_attr_extra_state_attributes[CONF_DEVICE] = device
        self.schedule_update_ha_stateschedule_update_ha_state()