light.py

Go to the documentation of this file.

"""Support for IHC lights."""
 
from __future__ import annotations
 
from typing import Any
 
from ihcsdk.ihccontroller import IHCController
 
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 
from .const import CONF_DIMMABLE, CONF_OFF_ID, CONF_ON_ID, DOMAIN, IHC_CONTROLLER
from .entity import IHCEntity
from .util import async_pulse, async_set_bool, async_set_int
 
 
def setup_platform(
    hass: HomeAssistant,
    config: ConfigType,
    add_entities: AddEntitiesCallback,
    discovery_info: DiscoveryInfoType | None = None,
) -> None:
    """Set up the IHC lights platform."""
    if discovery_info is None:
        return
    devices = []
    for name, device in discovery_info.items():
        ihc_id = device["ihc_id"]
        product_cfg = device["product_cfg"]
        product = device["product"]
        # Find controller that corresponds with device id
        controller_id = device["ctrl_id"]
        ihc_controller: IHCController = hass.data[DOMAIN][controller_id][IHC_CONTROLLER]
        ihc_off_id = product_cfg.get(CONF_OFF_ID)
        ihc_on_id = product_cfg.get(CONF_ON_ID)
        dimmable = product_cfg[CONF_DIMMABLE]
        light = IhcLight(
            ihc_controller,
            controller_id,
            name,
            ihc_id,
            ihc_off_id,
            ihc_on_id,
            dimmable,
            product,
        )
        devices.append(light)
    add_entities(devices)
 
 
class IhcLight(IHCEntity, LightEntity):
    """Representation of a IHC light.
 
    For dimmable lights, the associated IHC resource should be a light
    level (integer). For non dimmable light the IHC resource should be
    an on/off (boolean) resource
    """
 
    def __init__(
        self,
        ihc_controller: IHCController,
        controller_id: str,
        name: str,
        ihc_id: int,
        ihc_off_id: int,
        ihc_on_id: int,
        dimmable=False,
        product=None,
    ) -> None:
        """Initialize the light."""
        super().__init__(ihc_controller, controller_id, name, ihc_id, product)
        self._ihc_off_id_ihc_off_id = ihc_off_id
        self._ihc_on_id_ihc_on_id = ihc_on_id
        self._brightness_brightness = 0
        self._dimmable_dimmable = dimmable
        self._state_state = False
 
        if self._dimmable_dimmable:
            self._attr_color_mode_attr_color_mode = ColorMode.BRIGHTNESS
        else:
            self._attr_color_mode_attr_color_mode = ColorMode.ONOFF
        self._attr_supported_color_modes_attr_supported_color_modes = {self._attr_color_mode_attr_color_mode}
 
    @property
    def brightness(self) -> int:
        """Return the brightness of this light between 0..255."""
        return self._brightness_brightness
 
    @property
    def is_on(self) -> bool:
        """Return true if light is on."""
        return self._state_state
 
    async def async_turn_on(self, **kwargs: Any) -> None:
        """Turn the light on."""
        if ATTR_BRIGHTNESS in kwargs:
            brightness = kwargs[ATTR_BRIGHTNESS]
        elif (brightness := self._brightness_brightness) == 0:
            brightness = 255
 
        if self._dimmable_dimmable:
            await async_set_int(
                self.hasshass, self.ihc_controllerihc_controller, self.ihc_idihc_id, int(brightness * 100 / 255)
            )
        elif self._ihc_on_id_ihc_on_id:
            await async_pulse(self.hasshass, self.ihc_controllerihc_controller, self._ihc_on_id_ihc_on_id)
        else:
            await async_set_bool(self.hasshass, self.ihc_controllerihc_controller, self.ihc_idihc_id, True)
 
    async def async_turn_off(self, **kwargs: Any) -> None:
        """Turn the light off."""
        if self._dimmable_dimmable:
            await async_set_int(self.hasshass, self.ihc_controllerihc_controller, self.ihc_idihc_id, 0)
        elif self._ihc_off_id_ihc_off_id:
            await async_pulse(self.hasshass, self.ihc_controllerihc_controller, self._ihc_off_id_ihc_off_id)
        else:
            await async_set_bool(self.hasshass, self.ihc_controllerihc_controller, self.ihc_idihc_id, False)
 
    def on_ihc_change(self, ihc_id, value):
        """Handle IHC notifications."""
        if isinstance(value, bool):
            self._dimmable_dimmable = False
            self._state_state = value != 0
        else:
            self._dimmable_dimmable = True
            self._state_state = value > 0
            if self._state_state:
                self._brightness_brightness = int(value * 255 / 100)
        self.schedule_update_ha_stateschedule_update_ha_state()