python-kasa/kasa/smart/modules/thermostat.py

"""Module for a Thermostat."""

from __future__ import annotations

from typing import Annotated, Literal

from ...feature import Feature
from ...interfaces.thermostat import Thermostat as ThermostatInterface
from ...interfaces.thermostat import ThermostatState
from ...module import FeatureAttribute, Module
from ..smartmodule import SmartModule


class Thermostat(SmartModule, ThermostatInterface):
    """Implementation of a Thermostat."""

    @property
    def _all_features(self) -> dict[str, Feature]:
        """Get the features for this module and any sub modules."""
        ret: dict[str, Feature] = {}
        if temp_control := self._device.modules.get(Module.TemperatureControl):
            ret.update(**temp_control._module_features)
        if temp_sensor := self._device.modules.get(Module.TemperatureSensor):
            ret.update(**temp_sensor._module_features)
        return ret

    def query(self) -> dict:
        """Query to execute during the update cycle."""
        return {}

    @property
    def state(self) -> bool:
        """Return thermostat state."""
        return self._device.modules[Module.TemperatureControl].state

    async def set_state(self, enabled: bool) -> dict:
        """Set thermostat state."""
        return await self._device.modules[Module.TemperatureControl].set_state(enabled)

    @property
    def mode(self) -> ThermostatState:
        """Return thermostat state."""
        return self._device.modules[Module.TemperatureControl].mode

    @property
    def target_temperature(self) -> Annotated[float, FeatureAttribute()]:
        """Return target temperature."""
        return self._device.modules[Module.TemperatureControl].target_temperature

    async def set_target_temperature(
        self, target: float
    ) -> Annotated[dict, FeatureAttribute()]:
        """Set target temperature."""
        return await self._device.modules[
            Module.TemperatureControl
        ].set_target_temperature(target)

    @property
    def temperature(self) -> Annotated[float, FeatureAttribute()]:
        """Return current humidity in percentage."""
        return self._device.modules[Module.TemperatureSensor].temperature

    @property
    def temperature_unit(self) -> Literal["celsius", "fahrenheit"]:
        """Return current temperature unit."""
        return self._device.modules[Module.TemperatureSensor].temperature_unit

    async def set_temperature_unit(
        self, unit: Literal["celsius", "fahrenheit"]
    ) -> dict:
        """Set the device temperature unit."""
        return await self._device.modules[
            Module.TemperatureSensor
        ].set_temperature_unit(unit)