core/homeassistant/components/kira/sensor.py

53 lines
1.6 KiB
Python

"""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 = name
self._attr_extra_state_attributes = {CONF_DEVICE: STATE_UNKNOWN}
kira.registerCallback(self._update_callback)
def _update_callback(self, code):
code_name, device = code
_LOGGER.debug("Kira Code: %s", code_name)
self._attr_native_value = code_name
self._attr_extra_state_attributes[CONF_DEVICE] = device
self.schedule_update_ha_state()