github
/
ha-core
mirror of https://github.com/home-assistant/core
1
Fork 0
Code Releases Activity

Use ColorMode enum in broadlink (#70434)

This commit is contained in:
epenet 2022-04-23 07:49:46 +02:00 committed by GitHub
parent 643d04a717
commit 68697b6477
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
homeassistant/components/broadlink/light.py
View File

@ -8,10 +8,7 @@ from homeassistant.components.light import (
ATTR_COLOR_MODE,
ATTR_COLOR_TEMP,
ATTR_HS_COLOR,
COLOR_MODE_BRIGHTNESS,
COLOR_MODE_COLOR_TEMP,
COLOR_MODE_HS,
COLOR_MODE_UNKNOWN,
ColorMode,
LightEntity,
)
from homeassistant.config_entries import ConfigEntry
@ -56,11 +53,11 @@ class BroadlinkLight(BroadlinkEntity, LightEntity):
data = self._coordinator.data
if {"hue", "saturation"}.issubset(data):
self._attr_supported_color_modes.add(COLOR_MODE_HS)
self._attr_supported_color_modes.add(ColorMode.HS)
if "colortemp" in data:
self._attr_supported_color_modes.add(COLOR_MODE_COLOR_TEMP)
self._attr_supported_color_modes.add(ColorMode.COLOR_TEMP)
if not self.supported_color_modes:
self._attr_supported_color_modes = {COLOR_MODE_BRIGHTNESS}
self._attr_supported_color_modes = {ColorMode.BRIGHTNESS}
self._update_state(data)
@ -72,8 +69,8 @@ class BroadlinkLight(BroadlinkEntity, LightEntity):
if "brightness" in data:
self._attr_brightness = round(data["brightness"] * 2.55)
if self.supported_color_modes == {COLOR_MODE_BRIGHTNESS}:
self._attr_color_mode = COLOR_MODE_BRIGHTNESS
if self.supported_color_modes == {ColorMode.BRIGHTNESS}:
self._attr_color_mode = ColorMode.BRIGHTNESS
return
if {"hue", "saturation"}.issubset(data):
@ -84,12 +81,12 @@ class BroadlinkLight(BroadlinkEntity, LightEntity):
if "bulb_colormode" in data:
if data["bulb_colormode"] == BROADLINK_COLOR_MODE_RGB:
self._attr_color_mode = COLOR_MODE_HS
self._attr_color_mode = ColorMode.HS
elif data["bulb_colormode"] == BROADLINK_COLOR_MODE_WHITE:
self._attr_color_mode = COLOR_MODE_COLOR_TEMP
self._attr_color_mode = ColorMode.COLOR_TEMP
else:
# Scenes are not yet supported.
self._attr_color_mode = COLOR_MODE_UNKNOWN
self._attr_color_mode = ColorMode.UNKNOWN
async def async_turn_on(self, **kwargs):
"""Turn on the light."""
@ -99,7 +96,7 @@ class BroadlinkLight(BroadlinkEntity, LightEntity):
brightness = kwargs[ATTR_BRIGHTNESS]
state["brightness"] = round(brightness / 2.55)
if self.supported_color_modes == {COLOR_MODE_BRIGHTNESS}:
if self.supported_color_modes == {ColorMode.BRIGHTNESS}:
state["bulb_colormode"] = BROADLINK_COLOR_MODE_WHITE
elif ATTR_HS_COLOR in kwargs:
@ -115,9 +112,9 @@ class BroadlinkLight(BroadlinkEntity, LightEntity):
elif ATTR_COLOR_MODE in kwargs:
color_mode = kwargs[ATTR_COLOR_MODE]
if color_mode == COLOR_MODE_HS:
if color_mode == ColorMode.HS:
state["bulb_colormode"] = BROADLINK_COLOR_MODE_RGB
elif color_mode == COLOR_MODE_COLOR_TEMP:
elif color_mode == ColorMode.COLOR_TEMP:
state["bulb_colormode"] = BROADLINK_COLOR_MODE_WHITE
else:
# Scenes are not yet supported.