core/homeassistant/components/openuv/binary_sensor.py

"""Support for OpenUV binary sensors."""

from homeassistant.components.binary_sensor import (
    BinarySensorEntity,
    BinarySensorEntityDescription,
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
from homeassistant.util.dt import as_local

from .const import DATA_PROTECTION_WINDOW, LOGGER, TYPE_PROTECTION_WINDOW
from .coordinator import OpenUvConfigEntry
from .entity import OpenUvEntity

ATTR_PROTECTION_WINDOW_ENDING_TIME = "end_time"
ATTR_PROTECTION_WINDOW_ENDING_UV = "end_uv"
ATTR_PROTECTION_WINDOW_STARTING_TIME = "start_time"
ATTR_PROTECTION_WINDOW_STARTING_UV = "start_uv"

BINARY_SENSOR_DESCRIPTION_PROTECTION_WINDOW = BinarySensorEntityDescription(
    key=TYPE_PROTECTION_WINDOW,
    translation_key="protection_window",
)


async def async_setup_entry(
    hass: HomeAssistant,
    entry: OpenUvConfigEntry,
    async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
    """Set up OpenUV binary sensors for a config entry."""
    coordinators = entry.runtime_data

    async_add_entities(
        [
            OpenUvBinarySensor(
                coordinators[DATA_PROTECTION_WINDOW],
                BINARY_SENSOR_DESCRIPTION_PROTECTION_WINDOW,
            )
        ]
    )


class OpenUvBinarySensor(OpenUvEntity, BinarySensorEntity):
    """Define a binary sensor for OpenUV."""

    @callback
    def _handle_coordinator_update(self) -> None:
        """Update the entity from the latest data."""
        self._update_attrs()
        super()._handle_coordinator_update()

    def _update_attrs(self) -> None:
        data = self.coordinator.data

        if not data:
            LOGGER.warning("Skipping update due to missing data")
            return

        if self.entity_description.key == TYPE_PROTECTION_WINDOW:
            self._attr_is_on = data.get("is_on", False)
            self._attr_extra_state_attributes.update(
                {
                    attr_key: data[data_key]
                    for attr_key, data_key in (
                        (ATTR_PROTECTION_WINDOW_STARTING_UV, "from_uv"),
                        (ATTR_PROTECTION_WINDOW_ENDING_UV, "to_uv"),
                    )
                }
            )
            self._attr_extra_state_attributes.update(
                {
                    attr_key: as_local(data[data_key])
                    for attr_key, data_key in (
                        (ATTR_PROTECTION_WINDOW_STARTING_TIME, "from_time"),
                        (ATTR_PROTECTION_WINDOW_ENDING_TIME, "to_time"),
                    )
                }
            )