1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
import homeassistant.helpers.config_validation as cv
import voluptuous as vol
from homeassistant.const import CONF_MAC
from .driver import BekenConnection, makemsg
from homeassistant.components.light import (
LightEntity,
SUPPORT_BRIGHTNESS,
SUPPORT_COLOR,
SUPPORT_WHITE_VALUE,
ATTR_BRIGHTNESS,
ATTR_RGBW_COLOR,
COLOR_MODE_RGBW,
PLATFORM_SCHEMA
)
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Required("name"): cv.string,
vol.Required("address"): cv.string
})
SUPPORT_FEATURES_RGB = SUPPORT_BRIGHTNESS | SUPPORT_COLOR
SUPPORT_FEATURES_WHITE = SUPPORT_BRIGHTNESS
def setup_platform(hass, config, add_entities, discovery_info=None):
add_entities([ BekenLight(name=config["name"], address=config["address"]) ])
class BekenLight(LightEntity):
def __init__(self, **kwargs):
self._name = kwargs["name"]
self._address = kwargs["address"]
self._on = False
self._brightness = 255
self._rgbw = (255, 255, 255, 255) # r, g, b, w
@property
def color_mode(self):
return COLOR_MODE_RGBW
@property
def supported_color_modes(self):
return set([ COLOR_MODE_RGBW ])
@property
def supported_features(self):
return SUPPORT_BRIGHTNESS | SUPPORT_COLOR | SUPPORT_WHITE_VALUE
@property
def unique_id(self):
return self._address
@property
def name(self):
return self._name
@property
def is_on(self):
return self._on
@property
def brightness(self):
return self._brightness
@property
def rgbw_color(self):
return self._rgbw
def turn_on(self, **kwargs):
self._on = True
self._brightness = kwargs.get(ATTR_BRIGHTNESS)
self._rgbw = kwargs.get(ATTR_RGBW_COLOR)
def turn_off(self, **kwargs):
self._on = False
|