set color_mode for lights

niceboygithub · Nov 3, 2024 · fd9f0d0 · fd9f0d0
1 parent 264b887
commit fd9f0d0
Showing 1 changed file with 9 additions and 0 deletions.
diff --git 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 """