light.py

Go to the documentation of this file.

"""Support for Qwikswitch Relays and Dimmers."""
 
from __future__ import annotations
 
from homeassistant.components.light import ColorMode, LightEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 
from . import DOMAIN as QWIKSWITCH
from .entity import QSToggleEntity
 
 
async def async_setup_platform(
    hass: HomeAssistant,
    _: ConfigType,
    add_entities: AddEntitiesCallback,
    discovery_info: DiscoveryInfoType | None = None,
) -> None:
    """Add lights from the main Qwikswitch component."""
    if discovery_info is None:
        return
 
    qsusb = hass.data[QWIKSWITCH]
    devs = [QSLight(qsid, qsusb) for qsid in discovery_info[QWIKSWITCH]]
    add_entities(devs)
 
 
class QSLight(QSToggleEntity, LightEntity):
    """Light based on a Qwikswitch relay/dimmer module."""
 
    @property
    def brightness(self):
        """Return the brightness of this light (0-255)."""
        return self.devicedevice.value if self.devicedevice.is_dimmer else None
 
    @property
    def color_mode(self) -> ColorMode:
        """Return the color mode of the light."""
        return ColorMode.BRIGHTNESS if self.devicedevice.is_dimmer else ColorMode.ONOFF
 
    @property
    def supported_color_modes(self) -> set[ColorMode]:
        """Flag supported color modes."""
        return {self.color_modecolor_modecolor_mode}