jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 13:17:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use ColorMode enum in tradfri (#70544)

Browse Source
This commit is contained in:
epenet 2022-04-23 21:46:24 +02:00 committed by GitHub
parent 6e1b8a9387
commit 9a38172d1f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
homeassistant/components/tradfri/light.py
View File

@ -11,9 +11,7 @@ from homeassistant.components.light import (
ATTR_COLOR_TEMP,
ATTR_HS_COLOR,
ATTR_TRANSITION,
COLOR_MODE_BRIGHTNESS,
COLOR_MODE_COLOR_TEMP,
COLOR_MODE_HS,
ColorMode,
LightEntity,
LightEntityFeature,
)
@ -75,16 +73,16 @@ class TradfriLight(TradfriBaseEntity, LightEntity):
# Calculate supported color modes
self._attr_supported_color_modes = set()
if self._device.light_control.can_set_color:
self._attr_supported_color_modes.add(COLOR_MODE_HS)
self._attr_supported_color_modes.add(ColorMode.HS)
if self._device.light_control.can_set_temp:
self._attr_supported_color_modes.add(COLOR_MODE_COLOR_TEMP)
self._attr_supported_color_modes.add(ColorMode.COLOR_TEMP)
if (
not self._attr_supported_color_modes
and self._device.light_control.can_set_dimmer
):
# Must be the only supported mode according to docs for
# COLOR_MODE_BRIGHTNESS
self._attr_supported_color_modes.add(COLOR_MODE_BRIGHTNESS)
# ColorMode.BRIGHTNESS
self._attr_supported_color_modes.add(ColorMode.BRIGHTNESS)
if self._device_control:
self._attr_min_mireds = self._device_control.min_mireds