From fd9f0d02924f051e1122b315d60655f4e23b31d2 Mon Sep 17 00:00:00 2001
From: Necroneco <caibinqing@live.com>
Date: Sun, 3 Nov 2024 23:59:48 +0800
Subject: [PATCH] set color_mode for lights

---
 custom_components/aqara_gateway/light.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/custom_components/aqara_gateway/light.py b/custom_components/aqara_gateway/light.py
index 3d1b1c8a..329ac213 100755
--- a/custom_components/aqara_gateway/light.py
+++ b/custom_components/aqara_gateway/light.py
@@ -74,6 +74,15 @@ def __init__(
             color_modes = {ColorMode.RGB}
         self._attr_supported_color_modes = color_modes
 
+        if ColorMode.COLOR_TEMP in color_modes:
+            self._attr_color_mode = ColorMode.COLOR_TEMP
+        elif ColorMode.BRIGHTNESS in color_modes:
+            self._attr_color_mode = ColorMode.BRIGHTNESS
+        elif ColorMode.ONOFF in color_modes:
+            self._attr_color_mode = ColorMode.ONOFF
+        else:
+            self._attr_color_mode = ColorMode.UNKNOWN
+
     @property
     def is_on(self) -> bool:
         """return state """