-
Notifications
You must be signed in to change notification settings - Fork 18
/
light.py
72 lines (59 loc) · 2.48 KB
/
light.py
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
import logging
import homeassistant.util.color as color_util
from homeassistant.components.light import LightEntity
from .aiot_manager import (
AiotManager,
AiotToggleableEntityBase,
)
from .const import DOMAIN, HASS_DATA_AIOT_MANAGER
TYPE = "light"
_LOGGER = logging.getLogger(__name__)
DATA_KEY = f"{TYPE}.{DOMAIN}"
async def async_setup_entry(hass, config_entry, async_add_entities):
manager: AiotManager = hass.data[DOMAIN][HASS_DATA_AIOT_MANAGER]
await manager.async_add_entities(
config_entry, TYPE, AiotLightEntity, async_add_entities
)
class AiotLightEntity(AiotToggleableEntityBase, LightEntity):
def __init__(self, hass, device, res_params, channel=None, **kwargs):
AiotToggleableEntityBase.__init__(
self, hass, device, res_params, TYPE, channel, **kwargs
)
self._attr_color_mode = kwargs.get("color_mode")
async def async_turn_on(self, **kwargs):
"""Turn the specified light on."""
hs_color = kwargs.get("hs_color")
if hs_color:
await self.async_set_resource("color", hs_color)
brightness = kwargs.get("brightness")
if brightness:
await self.async_set_resource("brightness", brightness)
await super().async_turn_on(**kwargs)
def convert_attr_to_res(self, res_name, attr_value):
if res_name == "brightness":
# attr_value:0-255,亮度
return int(attr_value * 100 / 255)
elif res_name == "color":
# attr_value:hs颜色
rgb_color = color_util.color_hs_to_RGB(*attr_value)
return int(
"{}{}{}{}".format(
hex(int(self.brightness * 100 / 255))[2:4].zfill(2),
hex(rgb_color[0])[2:4].zfill(2),
hex(rgb_color[1])[2:4].zfill(2),
hex(rgb_color[2])[2:4].zfill(2),
),
16,
)
return super().convert_attr_to_res(res_name, attr_value)
def convert_res_to_attr(self, res_name, res_value):
if res_name == "brightness":
# res_value:0-100,亮度百分比
return int(int(res_value) * 255 / 100)
elif res_name == "color":
# res_value:十进制整数字符串
argb = hex(int(res_value))
return color_util.color_RGB_to_hs(
int(argb[4:6], 16), int(argb[6:8], 16), int(argb[8:10], 16)
)
return super().convert_res_to_attr(res_name, res_value)