|
7 | 7 | import logging
|
8 | 8 |
|
9 | 9 | from homeassistant.components.light import SUPPORT_BRIGHTNESS, \
|
10 |
| - ATTR_BRIGHTNESS, SUPPORT_COLOR, ATTR_HS_COLOR, \ |
11 |
| - SUPPORT_EFFECT, ATTR_EFFECT, ATTR_EFFECT_LIST, SUPPORT_COLOR_TEMP, \ |
12 |
| - ATTR_COLOR_TEMP, ATTR_MIN_MIREDS, ATTR_MAX_MIREDS |
| 10 | + ATTR_BRIGHTNESS, SUPPORT_COLOR, ATTR_HS_COLOR, SUPPORT_EFFECT, \ |
| 11 | + ATTR_EFFECT, ATTR_EFFECT_LIST, SUPPORT_COLOR_TEMP, \ |
| 12 | + ATTR_COLOR_TEMP, ATTR_MIN_MIREDS, ATTR_MAX_MIREDS, LightEntity |
13 | 13 | from homeassistant.util import color
|
14 | 14 |
|
15 | 15 | # noinspection PyUnresolvedReferences
|
@@ -50,10 +50,14 @@ async def async_setup_platform(hass, config, add_entities,
|
50 | 50 | elif channels and len(channels) >= 2:
|
51 | 51 | add_entities([EWeLinkLightGroup(registry, deviceid, channels)])
|
52 | 52 | else:
|
53 |
| - add_entities([EWeLinkToggle(registry, deviceid, channels)]) |
| 53 | + add_entities([EWeLinkLight(registry, deviceid, channels)]) |
54 | 54 |
|
55 | 55 |
|
56 |
| -class SonoffD1(EWeLinkToggle): |
| 56 | +class EWeLinkLight(EWeLinkToggle, LightEntity): |
| 57 | + pass |
| 58 | + |
| 59 | + |
| 60 | +class SonoffD1(EWeLinkLight): |
57 | 61 | _brightness = 0
|
58 | 62 |
|
59 | 63 | def _update_handler(self, state: dict, attrs: dict):
|
@@ -124,7 +128,7 @@ async def async_turn_on(self, **kwargs) -> None:
|
124 | 128 | ]
|
125 | 129 |
|
126 | 130 |
|
127 |
| -class SonoffLED(EWeLinkToggle): |
| 131 | +class SonoffLED(EWeLinkLight): |
128 | 132 | _brightness = 0
|
129 | 133 | _hs_color = None
|
130 | 134 | _mode = 0
|
@@ -212,7 +216,7 @@ async def async_turn_on(self, **kwargs) -> None:
|
212 | 216 | await self.registry.send(self.deviceid, payload)
|
213 | 217 |
|
214 | 218 |
|
215 |
| -class SonoffB1(EWeLinkToggle): |
| 219 | +class SonoffB1(EWeLinkLight): |
216 | 220 | _brightness = None
|
217 | 221 | _hs_color = None
|
218 | 222 | _temp = None
|
@@ -384,7 +388,7 @@ async def async_turn_on(self, **kwargs) -> None:
|
384 | 388 | DIFFUSER_EFFECTS = ["Color Light", "RGB Color", "Night Light"]
|
385 | 389 |
|
386 | 390 |
|
387 |
| -class SonoffDiffuserLight(EWeLinkToggle): |
| 391 | +class SonoffDiffuserLight(EWeLinkLight): |
388 | 392 | _brightness = 0
|
389 | 393 | _hs_color = None
|
390 | 394 | _mode = 0
|
@@ -525,7 +529,7 @@ async def async_turn_off(self, **kwargs) -> None:
|
525 | 529 | }
|
526 | 530 |
|
527 | 531 |
|
528 |
| -class Sonoff103(EWeLinkToggle): |
| 532 | +class Sonoff103(EWeLinkLight): |
529 | 533 | _brightness = None
|
530 | 534 | _mode = None
|
531 | 535 | _temp = None
|
@@ -667,7 +671,7 @@ async def async_turn_on(self, **kwargs) -> None:
|
667 | 671 | }
|
668 | 672 |
|
669 | 673 |
|
670 |
| -class SonoffB05(EWeLinkToggle): |
| 674 | +class SonoffB05(EWeLinkLight): |
671 | 675 | _brightness = None
|
672 | 676 | _hs_color = None
|
673 | 677 | _mode = None
|
|
0 commit comments