107 lines
3.3 KiB
Python
107 lines
3.3 KiB
Python
"""Support for Lutron Homeworks lights."""
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Any
|
|
|
|
from pyhomeworks.pyhomeworks import HW_LIGHT_CHANGED
|
|
|
|
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
|
|
from homeassistant.const import CONF_NAME
|
|
from homeassistant.core import HomeAssistant, callback
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
|
|
|
|
from . import CONF_ADDR, CONF_DIMMERS, CONF_RATE, HOMEWORKS_CONTROLLER, HomeworksDevice
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
def setup_platform(
|
|
hass: HomeAssistant,
|
|
config: ConfigType,
|
|
add_entities: AddEntitiesCallback,
|
|
discover_info: DiscoveryInfoType | None = None,
|
|
) -> None:
|
|
"""Set up Homeworks lights."""
|
|
if discover_info is None:
|
|
return
|
|
|
|
controller = hass.data[HOMEWORKS_CONTROLLER]
|
|
devs = []
|
|
for dimmer in discover_info[CONF_DIMMERS]:
|
|
dev = HomeworksLight(
|
|
controller, dimmer[CONF_ADDR], dimmer[CONF_NAME], dimmer[CONF_RATE]
|
|
)
|
|
devs.append(dev)
|
|
add_entities(devs, True)
|
|
|
|
|
|
class HomeworksLight(HomeworksDevice, LightEntity):
|
|
"""Homeworks Light."""
|
|
|
|
_attr_color_mode = ColorMode.BRIGHTNESS
|
|
_attr_supported_color_modes = {ColorMode.BRIGHTNESS}
|
|
|
|
def __init__(self, controller, addr, name, rate):
|
|
"""Create device with Addr, name, and rate."""
|
|
super().__init__(controller, addr, name)
|
|
self._rate = rate
|
|
self._level = 0
|
|
self._prev_level = 0
|
|
|
|
async def async_added_to_hass(self) -> None:
|
|
"""Call when entity is added to hass."""
|
|
signal = f"homeworks_entity_{self._addr}"
|
|
_LOGGER.debug("connecting %s", signal)
|
|
self.async_on_remove(
|
|
async_dispatcher_connect(self.hass, signal, self._update_callback)
|
|
)
|
|
self._controller.request_dimmer_level(self._addr)
|
|
|
|
def turn_on(self, **kwargs: Any) -> None:
|
|
"""Turn on the light."""
|
|
if ATTR_BRIGHTNESS in kwargs:
|
|
new_level = kwargs[ATTR_BRIGHTNESS]
|
|
elif self._prev_level == 0:
|
|
new_level = 255
|
|
else:
|
|
new_level = self._prev_level
|
|
self._set_brightness(new_level)
|
|
|
|
def turn_off(self, **kwargs: Any) -> None:
|
|
"""Turn off the light."""
|
|
self._set_brightness(0)
|
|
|
|
@property
|
|
def brightness(self):
|
|
"""Control the brightness."""
|
|
return self._level
|
|
|
|
def _set_brightness(self, level):
|
|
"""Send the brightness level to the device."""
|
|
self._controller.fade_dim(
|
|
float((level * 100.0) / 255.0), self._rate, 0, self._addr
|
|
)
|
|
|
|
@property
|
|
def extra_state_attributes(self):
|
|
"""Supported attributes."""
|
|
return {"homeworks_address": self._addr}
|
|
|
|
@property
|
|
def is_on(self):
|
|
"""Is the light on/off."""
|
|
return self._level != 0
|
|
|
|
@callback
|
|
def _update_callback(self, msg_type, values):
|
|
"""Process device specific messages."""
|
|
|
|
if msg_type == HW_LIGHT_CHANGED:
|
|
self._level = int((values[1] * 255.0) / 100.0)
|
|
if self._level != 0:
|
|
self._prev_level = self._level
|
|
self.async_write_ha_state()
|