From 76aea16f2940cc86d915c5d221cd09d50854a093 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Sat, 23 Apr 2022 21:47:13 +0200
Subject: [PATCH] Use ColorMode enum in twinkly (#70546)

---
 homeassistant/components/twinkly/light.py | 16 +++++++---------
 1 file changed, 7 insertions(+), 9 deletions(-)

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.