Skip to content

Add ColorMode enum to light platform #69223

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
frenck merged 2 commits into from
Apr 11, 2022
Merged

Add ColorMode enum to light platform #69223

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
4d90883
Add ColorMode enum to light platform
frenck Apr 3, 2022
6f0b6d6
Adjust tests
frenck Apr 4, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
32 changes: 14 additions & 18 deletions homeassistant/components/demo/light.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,11 +11,7 @@
ATTR_RGBW_COLOR,
ATTR_RGBWW_COLOR,
ATTR_WHITE,
COLOR_MODE_COLOR_TEMP,
COLOR_MODE_HS,
COLOR_MODE_RGBW,
COLOR_MODE_RGBWW,
COLOR_MODE_WHITE,
ColorMode,
LightEntity,
LightEntityFeature,
)
Expand All @@ -33,8 +29,8 @@

LIGHT_TEMPS = [240, 380]

SUPPORT_DEMO = {COLOR_MODE_HS, COLOR_MODE_COLOR_TEMP}
SUPPORT_DEMO_HS_WHITE = {COLOR_MODE_HS, COLOR_MODE_WHITE}
SUPPORT_DEMO = {ColorMode.HS, ColorMode.COLOR_TEMP}
SUPPORT_DEMO_HS_WHITE = {ColorMode.HS, ColorMode.WHITE}


async def async_setup_platform(
Expand Down Expand Up @@ -74,15 +70,15 @@ async def async_setup_platform(
name="Office RGBW Lights",
rgbw_color=(255, 0, 0, 255),
state=True,
supported_color_modes={COLOR_MODE_RGBW},
supported_color_modes={ColorMode.RGBW},
unique_id="light_4",
),
DemoLight(
available=True,
name="Living Room RGBWW Lights",
rgbww_color=(255, 0, 0, 255, 0),
state=True,
supported_color_modes={COLOR_MODE_RGBWW},
supported_color_modes={ColorMode.RGBWW},
unique_id="light_5",
),
DemoLight(
Expand Down Expand Up @@ -138,13 +134,13 @@ def __init__(
self._state = state
self._unique_id = unique_id
if hs_color:
self._color_mode = COLOR_MODE_HS
self._color_mode = ColorMode.HS
elif rgbw_color:
self._color_mode = COLOR_MODE_RGBW
self._color_mode = ColorMode.RGBW
elif rgbww_color:
self._color_mode = COLOR_MODE_RGBWW
self._color_mode = ColorMode.RGBWW
else:
self._color_mode = COLOR_MODE_COLOR_TEMP
self._color_mode = ColorMode.COLOR_TEMP
if not supported_color_modes:
supported_color_modes = SUPPORT_DEMO
self._color_modes = supported_color_modes
Expand Down Expand Up @@ -247,26 +243,26 @@ async def async_turn_on(self, **kwargs) -> None:
self._brightness = kwargs[ATTR_BRIGHTNESS]

if ATTR_COLOR_TEMP in kwargs:
self._color_mode = COLOR_MODE_COLOR_TEMP
self._color_mode = ColorMode.COLOR_TEMP
self._ct = kwargs[ATTR_COLOR_TEMP]

if ATTR_EFFECT in kwargs:
self._effect = kwargs[ATTR_EFFECT]

if ATTR_HS_COLOR in kwargs:
self._color_mode = COLOR_MODE_HS
self._color_mode = ColorMode.HS
self._hs_color = kwargs[ATTR_HS_COLOR]

if ATTR_RGBW_COLOR in kwargs:
self._color_mode = COLOR_MODE_RGBW
self._color_mode = ColorMode.RGBW
self._rgbw_color = kwargs[ATTR_RGBW_COLOR]

if ATTR_RGBWW_COLOR in kwargs:
self._color_mode = COLOR_MODE_RGBWW
self._color_mode = ColorMode.RGBWW
self._rgbww_color = kwargs[ATTR_RGBWW_COLOR]

if ATTR_WHITE in kwargs:
self._color_mode = COLOR_MODE_WHITE
self._color_mode = ColorMode.WHITE
self._brightness = kwargs[ATTR_WHITE]

# As we have disabled polling, we need to inform
Expand Down
Loading