mirror of https://github.com/home-assistant/core
98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
"""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
|
|
self._table = table
|
|
|
|
async def async_added_to_hass(self) -> None:
|
|
"""Add listeners after this object has been initialized."""
|
|
self._table.add_listener(self.async_write_ha_state)
|
|
|
|
async def async_update(self) -> None:
|
|
"""Force update the table state."""
|
|
await self._table.refresh()
|
|
|
|
@property
|
|
def available(self) -> bool:
|
|
"""Return true if the table is responding to heartbeats."""
|
|
return self._table.is_connected
|
|
|
|
@property
|
|
def unique_id(self):
|
|
"""Return the UUID of the table."""
|
|
return self._table.id
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the ame of the table."""
|
|
return self._name
|
|
|
|
@property
|
|
def is_on(self):
|
|
"""Return True if the table is on."""
|
|
return not self._table.is_sleeping
|
|
|
|
@property
|
|
def brightness(self):
|
|
"""Return the current brightness of the table's ring light."""
|
|
return self._table.brightness * 255
|
|
|
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
|
"""Put the table to sleep."""
|
|
await self._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_on:
|
|
await self._table.wakeup()
|
|
_LOGGER.debug("Sisyphus table %s: wakeup")
|
|
|
|
if "brightness" in kwargs:
|
|
await self._table.set_brightness(kwargs["brightness"] / 255.0)
|