Files
2024-02-29 11:33:00 +01:00

181 lines
7.3 KiB
Python

""" for Climate integration. """
from __future__ import annotations
from datetime import timedelta
import logging
from homeassistant.core import HomeAssistant, callback
from homeassistant.util import Throttle
from homeassistant.config_entries import ConfigEntry
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.components.climate import (
ClimateEntity,
ConfigEntry,
)
from homeassistant.helpers.update_coordinator import (
CoordinatorEntity,
DataUpdateCoordinator,
UpdateFailed,
)
from homeassistant.components.climate.const import (
HVACMode,
FAN_AUTO,
FAN_OFF,
)
from .const import (
DOMAIN,
SUPPORT_FLAGS,
SUPPORTED_HVAC_MODES,
SUPPORTED_FAN_MODES,
FAN_TRANSLATION,
HVAC_TRANSLATION,
)
from .coordinator import KoolnovaCoordinator
from homeassistant.const import (
ATTR_TEMPERATURE,
UnitOfTemperature,
)
from .koolnova.device import Koolnova, Area
from .koolnova.const import (
MIN_TEMP,
MAX_TEMP,
STEP_TEMP,
MIN_TEMP_ORDER,
MAX_TEMP_ORDER,
STEP_TEMP_ORDER,
SysState,
ZoneState,
ZoneClimMode,
ZoneFanMode,
)
_LOGGER = logging.getLogger(__name__)
async def async_setup_entry(hass: HomeAssistant,
entry: ConfigEntry,
async_add_entities: AddEntitiesCallback
) -> None:
"""Setup switch entries"""
entities = []
coordinator = hass.data[DOMAIN]["coordinator"]
device = hass.data[DOMAIN]["device"]
for area in device.areas:
entities.append(AreaClimateEntity(coordinator, device, area))
async_add_entities(entities)
class AreaClimateEntity(CoordinatorEntity, ClimateEntity):
""" Reperesentation of a climate entity """
# pylint: disable = too-many-instance-attributes
_attr_supported_features: int = SUPPORT_FLAGS
_attr_temperature_unit: str = UnitOfTemperature.CELSIUS
_attr_hvac_modes: list[HVACMode] = SUPPORTED_HVAC_MODES
_attr_fan_modes: list[str] = SUPPORTED_FAN_MODES
_attr_hvac_mode: HVACMode = HVACMode.OFF
_attr_fan_mode: str = FAN_OFF
_attr_min_temp: float = MIN_TEMP
_attr_max_temp: float = MAX_TEMP
_attr_precision: float = STEP_TEMP
_attr_target_temperature_high: float = MAX_TEMP_ORDER
_attr_target_temperature_low: float = MIN_TEMP_ORDER
_attr_target_temperature_step: float = STEP_TEMP_ORDER
_enable_turn_on_off_backwards_compatibility: bool = False
def __init__(self,
coordinator: KoolnovaCoordinator, # pylint: disable=unused-argument
device: Koolnova, # pylint: disable=unused-argument
area: Area, # pylint: disable=unused-argument
) -> None:
""" Class constructor """
super().__init__(coordinator)
self._device = device
self._area = area
self._attr_name = f"{device.name} {area.name} area"
self._attr_device_info = device.device_info
self._attr_unique_id = f"{DOMAIN}-{area.name}-area-climate"
self._attr_current_temperature = area.real_temp
self._attr_target_temperature = area.order_temp
self._attr_fan_mode = FAN_TRANSLATION[int(self._area.fan_mode)]
self._attr_hvac_mode = self._translate_to_hvac_mode()
def _translate_to_hvac_mode(self) -> int:
""" translate area state and clim mode to HA hvac mode """
ret = 0
if self._area.state == ZoneState.STATE_OFF:
ret = HVACMode.OFF
else:
ret = HVAC_TRANSLATION[int(self._area.clim_mode)]
return ret
async def async_set_temperature(self,
**kwargs,
) -> None:
""" set new target temperature """
_LOGGER.debug("[Climate {}] set target temp - kwargs: {}".format(self._area.id_zone, kwargs))
if "temperature" in kwargs:
target_temp = kwargs.get("temperature")
ret = await self._device.async_set_area_target_temp(zone_id = self._area.id_zone, temp = target_temp)
if not ret:
_LOGGER.error("Error sending target temperature for area id {}".format(self._area.id_zone))
else:
_LOGGER.warning("Target temperature not defined for climate id {}".format(self._area.id_zone))
await self.coordinator.async_request_refresh()
async def async_set_fan_mode(self,
fan_mode:str,
) -> None:
""" set new target fan mode """
_LOGGER.debug("[Climate {}] set new fan mode: {}".format(self._area.id_zone, fan_mode))
for k,v in FAN_TRANSLATION.items():
if v == fan_mode:
opt = k
break
ret = await self._device.async_set_area_fan_mode(zone_id = self._area.id_zone,
mode = ZoneFanMode(opt))
if not ret:
_LOGGER.exception("Error setting new fan value for area id {}".format(self._area.id_zone))
await self.coordinator.async_request_refresh()
async def async_set_hvac_mode(self,
hvac_mode:HVACMode,
) -> None:
""" set new target hvac mode """
_LOGGER.debug("[Climate {}] set new hvac mode: {}".format(self._area.id_zone, hvac_mode))
opt = 0
for k,v in HVAC_TRANSLATION.items():
if v == hvac_mode:
opt = k
break
ret = await self._device.async_set_area_clim_mode(zone_id = self._area.id_zone,
mode = ZoneClimMode(opt))
if not ret:
_LOGGER.exception("Error setting new hvac value for area id {}".format(self._area.id_zone))
await self.coordinator.async_request_refresh()
@callback
def _handle_coordinator_update(self) -> None:
""" Handle updated data from the coordinator """
for _cur_area in self.coordinator.data['areas']:
if _cur_area.id_zone == self._area.id_zone:
_LOGGER.debug("[UPDATE] [Climate {}] temp:{} - target:{} - state: {} - hvac:{} - fan:{}".format(_cur_area.id_zone,
_cur_area.real_temp,
_cur_area.order_temp,
_cur_area.state,
_cur_area.clim_mode,
_cur_area.fan_mode))
self._area = _cur_area
self._attr_current_temperature = _cur_area.real_temp
self._attr_target_temperature = _cur_area.order_temp
if _cur_area.state == ZoneState.STATE_OFF:
self._attr_hvac_mode = HVACMode.OFF
else:
self._attr_hvac_mode = HVAC_TRANSLATION[int(_cur_area.clim_mode)]
self._attr_fan_mode = FAN_TRANSLATION[int(_cur_area.fan_mode)]
self.async_write_ha_state()