core/homeassistant/components/homeworks/light.py

103 lines
3.0 KiB
Python

"""Support for Lutron Homeworks lights."""
import logging
from pyhomeworks.pyhomeworks import HW_LIGHT_CHANGED
from homeassistant.components.light import (
ATTR_BRIGHTNESS,
SUPPORT_BRIGHTNESS,
LightEntity,
)
from homeassistant.const import CONF_NAME
from homeassistant.core import callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from . import CONF_ADDR, CONF_DIMMERS, CONF_RATE, HOMEWORKS_CONTROLLER, HomeworksDevice
_LOGGER = logging.getLogger(__name__)
def setup_platform(hass, config, add_entities, discover_info=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."""
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):
"""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)
@property
def supported_features(self):
"""Supported features."""
return SUPPORT_BRIGHTNESS
def turn_on(self, **kwargs):
"""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):
"""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 device_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()