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
|
from __future__ import annotations
from typing import Any
from ..const.light import BulbColorMode, ColorMode
from .base_light import SwitchbotSequenceBaseLight
# Private mapping from device-specific color modes to original ColorMode enum
_BULB_COLOR_MODE_MAP = {
BulbColorMode.COLOR_TEMP: ColorMode.COLOR_TEMP,
BulbColorMode.RGB: ColorMode.RGB,
BulbColorMode.DYNAMIC: ColorMode.EFFECT,
BulbColorMode.UNKNOWN: ColorMode.OFF,
}
COLOR_BULB_CONTROL_HEADER = "570F4701"
class SwitchbotBulb(SwitchbotSequenceBaseLight):
"""Representation of a Switchbot bulb."""
_turn_on_command = f"{COLOR_BULB_CONTROL_HEADER}01"
_turn_off_command = f"{COLOR_BULB_CONTROL_HEADER}02"
_set_rgb_command = f"{COLOR_BULB_CONTROL_HEADER}12{{}}"
_set_color_temp_command = f"{COLOR_BULB_CONTROL_HEADER}13{{}}"
_set_brightness_command = f"{COLOR_BULB_CONTROL_HEADER}14{{}}"
_get_basic_info_command = ["570003", "570f4801"]
_effect_dict = {
"colorful": ["570F4701010300"],
"flickering": ["570F4701010301"],
"breathing": ["570F4701010302"],
}
@property
def color_modes(self) -> set[ColorMode]:
"""Return the supported color modes."""
return {ColorMode.RGB, ColorMode.COLOR_TEMP}
@property
def color_mode(self) -> ColorMode:
"""Return the current color mode."""
device_mode = BulbColorMode(self._get_adv_value("color_mode") or 10)
return _BULB_COLOR_MODE_MAP.get(device_mode, ColorMode.OFF)
async def get_basic_info(self) -> dict[str, Any] | None:
"""Get device basic settings."""
if not (
res := await self._get_multi_commands_results(self._get_basic_info_command)
):
return None
_version_info, _data = res
self._state["r"] = _data[3]
self._state["g"] = _data[4]
self._state["b"] = _data[5]
self._state["cw"] = int.from_bytes(_data[6:8], "big")
return {
"isOn": bool(_data[1] & 0b10000000),
"brightness": _data[2] & 0b01111111,
"r": self._state["r"],
"g": self._state["g"],
"b": self._state["b"],
"cw": self._state["cw"],
"color_mode": _data[10] & 0b00001111,
"firmware": _version_info[2] / 10.0,
}
|
