Source code for miio.integrations.light.yeelight.spec_helper

import logging
import os
from enum import IntEnum
from typing import Dict, NamedTuple

import attr
import yaml

_LOGGER = logging.getLogger(__name__)


class YeelightSubLightType(IntEnum):
    Main = 0
    Background = 1


class ColorTempRange(NamedTuple):
    """Color temperature range."""

    min: int
    max: int


@attr.s(auto_attribs=True)
class YeelightLampInfo:
    color_temp: ColorTempRange
    supports_color: bool


@attr.s(auto_attribs=True)
class YeelightModelInfo:
    model: str
    night_light: bool
    lamps: Dict[YeelightSubLightType, YeelightLampInfo]


class YeelightSpecHelper:
    _models: Dict[str, YeelightModelInfo] = {}

    def __init__(self):
        if not YeelightSpecHelper._models:
            self._parse_specs_yaml()

    def _parse_specs_yaml(self):
        generic_info = YeelightModelInfo(
            "generic",
            False,
            {
                YeelightSubLightType.Main: YeelightLampInfo(
                    ColorTempRange(1700, 6500), False
                )
            },
        )
        YeelightSpecHelper._models["generic"] = generic_info
        # read the yaml file to populate the internal model cache
        with open(os.path.dirname(__file__) + "/specs.yaml") as filedata:
            models = yaml.safe_load(filedata)
            for key, value in models.items():
                lamps = {
                    YeelightSubLightType.Main: YeelightLampInfo(
                        ColorTempRange(*value["color_temp"]),
                        value["supports_color"],
                    )
                }

                if "background" in value:
                    lamps[YeelightSubLightType.Background] = YeelightLampInfo(
                        ColorTempRange(*value["background"]["color_temp"]),
                        value["background"]["supports_color"],
                    )

                info = YeelightModelInfo(key, value["night_light"], lamps)
                YeelightSpecHelper._models[key] = info

    @property
    def supported_models(self):
        return self._models.keys()

    def get_model_info(self, model) -> YeelightModelInfo:
        if model not in self._models:
            _LOGGER.warning(
                "Unknown model %s, please open an issue and supply features for this light. Returning generic information.",
                model,
            )
            return self._models["generic"]
        return self._models[model]