98 lines
2.5 KiB
Python
98 lines
2.5 KiB
Python
"""
|
|
Demo light platform that implements lights.
|
|
|
|
For more details about this platform, please refer to the documentation
|
|
https://home-assistant.io/components/demo/
|
|
"""
|
|
import random
|
|
|
|
from homeassistant.components.light import (
|
|
ATTR_BRIGHTNESS, ATTR_COLOR_TEMP, ATTR_RGB_COLOR, SUPPORT_BRIGHTNESS,
|
|
SUPPORT_COLOR_TEMP, SUPPORT_RGB_COLOR, Light)
|
|
|
|
LIGHT_COLORS = [
|
|
[237, 224, 33],
|
|
[255, 63, 111],
|
|
]
|
|
|
|
LIGHT_TEMPS = [240, 380]
|
|
|
|
SUPPORT_DEMO = SUPPORT_BRIGHTNESS | SUPPORT_COLOR_TEMP | SUPPORT_RGB_COLOR
|
|
|
|
|
|
def setup_platform(hass, config, add_devices_callback, discovery_info=None):
|
|
"""Setup the demo light platform."""
|
|
add_devices_callback([
|
|
DemoLight("Bed Light", False),
|
|
DemoLight("Ceiling Lights", True, LIGHT_COLORS[0], LIGHT_TEMPS[1]),
|
|
DemoLight("Kitchen Lights", True, LIGHT_COLORS[1], LIGHT_TEMPS[0])
|
|
])
|
|
|
|
|
|
class DemoLight(Light):
|
|
"""Represenation of a demo light."""
|
|
|
|
# pylint: disable=too-many-arguments
|
|
def __init__(self, name, state, rgb=None, ct=None, brightness=180):
|
|
"""Initialize the light."""
|
|
self._name = name
|
|
self._state = state
|
|
self._rgb = rgb or random.choice(LIGHT_COLORS)
|
|
self._ct = ct or random.choice(LIGHT_TEMPS)
|
|
self._brightness = brightness
|
|
|
|
@property
|
|
def should_poll(self):
|
|
"""No polling needed for a demo light."""
|
|
return False
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the light if any."""
|
|
return self._name
|
|
|
|
@property
|
|
def brightness(self):
|
|
"""Return the brightness of this light between 0..255."""
|
|
return self._brightness
|
|
|
|
@property
|
|
def rgb_color(self):
|
|
"""Return the RBG color value."""
|
|
return self._rgb
|
|
|
|
@property
|
|
def color_temp(self):
|
|
"""Return the CT color temperature."""
|
|
return self._ct
|
|
|
|
@property
|
|
def is_on(self):
|
|
"""Return true if light is on."""
|
|
return self._state
|
|
|
|
@property
|
|
def supported_features(self):
|
|
"""Flag supported features."""
|
|
return SUPPORT_DEMO
|
|
|
|
def turn_on(self, **kwargs):
|
|
"""Turn the light on."""
|
|
self._state = True
|
|
|
|
if ATTR_RGB_COLOR in kwargs:
|
|
self._rgb = kwargs[ATTR_RGB_COLOR]
|
|
|
|
if ATTR_COLOR_TEMP in kwargs:
|
|
self._ct = kwargs[ATTR_COLOR_TEMP]
|
|
|
|
if ATTR_BRIGHTNESS in kwargs:
|
|
self._brightness = kwargs[ATTR_BRIGHTNESS]
|
|
|
|
self.update_ha_state()
|
|
|
|
def turn_off(self, **kwargs):
|
|
"""Turn the light off."""
|
|
self._state = False
|
|
self.update_ha_state()
|