138 lines
3.7 KiB
Python
138 lines
3.7 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_EFFECT,
|
|
ATTR_RGB_COLOR, ATTR_WHITE_VALUE, ATTR_XY_COLOR, SUPPORT_BRIGHTNESS,
|
|
SUPPORT_COLOR_TEMP, SUPPORT_EFFECT, SUPPORT_RGB_COLOR, SUPPORT_WHITE_VALUE,
|
|
Light)
|
|
|
|
LIGHT_COLORS = [
|
|
[237, 224, 33],
|
|
[255, 63, 111],
|
|
]
|
|
|
|
LIGHT_EFFECT_LIST = ['rainbow', 'none']
|
|
|
|
LIGHT_TEMPS = [240, 380]
|
|
|
|
SUPPORT_DEMO = (SUPPORT_BRIGHTNESS | SUPPORT_COLOR_TEMP | SUPPORT_EFFECT |
|
|
SUPPORT_RGB_COLOR | SUPPORT_WHITE_VALUE)
|
|
|
|
|
|
def setup_platform(hass, config, add_devices_callback, discovery_info=None):
|
|
"""Setup the demo light platform."""
|
|
add_devices_callback([
|
|
DemoLight("Bed Light", False, effect_list=LIGHT_EFFECT_LIST,
|
|
effect=LIGHT_EFFECT_LIST[0]),
|
|
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."""
|
|
|
|
def __init__(
|
|
self, name, state, rgb=None, ct=None, brightness=180,
|
|
xy_color=(.5, .5), white=200, effect_list=None, effect=None):
|
|
"""Initialize the light."""
|
|
self._name = name
|
|
self._state = state
|
|
self._rgb = rgb
|
|
self._ct = ct or random.choice(LIGHT_TEMPS)
|
|
self._brightness = brightness
|
|
self._xy_color = xy_color
|
|
self._white = white
|
|
self._effect_list = effect_list
|
|
self._effect = effect
|
|
|
|
@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 xy_color(self):
|
|
"""Return the XY color value [float, float]."""
|
|
return self._xy_color
|
|
|
|
@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 white_value(self):
|
|
"""Return the white value of this light between 0..255."""
|
|
return self._white
|
|
|
|
@property
|
|
def effect_list(self):
|
|
"""Return the list of supported effects."""
|
|
return self._effect_list
|
|
|
|
@property
|
|
def effect(self):
|
|
"""Return the current effect."""
|
|
return self._effect
|
|
|
|
@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]
|
|
|
|
if ATTR_XY_COLOR in kwargs:
|
|
self._xy_color = kwargs[ATTR_XY_COLOR]
|
|
|
|
if ATTR_WHITE_VALUE in kwargs:
|
|
self._white = kwargs[ATTR_WHITE_VALUE]
|
|
|
|
if ATTR_EFFECT in kwargs:
|
|
self._effect = kwargs[ATTR_EFFECT]
|
|
|
|
self.schedule_update_ha_state()
|
|
|
|
def turn_off(self, **kwargs):
|
|
"""Turn the light off."""
|
|
self._state = False
|
|
self.schedule_update_ha_state()
|