diff --git a/homeassistant/components/twinkly/light.py b/homeassistant/components/twinkly/light.py index 440dab9a6ff..9068c04e01b 100644 --- a/homeassistant/components/twinkly/light.py +++ b/homeassistant/components/twinkly/light.py @@ -12,9 +12,7 @@ from homeassistant.components.light import ( ATTR_BRIGHTNESS, ATTR_RGB_COLOR, ATTR_RGBW_COLOR, - COLOR_MODE_BRIGHTNESS, - COLOR_MODE_RGB, - COLOR_MODE_RGBW, + ColorMode, LightEntity, ) from homeassistant.config_entries import ConfigEntry @@ -70,16 +68,16 @@ class TwinklyLight(LightEntity): self._conf = conf if device_info.get(DEV_LED_PROFILE) == DEV_PROFILE_RGBW: - self._attr_supported_color_modes = {COLOR_MODE_RGBW} - self._attr_color_mode = COLOR_MODE_RGBW + self._attr_supported_color_modes = {ColorMode.RGBW} + self._attr_color_mode = ColorMode.RGBW self._attr_rgbw_color = (255, 255, 255, 0) elif device_info.get(DEV_LED_PROFILE) == DEV_PROFILE_RGB: - self._attr_supported_color_modes = {COLOR_MODE_RGB} - self._attr_color_mode = COLOR_MODE_RGB + self._attr_supported_color_modes = {ColorMode.RGB} + self._attr_color_mode = ColorMode.RGB self._attr_rgb_color = (255, 255, 255) else: - self._attr_supported_color_modes = {COLOR_MODE_BRIGHTNESS} - self._attr_color_mode = COLOR_MODE_BRIGHTNESS + self._attr_supported_color_modes = {ColorMode.BRIGHTNESS} + self._attr_color_mode = ColorMode.BRIGHTNESS # Those are saved in the config entry in order to have meaningful values even # if the device is currently offline.