light.py

Go to the documentation of this file.

"""Support for Homekit lights."""
 
from __future__ import annotations
 
from typing import Any
 
from aiohomekit.model.characteristics import CharacteristicsTypes
from aiohomekit.model.services import Service, ServicesTypes
from propcache import cached_property
 
from homeassistant.components.light import (
    ATTR_BRIGHTNESS,
    ATTR_COLOR_TEMP,
    ATTR_HS_COLOR,
    ColorMode,
    LightEntity,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity_platform import AddEntitiesCallback
import homeassistant.util.color as color_util
 
from . import KNOWN_DEVICES
from .connection import HKDevice
from .entity import HomeKitEntity
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    config_entry: ConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up Homekit lightbulb."""
    hkid: str = config_entry.data["AccessoryPairingID"]
    conn: HKDevice = hass.data[KNOWN_DEVICES][hkid]
 
    @callback
    def async_add_service(service: Service) -> bool:
        if service.type != ServicesTypes.LIGHTBULB:
            return False
        info = {"aid": service.accessory.aid, "iid": service.iid}
        entity = HomeKitLight(conn, info)
        conn.async_migrate_unique_id(
            entity.old_unique_id, entity.unique_id, Platform.LIGHT
        )
        async_add_entities([entity])
        return True
 
    conn.add_listener(async_add_service)
 
 
class HomeKitLight(HomeKitEntity, LightEntity):
    """Representation of a Homekit light."""
 
    @callback
    def _async_reconfigure(self) -> None:
        """Reconfigure entity."""
        self._async_clear_property_cache_async_clear_property_cache(
            ("supported_features", "min_mireds", "max_mireds", "supported_color_modes")
        )
        super()._async_reconfigure()
 
    def get_characteristic_types(self) -> list[str]:
        """Define the homekit characteristics the entity cares about."""
        return [
            CharacteristicsTypes.ON,
            CharacteristicsTypes.BRIGHTNESS,
            CharacteristicsTypes.COLOR_TEMPERATURE,
            CharacteristicsTypes.HUE,
            CharacteristicsTypes.SATURATION,
        ]
 
    @property
    def is_on(self) -> bool:
        """Return true if device is on."""
        return self.serviceservice.value(CharacteristicsTypes.ON)
 
    @property
    def brightness(self) -> int:
        """Return the brightness of this light between 0..255."""
        return self.serviceservice.value(CharacteristicsTypes.BRIGHTNESS) * 255 / 100
 
    @property
    def hs_color(self) -> tuple[float, float]:
        """Return the color property."""
        return (
            self.serviceservice.value(CharacteristicsTypes.HUE),
            self.serviceservice.value(CharacteristicsTypes.SATURATION),
        )
 
    @cached_property
    def min_mireds(self) -> int:
        """Return minimum supported color temperature."""
        if not self.serviceservice.has(CharacteristicsTypes.COLOR_TEMPERATURE):
            return super().min_mireds
        min_value = self.serviceservice[CharacteristicsTypes.COLOR_TEMPERATURE].minValue
        return int(min_value) if min_value else super().min_mireds
 
    @cached_property
    def max_mireds(self) -> int:
        """Return the maximum color temperature."""
        if not self.serviceservice.has(CharacteristicsTypes.COLOR_TEMPERATURE):
            return super().max_mireds
        max_value = self.serviceservice[CharacteristicsTypes.COLOR_TEMPERATURE].maxValue
        return int(max_value) if max_value else super().max_mireds
 
    @property
    def color_temp(self) -> int:
        """Return the color temperature."""
        return self.serviceservice.value(CharacteristicsTypes.COLOR_TEMPERATURE)
 
    @property
    def color_mode(self) -> str:
        """Return the color mode of the light."""
        # aiohomekit does not keep track of the light's color mode, report
        # hs for light supporting both hs and ct
        if self.serviceservice.has(CharacteristicsTypes.HUE) or self.serviceservice.has(
            CharacteristicsTypes.SATURATION
        ):
            return ColorMode.HS
 
        if self.serviceservice.has(CharacteristicsTypes.COLOR_TEMPERATURE):
            return ColorMode.COLOR_TEMP
 
        if self.serviceservice.has(CharacteristicsTypes.BRIGHTNESS):
            return ColorMode.BRIGHTNESS
 
        return ColorMode.ONOFF
 
    @cached_property
    def supported_color_modes(self) -> set[ColorMode]:
        """Flag supported color modes."""
        color_modes: set[ColorMode] = set()
 
        if self.serviceservice.has(CharacteristicsTypes.HUE) or self.serviceservice.has(
            CharacteristicsTypes.SATURATION
        ):
            color_modes.add(ColorMode.HS)
            color_modes.add(ColorMode.COLOR_TEMP)
 
        elif self.serviceservice.has(CharacteristicsTypes.COLOR_TEMPERATURE):
            color_modes.add(ColorMode.COLOR_TEMP)
 
        if not color_modes and self.serviceservice.has(CharacteristicsTypes.BRIGHTNESS):
            color_modes.add(ColorMode.BRIGHTNESS)
 
        if not color_modes:
            color_modes.add(ColorMode.ONOFF)
 
        return color_modes
 
    async def async_turn_on(self, **kwargs: Any) -> None:
        """Turn the specified light on."""
        hs_color = kwargs.get(ATTR_HS_COLOR)
        temperature = kwargs.get(ATTR_COLOR_TEMP)
        brightness = kwargs.get(ATTR_BRIGHTNESS)
 
        characteristics: dict[str, Any] = {}
 
        if brightness is not None:
            characteristics[CharacteristicsTypes.BRIGHTNESS] = int(
                brightness * 100 / 255
            )
 
        # If they send both temperature and hs_color, and the device
        # does not support both, temperature will win. This is not
        # expected to happen in the UI, but it is possible via a manual
        # service call.
        if temperature is not None:
            if self.serviceservice.has(CharacteristicsTypes.COLOR_TEMPERATURE):
                characteristics[CharacteristicsTypes.COLOR_TEMPERATURE] = int(
                    temperature
                )
            elif hs_color is None:
                # Some HomeKit devices implement color temperature with HS
                # since the spec "technically" does not permit the COLOR_TEMPERATURE
                # characteristic and the HUE and SATURATION characteristics to be
                # present at the same time.
                hue_sat = color_util.color_temperature_to_hs(
                    color_util.color_temperature_mired_to_kelvin(temperature)
                )
                characteristics[CharacteristicsTypes.HUE] = hue_sat[0]
                characteristics[CharacteristicsTypes.SATURATION] = hue_sat[1]
 
        if hs_color is not None:
            characteristics[CharacteristicsTypes.HUE] = hs_color[0]
            characteristics[CharacteristicsTypes.SATURATION] = hs_color[1]
 
        characteristics[CharacteristicsTypes.ON] = True
 
        await self.async_put_characteristicsasync_put_characteristics(characteristics)
 
    async def async_turn_off(self, **kwargs: Any) -> None:
        """Turn the specified light off."""
        await self.async_put_characteristicsasync_put_characteristics({CharacteristicsTypes.ON: False})