Add light specs for yeelight

miio/integrations/yeelight/__init__.py

@@ -8,7 +8,7 @@
 from miio.exceptions import DeviceException
 from miio.utils import int_to_rgb, rgb_to_int
 
-SUPPORTED_MODELS = ["yeelink.light.color1"]
+from .spec_hellper import YeelightSpecHellper
 
 
 class YeelightException(DeviceException):
@@ -259,7 +259,7 @@ class Yeelight(Device):
     which however requires enabling the developer mode on the bulbs.
     """
 
-    _supported_models = SUPPORTED_MODELS
+    _supported_models = YeelightSpecHellper.supported_models()
 
     @command(default_output=format_output("", "{result.cli_format}"))
     def status(self) -> YeelightStatus:

miio/integrations/yeelight/spec_hellper.py

@@ -0,0 +1,48 @@
+import logging
+import os
+from typing import Tuple
+
+import yaml
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class YeelightModelInfo:
+    model: str
+    color_temp: Tuple[int, int] = (1700, 6500)
+    has_night_light: bool = False
+    has_background_light: bool = False
+    supports_color: bool = False
+
+    def __init__(self, model, spec=None):
+        self.model = model
+        if spec:
+            if "color_temp" in spec:
+                self.color_temp = tuple(spec["color_temp"])
+            if "night_light" in spec:
+                self.has_night_light = spec["night_light"]
+            if "background_light" in spec:
+                self.has_background_light = spec["background_light"]
+            if "supports_color" in spec:
+                self.supports_color = spec["supports_color"]
+
+
+class YeelightSpecHellper:
+    @staticmethod
+    def specs() -> dict:
+        with open(os.path.dirname(__file__) + "/specs.yaml") as filedata:
+            return yaml.safe_load(filedata)
+
+    @staticmethod
+    def supported_models():
+        return YeelightSpecHellper.specs().keys()
+
+    @staticmethod
+    def get_model_info(model) -> YeelightModelInfo:
+        if model not in YeelightSpecHellper.specs():
+            _LOGGER.warning(
+                "Unknown model %s, please open an issue and supply features for this light. Returning generic information."
+                % model
+            )
+            return YeelightModelInfo(model="generic")
+        return YeelightModelInfo(model, YeelightSpecHellper.specs().get(model))

miio/integrations/yeelight/specs.yaml

@@ -0,0 +1,180 @@
+yeelink.light.bslamp1:
+  color_temp: [1700, 6500]
+  night_light: False
+  background_light: False
+  supports_color: True
+yeelink.light.bslamp2:
+  color_temp: [1700, 6500]
+  night_light: True
+  background_light: False
+  supports_color: True
+yeelink.light.bslamp3:
+  color_temp: [1700, 6500]
+  night_light: True
+  background_light: False
+  supports_color: True
+yeelink.light.ceil26:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: False
+  supports_color: False
+yeelink.light.ceila:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: False
+  supports_color: False
+yeelink.light.ceiling1:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: False
+  supports_color: False
+yeelink.light.ceiling2:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: False
+  supports_color: False
+yeelink.light.ceiling3:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: False
+  supports_color: False
+yeelink.light.ceiling4:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: True
+  supports_color: False
+yeelink.light.ceiling5:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: False
+  supports_color: False
+yeelink.light.ceiling6:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: False
+  supports_color: False
+yeelink.light.ceiling10:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: True
+  supports_color: False
+yeelink.light.ceiling13:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: False
+  supports_color: False
+yeelink.light.ceiling15:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: False
+  supports_color: False
+yeelink.light.ceiling18:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: False
+  supports_color: False
+yeelink.light.ceiling19:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: True
+  supports_color: False
+yeelink.light.ceiling20:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: True
+  supports_color: False
+yeelink.light.ceiling22:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: False
+  supports_color: False
+yeelink.light.ceiling24:
+  color_temp: [2700, 6500]
+  night_light: True
+  background_light: False
+  supports_color: False
+yeelink.light.color1:
+  color_temp: [1700, 6500]
+  night_light: False
+  background_light: False
+  supports_color: True
+yeelink.light.color2:
+  color_temp: [2700, 6500]
+  night_light: False
+  background_light: False
+  supports_color: True
+yeelink.light.color4:
+  color_temp: [1700, 6500]
+  night_light: False
+  background_light: False
+  supports_color: True
+yeelink.light.colorc:
+  color_temp: [2700, 6500]
+  night_light: False
+  background_light: False
+  supports_color: True
+yeelink.light.color:
+  color_temp: [1700, 6500]
+  night_light: False
+  background_light: False
+  supports_color: True
+yeelink.light.ct_bulb:
+  color_temp: [2700, 6500]
+  night_light: False
+  background_light: False
+  supports_color: False
+yeelink.light.ct2:
+  color_temp: [2700, 6500]
+  night_light: False
+  background_light: False
+  supports_color: False
+yeelink.light.lamp1:
+  color_temp: [2700, 5000]
+  night_light: False
+  background_light: False
+  supports_color: False
+yeelink.light.lamp4:
+  color_temp: [2600, 5000]
+  night_light: False
+  background_light: False
+  supports_color: False
+yeelink.light.lamp15:
+  color_temp: [2700, 6500]
+  night_light: False
+  background_light: True
+  supports_color: False
+yeelink.light.mono1:
+  color_temp: [2700, 2700]
+  night_light: False
+  background_light: False
+  supports_color: False
+yeelink.light.mono5:
+  color_temp: [2700, 2700]
+  night_light: False
+  background_light: False
+  supports_color: False
+yeelink.light.mono:
+  color_temp: [2700, 2700]
+  night_light: False
+  background_light: False
+  supports_color: False
+yeelink.light.monob:
+  color_temp: [2700, 2700]
+  night_light: False
+  background_light: False
+  supports_color: False
+yeelink.light.strip1:
+  color_temp: [1700, 6500]
+  night_light: False
+  background_light: False
+  supports_color: True
+yeelink.light.strip2:
+  color_temp: [2700, 6500]
+  night_light: False
+  background_light: False
+  supports_color: True
+yeelink.light.strip4:
+  color_temp: [2700, 6500]
+  night_light: False
+  background_light: False
+  supports_color: True

miio/integrations/yeelight/tests/test_yeelight_spec_hellper.py

@@ -0,0 +1,20 @@
+from unittest import TestCase
+
+from miio.integrations.yeelight.spec_hellper import YeelightSpecHellper
+
+
+class TestYeelightSpecHellper(TestCase):
+    def test_get_model_info(self):
+        model_info = YeelightSpecHellper.get_model_info("yeelink.light.bslamp1")
+        assert model_info.model == "yeelink.light.bslamp1"
+        assert model_info.color_temp == (1700, 6500)
+        assert model_info.has_night_light is False
+        assert model_info.has_background_light is False
+        assert model_info.supports_color is True
+
+        model_info = YeelightSpecHellper.get_model_info("notreal")
+        assert model_info.model == "generic"
+        assert model_info.color_temp == (1700, 6500)
+        assert model_info.has_night_light is False
+        assert model_info.has_background_light is False
+        assert model_info.supports_color is False