tuya-local/custom_components/tuya_local/climate.py

"""
Setup for different kinds of Tuya climate devices
"""
from . import DOMAIN
from .const import (
    CONF_CLIMATE,
    CONF_DEVICE_ID,
    CONF_TYPE,
    CONF_TYPE_AUTO,
    CONF_TYPE_DEHUMIDIFIER,
    CONF_TYPE_FAN,
    CONF_TYPE_GECO_HEATER,
    CONF_TYPE_GPCV_HEATER,
    CONF_TYPE_GPPH_HEATER,
    CONF_TYPE_KOGAN_HEATER,
    CONF_CLIMATE,
)
from .dehumidifier.climate import GoldairDehumidifier
from .fan.climate import GoldairFan
from .geco_heater.climate import GoldairGECOHeater
from .gpcv_heater.climate import GoldairGPCVHeater
from .heater.climate import GoldairHeater
from .kogan_heater.climate import KoganHeater


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

    if discovery_info[CONF_TYPE] == CONF_TYPE_AUTO:
        discovery_info[CONF_TYPE] = await device.async_inferred_type()

        if discovery_info[CONF_TYPE] is None:
            raise ValueError(f"Unable to detect type for device {device.name}")

    if discovery_info[CONF_TYPE] == CONF_TYPE_GPPH_HEATER:
        data[CONF_CLIMATE] = GoldairHeater(device)
    elif discovery_info[CONF_TYPE] == CONF_TYPE_DEHUMIDIFIER:
        data[CONF_CLIMATE] = GoldairDehumidifier(device)
    elif discovery_info[CONF_TYPE] == CONF_TYPE_FAN:
        data[CONF_CLIMATE] = GoldairFan(device)
    elif discovery_info[CONF_TYPE] == CONF_TYPE_GECO_HEATER:
        data[CONF_CLIMATE] = GoldairGECOHeater(device)
    elif discovery_info[CONF_TYPE] == CONF_TYPE_GPCV_HEATER:
        data[CONF_CLIMATE] = GoldairGPCVHeater(device)
    elif discovery_info[CONF_TYPE] == CONF_TYPE_KOGAN_HEATER:
        data[CONF_CLIMATE] = KoganHeater(device)
    else:
        raise ValueError("This device does not support working as a climate device")

    if CONF_CLIMATE in data:
        async_add_entities([data[CONF_CLIMATE]])


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