light.py

Go to the documentation of this file.

"""Support for Crownstone devices."""
 
from __future__ import annotations
 
from functools import partial
from typing import TYPE_CHECKING, Any
 
from crownstone_cloud.cloud_models.crownstones import Crownstone
from crownstone_cloud.const import DIMMING_ABILITY
from crownstone_cloud.exceptions import CrownstoneAbilityError
from crownstone_uart import CrownstoneUart
 
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
from .const import (
    CROWNSTONE_INCLUDE_TYPES,
    CROWNSTONE_SUFFIX,
    DOMAIN,
    SIG_CROWNSTONE_STATE_UPDATE,
    SIG_UART_STATE_CHANGE,
)
from .entity import CrownstoneEntity
from .helpers import map_from_to
 
if TYPE_CHECKING:
    from .entry_manager import CrownstoneEntryManager
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    config_entry: ConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up crownstones from a config entry."""
    manager: CrownstoneEntryManager = hass.data[DOMAIN][config_entry.entry_id]
 
    entities: list[CrownstoneLightEntity] = []
 
    # Add Crownstone entities that support switching/dimming
    for sphere in manager.cloud.cloud_data:
        for crownstone in sphere.crownstones:
            if crownstone.type in CROWNSTONE_INCLUDE_TYPES:
                # Crownstone can communicate with Crownstone USB
                if manager.uart and sphere.cloud_id == manager.usb_sphere_id:
                    entities.append(CrownstoneLightEntity(crownstone, manager.uart))
                # Crownstone can't communicate with Crownstone USB
                else:
                    entities.append(CrownstoneLightEntity(crownstone))
 
    async_add_entities(entities)
 
 
def crownstone_state_to_hass(value: int) -> int:
    """Crownstone 0..100 to hass 0..255."""
    return map_from_to(value, 0, 100, 0, 255)
 
 
def hass_to_crownstone_state(value: int) -> int:
    """Hass 0..255 to Crownstone 0..100."""
    return map_from_to(value, 0, 255, 0, 100)
 
 
class CrownstoneLightEntity(CrownstoneEntity, LightEntity):
    """Representation of a crownstone.
 
    Light platform is used to support dimming.
    """
 
    _attr_name = None
    _attr_translation_key = "german_power_outlet"
 
    def __init__(
        self, crownstone_data: Crownstone, usb: CrownstoneUart | None = None
    ) -> None:
        """Initialize the crownstone."""
        super().__init__(crownstone_data)
        self.usbusb = usb
        # Entity class attributes
        self._attr_unique_id_attr_unique_id = f"{self.cloud_id}-{CROWNSTONE_SUFFIX}"
 
    @property
    def brightness(self) -> int | None:
        """Return the brightness if dimming enabled."""
        return crownstone_state_to_hass(self.devicedevice.state)
 
    @property
    def is_on(self) -> bool:
        """Return if the device is on."""
        return crownstone_state_to_hass(self.devicedevice.state) > 0
 
    @property
    def color_mode(self) -> str:
        """Return the color mode of the light."""
        if self.devicedevice.abilities.get(DIMMING_ABILITY).is_enabled:
            return ColorMode.BRIGHTNESS
        return ColorMode.ONOFF
 
    @property
    def supported_color_modes(self) -> set[str] | None:
        """Flag supported color modes."""
        return {self.color_modecolor_modecolor_mode}
 
    async def async_added_to_hass(self) -> None:
        """Set up a listener when this entity is added to HA."""
        # new state received
        self.async_on_removeasync_on_remove(
            async_dispatcher_connect(
                self.hasshass, SIG_CROWNSTONE_STATE_UPDATE, self.async_write_ha_stateasync_write_ha_state
            )
        )
        # updates state attributes when usb connects/disconnects
        self.async_on_removeasync_on_remove(
            async_dispatcher_connect(
                self.hasshass, SIG_UART_STATE_CHANGE, self.async_write_ha_stateasync_write_ha_state
            )
        )
 
    async def async_turn_on(self, **kwargs: Any) -> None:
        """Turn on this light via dongle or cloud."""
        if ATTR_BRIGHTNESS in kwargs:
            if self.usbusb is not None and self.usbusb.is_ready():
                await self.hasshass.async_add_executor_job(
                    partial(
                        self.usbusb.dim_crownstone,
                        self.devicedevice.unique_id,
                        hass_to_crownstone_state(kwargs[ATTR_BRIGHTNESS]),
                    )
                )
            else:
                try:
                    await self.devicedevice.async_set_brightness(
                        hass_to_crownstone_state(kwargs[ATTR_BRIGHTNESS])
                    )
                except CrownstoneAbilityError as ability_error:
                    raise HomeAssistantError(ability_error) from ability_error
 
            # assume brightness is set on device
            self.devicedevice.state = hass_to_crownstone_state(kwargs[ATTR_BRIGHTNESS])
            self.async_write_ha_stateasync_write_ha_state()
 
        elif self.usbusb is not None and self.usbusb.is_ready():
            await self.hasshass.async_add_executor_job(
                partial(self.usbusb.switch_crownstone, self.devicedevice.unique_id, on=True)
            )
            self.devicedevice.state = 100
            self.async_write_ha_stateasync_write_ha_state()
 
        else:
            await self.devicedevice.async_turn_on()
            self.devicedevice.state = 100
            self.async_write_ha_stateasync_write_ha_state()
 
    async def async_turn_off(self, **kwargs: Any) -> None:
        """Turn off this device via dongle or cloud."""
        if self.usbusb is not None and self.usbusb.is_ready():
            await self.hasshass.async_add_executor_job(
                partial(self.usbusb.switch_crownstone, self.devicedevice.unique_id, on=False)
            )
 
        else:
            await self.devicedevice.async_turn_off()
 
        self.devicedevice.state = 0
        self.async_write_ha_stateasync_write_ha_state()