water_heater.py

Go to the documentation of this file.

"""Support for Genius Hub water_heater devices."""
 
from __future__ import annotations
 
from homeassistant.components.water_heater import (
    WaterHeaterEntity,
    WaterHeaterEntityFeature,
)
from homeassistant.const import STATE_OFF
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
from . import GeniusHubConfigEntry
from .entity import GeniusHeatingZone
 
STATE_AUTO = "auto"
STATE_MANUAL = "manual"
 
# Genius Hub HW zones support only Off, Override/Boost & Timer modes
HA_OPMODE_TO_GH = {STATE_OFF: "off", STATE_AUTO: "timer", STATE_MANUAL: "override"}
GH_STATE_TO_HA = {
    "off": STATE_OFF,
    "timer": STATE_AUTO,
    "footprint": None,
    "away": None,
    "override": STATE_MANUAL,
    "early": None,
    "test": None,
    "linked": None,
    "other": None,
}
 
GH_HEATERS = ["hot water temperature"]
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    entry: GeniusHubConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up the Genius Hub water heater entities."""
 
    broker = entry.runtime_data
 
    async_add_entities(
        GeniusWaterHeater(broker, z)
        for z in broker.client.zone_objs
        if z.data.get("type") in GH_HEATERS
    )
 
 
class GeniusWaterHeater(GeniusHeatingZone, WaterHeaterEntity):
    """Representation of a Genius Hub water_heater device."""
 
    _attr_supported_features = (
        WaterHeaterEntityFeature.TARGET_TEMPERATURE
        | WaterHeaterEntityFeature.OPERATION_MODE
    )
 
    def __init__(self, broker, zone) -> None:
        """Initialize the water_heater device."""
        super().__init__(broker, zone)
 
        self._max_temp_max_temp = 80.0
        self._min_temp_min_temp = 30.0
 
    @property
    def operation_list(self) -> list[str]:
        """Return the list of available operation modes."""
        return list(HA_OPMODE_TO_GH)
 
    @property
    def current_operation(self) -> str | None:
        """Return the current operation mode."""
        return GH_STATE_TO_HA[self._zone_zone.data["mode"]]
 
    async def async_set_operation_mode(self, operation_mode: str) -> None:
        """Set a new operation mode for this boiler."""
        await self._zone_zone.set_mode(HA_OPMODE_TO_GH[operation_mode])