"""Support for KNX/IP time.""" from __future__ import annotations from datetime import time as dt_time from xknx import XKNX from xknx.devices import TimeDevice as XknxTimeDevice from xknx.dpt.dpt_10 import KNXTime as XknxTime from homeassistant import config_entries from homeassistant.components.time import TimeEntity from homeassistant.const import ( CONF_ENTITY_CATEGORY, CONF_NAME, STATE_UNAVAILABLE, STATE_UNKNOWN, Platform, ) from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.restore_state import RestoreEntity from homeassistant.helpers.typing import ConfigType from . import KNXModule from .const import ( CONF_RESPOND_TO_READ, CONF_STATE_ADDRESS, CONF_SYNC_STATE, KNX_ADDRESS, KNX_MODULE_KEY, ) from .knx_entity import KnxYamlEntity async def async_setup_entry( hass: HomeAssistant, config_entry: config_entries.ConfigEntry, async_add_entities: AddEntitiesCallback, ) -> None: """Set up entities for KNX platform.""" knx_module = hass.data[KNX_MODULE_KEY] config: list[ConfigType] = knx_module.config_yaml[Platform.TIME] async_add_entities( KNXTimeEntity(knx_module, entity_config) for entity_config in config ) def _create_xknx_device(xknx: XKNX, config: ConfigType) -> XknxTimeDevice: """Return a XKNX DateTime object to be used within XKNX.""" return XknxTimeDevice( xknx, name=config[CONF_NAME], localtime=False, group_address=config[KNX_ADDRESS], group_address_state=config.get(CONF_STATE_ADDRESS), respond_to_read=config[CONF_RESPOND_TO_READ], sync_state=config[CONF_SYNC_STATE], ) class KNXTimeEntity(KnxYamlEntity, TimeEntity, RestoreEntity): """Representation of a KNX time.""" _device: XknxTimeDevice def __init__(self, knx_module: KNXModule, config: ConfigType) -> None: """Initialize a KNX time.""" super().__init__( knx_module=knx_module, device=_create_xknx_device(knx_module.xknx, config), ) self._attr_entity_category = config.get(CONF_ENTITY_CATEGORY) self._attr_unique_id = str(self._device.remote_value.group_address) async def async_added_to_hass(self) -> None: """Restore last state.""" await super().async_added_to_hass() if ( not self._device.remote_value.readable and (last_state := await self.async_get_last_state()) is not None and last_state.state not in (STATE_UNKNOWN, STATE_UNAVAILABLE) ): self._device.remote_value.value = XknxTime.from_time( dt_time.fromisoformat(last_state.state) ) @property def native_value(self) -> dt_time | None: """Return the latest value.""" return self._device.value async def async_set_value(self, value: dt_time) -> None: """Change the value.""" await self._device.set(value)