light.py

Go to the documentation of this file.

"""Support for the Elgato Avea lights."""
 
from __future__ import annotations
 
from typing import Any
 
import avea
 
from homeassistant.components.light import (
    ATTR_BRIGHTNESS,
    ATTR_HS_COLOR,
    ColorMode,
    LightEntity,
)
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import PlatformNotReady
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
import homeassistant.util.color as color_util
 
 
def setup_platform(
    hass: HomeAssistant,
    config: ConfigType,
    add_entities: AddEntitiesCallback,
    discovery_info: DiscoveryInfoType | None = None,
) -> None:
    """Set up the Avea platform."""
    try:
        nearby_bulbs = avea.discover_avea_bulbs()
        for bulb in nearby_bulbs:
            bulb.get_name()
            bulb.get_brightness()
    except OSError as err:
        raise PlatformNotReady from err
 
    add_entities(AveaLight(bulb) for bulb in nearby_bulbs)
 
 
class AveaLight(LightEntity):
    """Representation of an Avea."""
 
    _attr_color_mode = ColorMode.HS
    _attr_supported_color_modes = {ColorMode.HS}
 
    def __init__(self, light):
        """Initialize an AveaLight."""
        self._light_light = light
        self._attr_name_attr_name = light.name
        self._attr_brightness_attr_brightness = light.brightness
 
    def turn_on(self, **kwargs: Any) -> None:
        """Instruct the light to turn on."""
        if not kwargs:
            self._light_light.set_brightness(4095)
        else:
            if ATTR_BRIGHTNESS in kwargs:
                bright = round((kwargs[ATTR_BRIGHTNESS] / 255) * 4095)
                self._light_light.set_brightness(bright)
            if ATTR_HS_COLOR in kwargs:
                rgb = color_util.color_hs_to_RGB(*kwargs[ATTR_HS_COLOR])
                self._light_light.set_rgb(rgb[0], rgb[1], rgb[2])
 
    def turn_off(self, **kwargs: Any) -> None:
        """Instruct the light to turn off."""
        self._light_light.set_brightness(0)
 
    def update(self) -> None:
        """Fetch new state data for this light.
 
        This is the only method that should fetch new data for Home Assistant.
        """
        if (brightness := self._light_light.get_brightness()) is not None:
            self._attr_is_on_attr_is_on = brightness != 0
            self._attr_brightness_attr_brightness = round(255 * (brightness / 4095))