jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-25 22:27:07 +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_COLOR_TEMP,
ATTR_HS_COLOR, ATTR_HS_COLOR,
ATTR_TRANSITION, ATTR_TRANSITION,
COLOR_MODE_BRIGHTNESS, ColorMode,
COLOR_MODE_COLOR_TEMP,
COLOR_MODE_HS,
LightEntity, LightEntity,
LightEntityFeature, LightEntityFeature,
) )
@ -75,16 +73,16 @@ class TradfriLight(TradfriBaseEntity, LightEntity):
# Calculate supported color modes # Calculate supported color modes
self._attr_supported_color_modes = set() self._attr_supported_color_modes = set()
if self._device.light_control.can_set_color: 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: 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 ( if (
not self._attr_supported_color_modes not self._attr_supported_color_modes
and self._device.light_control.can_set_dimmer and self._device.light_control.can_set_dimmer
): ):
# Must be the only supported mode according to docs for # Must be the only supported mode according to docs for
# COLOR_MODE_BRIGHTNESS # ColorMode.BRIGHTNESS
self._attr_supported_color_modes.add(COLOR_MODE_BRIGHTNESS) self._attr_supported_color_modes.add(ColorMode.BRIGHTNESS)
if self._device_control: if self._device_control:
self._attr_min_mireds = self._device_control.min_mireds self._attr_min_mireds = self._device_control.min_mireds