123 lines
		
	
	
		
			4.0 KiB
		
	
	
	
		
			Python
		
	
	
			
		
		
	
	
			123 lines
		
	
	
		
			4.0 KiB
		
	
	
	
		
			Python
		
	
	
"""Intellifire Climate Entities."""
 | 
						|
 | 
						|
from __future__ import annotations
 | 
						|
 | 
						|
from typing import Any
 | 
						|
 | 
						|
from homeassistant.components.climate import (
 | 
						|
    ClimateEntity,
 | 
						|
    ClimateEntityDescription,
 | 
						|
    ClimateEntityFeature,
 | 
						|
    HVACMode,
 | 
						|
)
 | 
						|
from homeassistant.config_entries import ConfigEntry
 | 
						|
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
 | 
						|
from homeassistant.core import HomeAssistant
 | 
						|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 | 
						|
 | 
						|
from . import IntellifireDataUpdateCoordinator
 | 
						|
from .const import DEFAULT_THERMOSTAT_TEMP, DOMAIN, LOGGER
 | 
						|
from .entity import IntellifireEntity
 | 
						|
 | 
						|
INTELLIFIRE_CLIMATES: tuple[ClimateEntityDescription, ...] = (
 | 
						|
    ClimateEntityDescription(key="climate", name="Thermostat"),
 | 
						|
)
 | 
						|
 | 
						|
 | 
						|
async def async_setup_entry(
 | 
						|
    hass: HomeAssistant,
 | 
						|
    entry: ConfigEntry,
 | 
						|
    async_add_entities: AddEntitiesCallback,
 | 
						|
) -> None:
 | 
						|
    """Configure the fan entry.."""
 | 
						|
    coordinator: IntellifireDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
 | 
						|
 | 
						|
    if coordinator.data.has_thermostat:
 | 
						|
        async_add_entities(
 | 
						|
            IntellifireClimate(
 | 
						|
                coordinator=coordinator,
 | 
						|
                description=description,
 | 
						|
            )
 | 
						|
            for description in INTELLIFIRE_CLIMATES
 | 
						|
        )
 | 
						|
 | 
						|
 | 
						|
class IntellifireClimate(IntellifireEntity, ClimateEntity):
 | 
						|
    """Intellifire climate entity."""
 | 
						|
 | 
						|
    entity_description: ClimateEntityDescription
 | 
						|
 | 
						|
    _attr_hvac_modes = [HVACMode.HEAT, HVACMode.OFF]
 | 
						|
    _attr_min_temp = 0
 | 
						|
    _attr_max_temp = 37
 | 
						|
    _attr_supported_features = (
 | 
						|
        ClimateEntityFeature.TARGET_TEMPERATURE
 | 
						|
        | ClimateEntityFeature.TURN_OFF
 | 
						|
        | ClimateEntityFeature.TURN_ON
 | 
						|
    )
 | 
						|
    _attr_target_temperature_step = 1.0
 | 
						|
    _attr_temperature_unit = UnitOfTemperature.CELSIUS
 | 
						|
    last_temp = DEFAULT_THERMOSTAT_TEMP
 | 
						|
    _enable_turn_on_off_backwards_compatibility = False
 | 
						|
 | 
						|
    def __init__(
 | 
						|
        self,
 | 
						|
        coordinator: IntellifireDataUpdateCoordinator,
 | 
						|
        description: ClimateEntityDescription,
 | 
						|
    ) -> None:
 | 
						|
        """Configure climate entry - and override last_temp if the thermostat is currently on."""
 | 
						|
        super().__init__(coordinator, description)
 | 
						|
 | 
						|
        if coordinator.data.thermostat_on:
 | 
						|
            self.last_temp = coordinator.data.thermostat_setpoint_c
 | 
						|
 | 
						|
    @property
 | 
						|
    def hvac_mode(self) -> HVACMode:
 | 
						|
        """Return current hvac mode."""
 | 
						|
        if self.coordinator.read_api.data.thermostat_on:
 | 
						|
            return HVACMode.HEAT
 | 
						|
        return HVACMode.OFF
 | 
						|
 | 
						|
    async def async_set_temperature(self, **kwargs: Any) -> None:
 | 
						|
        """Turn on thermostat by setting a target temperature."""
 | 
						|
        raw_target_temp = kwargs[ATTR_TEMPERATURE]
 | 
						|
        self.last_temp = int(raw_target_temp)
 | 
						|
        LOGGER.debug(
 | 
						|
            "Setting target temp to %sc %sf",
 | 
						|
            int(raw_target_temp),
 | 
						|
            (raw_target_temp * 9 / 5) + 32,
 | 
						|
        )
 | 
						|
        await self.coordinator.control_api.set_thermostat_c(
 | 
						|
            temp_c=self.last_temp,
 | 
						|
        )
 | 
						|
 | 
						|
    @property
 | 
						|
    def current_temperature(self) -> float:
 | 
						|
        """Return the current temperature."""
 | 
						|
        return float(self.coordinator.read_api.data.temperature_c)
 | 
						|
 | 
						|
    @property
 | 
						|
    def target_temperature(self) -> float:
 | 
						|
        """Return target temperature."""
 | 
						|
        return float(self.coordinator.read_api.data.thermostat_setpoint_c)
 | 
						|
 | 
						|
    async def async_set_hvac_mode(self, hvac_mode: HVACMode) -> None:
 | 
						|
        """Set HVAC mode to normal or thermostat control."""
 | 
						|
        LOGGER.debug(
 | 
						|
            "Setting mode to [%s] - using last temp: %s", hvac_mode, self.last_temp
 | 
						|
        )
 | 
						|
 | 
						|
        if hvac_mode == HVACMode.OFF:
 | 
						|
            await self.coordinator.control_api.turn_off_thermostat()
 | 
						|
            return
 | 
						|
 | 
						|
        # hvac_mode == HVACMode.HEAT
 | 
						|
        # 1) Set the desired target temp
 | 
						|
        await self.coordinator.control_api.set_thermostat_c(
 | 
						|
            temp_c=self.last_temp,
 | 
						|
        )
 | 
						|
 | 
						|
        # 2) Make sure the fireplace is on!
 | 
						|
        if not self.coordinator.read_api.data.is_on:
 | 
						|
            await self.coordinator.control_api.flame_on()
 |