core/homeassistant/components/velbus/light.py

78 lines
2.2 KiB
Python

"""Support for Velbus light."""
import logging
from velbus.util import VelbusException
from homeassistant.components.light import (
ATTR_BRIGHTNESS,
ATTR_TRANSITION,
SUPPORT_BRIGHTNESS,
SUPPORT_TRANSITION,
Light,
)
from . import VelbusEntity
from .const import DOMAIN
_LOGGER = logging.getLogger(__name__)
async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
"""Old way."""
pass
async def async_setup_entry(hass, entry, async_add_entities):
"""Set up Velbus light based on config_entry."""
cntrl = hass.data[DOMAIN][entry.entry_id]["cntrl"]
modules_data = hass.data[DOMAIN][entry.entry_id]["light"]
entities = []
for address, channel in modules_data:
module = cntrl.get_module(address)
entities.append(VelbusLight(module, channel))
async_add_entities(entities)
class VelbusLight(VelbusEntity, Light):
"""Representation of a Velbus light."""
@property
def supported_features(self):
"""Flag supported features."""
return SUPPORT_BRIGHTNESS | SUPPORT_TRANSITION
@property
def is_on(self):
"""Return true if the light is on."""
return self._module.is_on(self._channel)
@property
def brightness(self):
"""Return the brightness of the light."""
return self._module.get_dimmer_state(self._channel)
def turn_on(self, **kwargs):
"""Instruct the Velbus light to turn on."""
try:
if ATTR_BRIGHTNESS in kwargs:
self._module.set_dimmer_state(
self._channel,
kwargs[ATTR_BRIGHTNESS],
kwargs.get(ATTR_TRANSITION, 0),
)
else:
self._module.restore_dimmer_state(
self._channel, kwargs.get(ATTR_TRANSITION, 0),
)
except VelbusException as err:
_LOGGER.error("A Velbus error occurred: %s", err)
def turn_off(self, **kwargs):
"""Instruct the velbus light to turn off."""
try:
self._module.set_dimmer_state(
self._channel, 0, kwargs.get(ATTR_TRANSITION, 0),
)
except VelbusException as err:
_LOGGER.error("A Velbus error occurred: %s", err)