From e8ff812d3989662f1bb9494682c697f05dc9afda Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Sat, 23 Apr 2022 21:20:23 +0200
Subject: [PATCH] Use ColorMode enum in freedompro (#70495)

---
 homeassistant/components/freedompro/light.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/homeassistant/components/freedompro/light.py b/homeassistant/components/freedompro/light.py
index 95731da914e..7659a136927 100644
--- a/homeassistant/components/freedompro/light.py
+++ b/homeassistant/components/freedompro/light.py
@@ -6,9 +6,7 @@ from pyfreedompro import put_state
 from homeassistant.components.light import (
     ATTR_BRIGHTNESS,
     ATTR_HS_COLOR,
-    COLOR_MODE_BRIGHTNESS,
-    COLOR_MODE_HS,
-    COLOR_MODE_ONOFF,
+    ColorMode,
     LightEntity,
 )
 from homeassistant.config_entries import ConfigEntry
@@ -55,11 +53,11 @@ class Device(CoordinatorEntity, LightEntity):
         )
         self._attr_is_on = False
         self._attr_brightness = 0
-        color_mode = COLOR_MODE_ONOFF
+        color_mode = ColorMode.ONOFF
         if "hue" in device["characteristics"]:
-            color_mode = COLOR_MODE_HS
+            color_mode = ColorMode.HS
         elif "brightness" in device["characteristics"]:
-            color_mode = COLOR_MODE_BRIGHTNESS
+            color_mode = ColorMode.BRIGHTNESS
         self._attr_color_mode = color_mode
         self._attr_supported_color_modes = {color_mode}