tuya-local/custom_components/tuya_local/climate.py

"""
Setup for different kinds of Tuya climate devices
"""
import logging

from . import DOMAIN
from .const import (
    CONF_CLIMATE,
    CONF_DEVICE_ID,
    CONF_TYPE,
)
from .generic.climate import TuyaLocalClimate
from .helpers.device_config import get_config

_LOGGER = logging.getLogger(__name__)


async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
    """Set up the Tuya device according to its type."""
    data = hass.data[DOMAIN][discovery_info[CONF_DEVICE_ID]]
    device = data["device"]
    climates = []

    cfg = get_config(discovery_info[CONF_TYPE])
    if cfg is None:
        raise ValueError(f"No device config found for {discovery_info}")
    ecfg = cfg.primary_entity
    if ecfg.entity == "climate" and discovery_info.get(ecfg.config_id, False):
        legacy_class = ecfg.legacy_class
        if legacy_class is None:
            data[ecfg.config_id] = TuyaLocalClimate(device, ecfg)
        else:
            data[ecfg.config_id] = legacy_class(device)
        climates.append(data[ecfg.config_id])
        if ecfg.deprecated:
            _LOGGER.warning(ecfg.deprecation_message)
        _LOGGER.debug(f"Adding climate for {ecfg.name}")

    for ecfg in cfg.secondary_entities():
        if ecfg.entity == "climate" and discovery_info.get(ecfg.config_id, False):
            legacy_class = ecfg.legacy_class
            if legacy_class is None:
                data[ecfg.config_id] = TuyaLocalClimate(device, ecfg)
            else:
                data[ecfg.config_id] = legacy_class(device)
            climates.append(data[ecfg.config_id])
            if ecfg.deprecated:
                _LOGGER.warning(ecfg.deprecation_message)
            _LOGGER.debug(f"Adding climate for {ecfg.name}")

    if not climates:
        raise ValueError(f"{device.name} does not support use as a climate device.")

    async_add_entities(climates)


async def async_setup_entry(hass, config_entry, async_add_entities):
    config = {**config_entry.data, **config_entry.options}
    await async_setup_platform(hass, {}, async_add_entities, config)