light.py

Go to the documentation of this file.

"""Support for the light on the Sisyphus Kinetic Art Table."""
 
from __future__ import annotations
 
import logging
from typing import Any
 
import aiohttp
 
from homeassistant.components.light import ColorMode, LightEntity
from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import PlatformNotReady
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 
from . import DATA_SISYPHUS
 
_LOGGER = logging.getLogger(__name__)
 
 
async def async_setup_platform(
    hass: HomeAssistant,
    config: ConfigType,
    add_entities: AddEntitiesCallback,
    discovery_info: DiscoveryInfoType | None = None,
) -> None:
    """Set up a single Sisyphus table."""
    if not discovery_info:
        return
    host = discovery_info[CONF_HOST]
    try:
        table_holder = hass.data[DATA_SISYPHUS][host]
        table = await table_holder.get_table()
    except aiohttp.ClientError as err:
        raise PlatformNotReady from err
 
    add_entities([SisyphusLight(table_holder.name, table)], update_before_add=True)
 
 
class SisyphusLight(LightEntity):
    """Representation of a Sisyphus table as a light."""
 
    _attr_color_mode = ColorMode.BRIGHTNESS
    _attr_supported_color_modes = {ColorMode.BRIGHTNESS}
 
    def __init__(self, name, table):
        """Initialize the Sisyphus table."""
        self._name_name = name
        self._table_table = table
 
    async def async_added_to_hass(self) -> None:
        """Add listeners after this object has been initialized."""
        self._table_table.add_listener(self.async_write_ha_stateasync_write_ha_state)
 
    async def async_update(self) -> None:
        """Force update the table state."""
        await self._table_table.refresh()
 
    @property
    def available(self) -> bool:
        """Return true if the table is responding to heartbeats."""
        return self._table_table.is_connected
 
    @property
    def unique_id(self):
        """Return the UUID of the table."""
        return self._table_table.id
 
    @property
    def name(self):
        """Return the ame of the table."""
        return self._name_name
 
    @property
    def is_on(self):
        """Return True if the table is on."""
        return not self._table_table.is_sleeping
 
    @property
    def brightness(self):
        """Return the current brightness of the table's ring light."""
        return self._table_table.brightness * 255
 
    async def async_turn_off(self, **kwargs: Any) -> None:
        """Put the table to sleep."""
        await self._table_table.sleep()
        _LOGGER.debug("Sisyphus table %s: sleep")
 
    async def async_turn_on(self, **kwargs: Any) -> None:
        """Wake up the table if necessary, optionally changes brightness."""
        if not self.is_onis_onis_on:
            await self._table_table.wakeup()
            _LOGGER.debug("Sisyphus table %s: wakeup")
 
        if "brightness" in kwargs:
            await self._table_table.set_brightness(kwargs["brightness"] / 255.0)