core/homeassistant/components/econet/switch.py

"""Support for using switch with ecoNet thermostats."""

import logging
from typing import Any

from pyeconet.equipment import EquipmentType
from pyeconet.equipment.thermostat import Thermostat, ThermostatOperationMode

from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback

from . import EconetConfigEntry
from .entity import EcoNetEntity

_LOGGER = logging.getLogger(__name__)


async def async_setup_entry(
    hass: HomeAssistant,
    entry: EconetConfigEntry,
    async_add_entities: AddConfigEntryEntitiesCallback,
) -> None:
    """Set up the econet thermostat switch entity."""
    equipment = entry.runtime_data
    async_add_entities(
        EcoNetSwitchAuxHeatOnly(thermostat)
        for thermostat in equipment[EquipmentType.THERMOSTAT]
        if ThermostatOperationMode.EMERGENCY_HEAT in thermostat.modes
    )


class EcoNetSwitchAuxHeatOnly(EcoNetEntity[Thermostat], SwitchEntity):
    """Representation of an aux_heat_only EcoNet switch."""

    def __init__(self, thermostat: Thermostat) -> None:
        """Initialize EcoNet platform."""
        super().__init__(thermostat)
        self._attr_name = f"{thermostat.device_name} emergency heat"
        self._attr_unique_id = (
            f"{thermostat.device_id}_{thermostat.device_name}_auxheat"
        )

    def turn_on(self, **kwargs: Any) -> None:
        """Set the hvacMode to auxHeatOnly."""
        self._econet.set_mode(ThermostatOperationMode.EMERGENCY_HEAT)

    def turn_off(self, **kwargs: Any) -> None:
        """Set the hvacMode back to the prior setting."""
        self._econet.set_mode(ThermostatOperationMode.HEATING)

    @property
    def is_on(self) -> bool:
        """Return true if auxHeatOnly mode is active."""
        return self._econet.mode == ThermostatOperationMode.EMERGENCY_HEAT