climate.py

Go to the documentation of this file.

"""Support for the EPH Controls Ember themostats."""
 
from __future__ import annotations
 
from datetime import timedelta
import logging
from typing import Any
 
from pyephember.pyephember import (
    EphEmber,
    ZoneMode,
    zone_current_temperature,
    zone_is_active,
    zone_is_boost_active,
    zone_is_hot_water,
    zone_mode,
    zone_name,
    zone_target_temperature,
)
import voluptuous as vol
 
from homeassistant.components.climate import (
    PLATFORM_SCHEMA as CLIMATE_PLATFORM_SCHEMA,
    ClimateEntity,
    ClimateEntityFeature,
    HVACAction,
    HVACMode,
)
from homeassistant.const import (
    ATTR_TEMPERATURE,
    CONF_PASSWORD,
    CONF_USERNAME,
    UnitOfTemperature,
)
from homeassistant.core import HomeAssistant
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 
_LOGGER = logging.getLogger(__name__)
 
# Return cached results if last scan was less then this time ago
SCAN_INTERVAL = timedelta(seconds=120)
 
OPERATION_LIST = [HVACMode.HEAT_COOL, HVACMode.HEAT, HVACMode.OFF]
 
PLATFORM_SCHEMA = CLIMATE_PLATFORM_SCHEMA.extend(
    {vol.Required(CONF_USERNAME): cv.string, vol.Required(CONF_PASSWORD): cv.string}
)
 
EPH_TO_HA_STATE = {
    "AUTO": HVACMode.HEAT_COOL,
    "ON": HVACMode.HEAT,
    "OFF": HVACMode.OFF,
}
 
HA_STATE_TO_EPH = {value: key for key, value in EPH_TO_HA_STATE.items()}
 
 
def setup_platform(
    hass: HomeAssistant,
    config: ConfigType,
    add_entities: AddEntitiesCallback,
    discovery_info: DiscoveryInfoType | None = None,
) -> None:
    """Set up the ephember thermostat."""
    username = config.get(CONF_USERNAME)
    password = config.get(CONF_PASSWORD)
 
    try:
        ember = EphEmber(username, password)
        zones = ember.get_zones()
        for zone in zones:
            add_entities([EphEmberThermostat(ember, zone)])
    except RuntimeError:
        _LOGGER.error("Cannot connect to EphEmber")
        return
 
    return
 
 
class EphEmberThermostat(ClimateEntity):
    """Representation of a EphEmber thermostat."""
 
    _attr_hvac_modes = OPERATION_LIST
    _attr_temperature_unit = UnitOfTemperature.CELSIUS
    _enable_turn_on_off_backwards_compatibility = False
 
    def __init__(self, ember, zone):
        """Initialize the thermostat."""
        self._ember_ember = ember
        self._zone_name_zone_name = zone_name(zone)
        self._zone_zone = zone
        self._hot_water_hot_water = zone_is_hot_water(zone)
 
        self._attr_name_attr_name = self._zone_name_zone_name
 
        self._attr_supported_features_attr_supported_features = (
            ClimateEntityFeature.TARGET_TEMPERATURE | ClimateEntityFeature.AUX_HEAT
        )
        self._attr_target_temperature_step_attr_target_temperature_step = 0.5
        if self._hot_water_hot_water:
            self._attr_supported_features_attr_supported_features = ClimateEntityFeature.AUX_HEAT
            self._attr_target_temperature_step_attr_target_temperature_step = None
        self._attr_supported_features_attr_supported_features |= (
            ClimateEntityFeature.TURN_OFF | ClimateEntityFeature.TURN_ON
        )
 
    @property
    def current_temperature(self):
        """Return the current temperature."""
        return zone_current_temperature(self._zone_zone)
 
    @property
    def target_temperature(self):
        """Return the temperature we try to reach."""
        return zone_target_temperature(self._zone_zone)
 
    @property
    def hvac_action(self) -> HVACAction:
        """Return current HVAC action."""
        if zone_is_active(self._zone_zone):
            return HVACAction.HEATING
 
        return HVACAction.IDLE
 
    @property
    def hvac_mode(self) -> HVACMode:
        """Return current operation ie. heat, cool, idle."""
        mode = zone_mode(self._zone_zone)
        return self.map_mode_eph_hassmap_mode_eph_hass(mode)
 
    def set_hvac_mode(self, hvac_mode: HVACMode) -> None:
        """Set the operation mode."""
        mode = self.map_mode_hass_ephmap_mode_hass_eph(hvac_mode)
        if mode is not None:
            self._ember_ember.set_mode_by_name(self._zone_name_zone_name, mode)
        else:
            _LOGGER.error("Invalid operation mode provided %s", hvac_mode)
 
    @property
    def is_aux_heat(self):
        """Return true if aux heater."""
 
        return zone_is_boost_active(self._zone_zone)
 
    def turn_aux_heat_on(self) -> None:
        """Turn auxiliary heater on."""
        self._ember_ember.activate_boost_by_name(
            self._zone_name_zone_name, zone_target_temperature(self._zone_zone)
        )
 
    def turn_aux_heat_off(self) -> None:
        """Turn auxiliary heater off."""
        self._ember_ember.deactivate_boost_by_name(self._zone_name_zone_name)
 
    def set_temperature(self, **kwargs: Any) -> None:
        """Set new target temperature."""
        if (temperature := kwargs.get(ATTR_TEMPERATURE)) is None:
            return
 
        if self._hot_water_hot_water:
            return
 
        if temperature == self.target_temperaturetarget_temperaturetarget_temperature:
            return
 
        if temperature > self.max_tempmax_tempmax_temp or temperature < self.min_tempmin_tempmin_temp:
            return
 
        self._ember_ember.set_target_temperture_by_name(self._zone_name_zone_name, temperature)
 
    @property
    def min_temp(self):
        """Return the minimum temperature."""
        # Hot water temp doesn't support being changed
        if self._hot_water_hot_water:
            return zone_target_temperature(self._zone_zone)
 
        return 5.0
 
    @property
    def max_temp(self):
        """Return the maximum temperature."""
        if self._hot_water_hot_water:
            return zone_target_temperature(self._zone_zone)
 
        return 35.0
 
    def update(self) -> None:
        """Get the latest data."""
        self._zone_zone = self._ember_ember.get_zone(self._zone_name_zone_name)
 
    @staticmethod
    def map_mode_hass_eph(operation_mode):
        """Map from Home Assistant mode to eph mode."""
        return getattr(ZoneMode, HA_STATE_TO_EPH.get(operation_mode), None)
 
    @staticmethod
    def map_mode_eph_hass(operation_mode):
        """Map from eph mode to Home Assistant mode."""
        return EPH_TO_HA_STATE.get(operation_mode.name, HVACMode.HEAT_COOL)