From 648c97378500de0e15eca7c4fbe8926472bab380 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Sat, 23 Apr 2022 21:14:34 +0200
Subject: [PATCH] Use ColorMode enum in hive (#70501)

---
 homeassistant/components/hive/light.py | 23 ++++++++++-------------
 1 file changed, 10 insertions(+), 13 deletions(-)

diff --git a/homeassistant/components/hive/light.py b/homeassistant/components/hive/light.py
index a5ab2c11f35..ba095896b64 100644
--- a/homeassistant/components/hive/light.py
+++ b/homeassistant/components/hive/light.py
@@ -7,10 +7,7 @@ from homeassistant.components.light import (
     ATTR_BRIGHTNESS,
     ATTR_COLOR_TEMP,
     ATTR_HS_COLOR,
-    COLOR_MODE_BRIGHTNESS,
-    COLOR_MODE_COLOR_TEMP,
-    COLOR_MODE_HS,
-    COLOR_MODE_ONOFF,
+    ColorMode,
     LightEntity,
 )
 from homeassistant.config_entries import ConfigEntry
@@ -144,25 +141,25 @@ class HiveDeviceLight(HiveEntity, LightEntity):
     def color_mode(self) -> str:
         """Return the color mode of the light."""
         if self.device["hiveType"] == "warmwhitelight":
-            return COLOR_MODE_BRIGHTNESS
+            return ColorMode.BRIGHTNESS
         if self.device["hiveType"] == "tuneablelight":
-            return COLOR_MODE_COLOR_TEMP
+            return ColorMode.COLOR_TEMP
         if self.device["hiveType"] == "colourtuneablelight":
             if self.device["status"]["mode"] == "COLOUR":
-                return COLOR_MODE_HS
-            return COLOR_MODE_COLOR_TEMP
-        return COLOR_MODE_ONOFF
+                return ColorMode.HS
+            return ColorMode.COLOR_TEMP
+        return ColorMode.ONOFF
 
     @property
     def supported_color_modes(self) -> set[str] | None:
         """Flag supported color modes."""
         if self.device["hiveType"] == "warmwhitelight":
-            return {COLOR_MODE_BRIGHTNESS}
+            return {ColorMode.BRIGHTNESS}
         if self.device["hiveType"] == "tuneablelight":
-            return {COLOR_MODE_COLOR_TEMP}
+            return {ColorMode.COLOR_TEMP}
         if self.device["hiveType"] == "colourtuneablelight":
-            return {COLOR_MODE_COLOR_TEMP, COLOR_MODE_HS}
-        return {COLOR_MODE_ONOFF}
+            return {ColorMode.COLOR_TEMP, ColorMode.HS}
+        return {ColorMode.ONOFF}
 
     async def async_update(self):
         """Update all Node data from Hive."""