switch.py

Go to the documentation of this file.

"""Support for the for Danfoss Air HRV sswitches."""
 
from __future__ import annotations
 
import logging
from typing import Any
 
from pydanfossair.commands import ReadCommand, UpdateCommand
 
from homeassistant.components.switch import SwitchEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 
from . import DOMAIN as DANFOSS_AIR_DOMAIN
 
_LOGGER = logging.getLogger(__name__)
 
 
def setup_platform(
    hass: HomeAssistant,
    config: ConfigType,
    add_entities: AddEntitiesCallback,
    discovery_info: DiscoveryInfoType | None = None,
) -> None:
    """Set up the Danfoss Air HRV switch platform."""
    data = hass.data[DANFOSS_AIR_DOMAIN]
 
    switches = [
        [
            "Danfoss Air Boost",
            ReadCommand.boost,
            UpdateCommand.boost_activate,
            UpdateCommand.boost_deactivate,
        ],
        [
            "Danfoss Air Bypass",
            ReadCommand.bypass,
            UpdateCommand.bypass_activate,
            UpdateCommand.bypass_deactivate,
        ],
        [
            "Danfoss Air Automatic Bypass",
            ReadCommand.automatic_bypass,
            UpdateCommand.bypass_activate,
            UpdateCommand.bypass_deactivate,
        ],
    ]
 
    add_entities(
        DanfossAir(data, switch[0], switch[1], switch[2], switch[3])
        for switch in switches
    )
 
 
class DanfossAir(SwitchEntity):
    """Representation of a Danfoss Air HRV Switch."""
 
    def __init__(self, data, name, state_command, on_command, off_command):
        """Initialize the switch."""
        self._data_data = data
        self._name_name = name
        self._state_command_state_command = state_command
        self._on_command_on_command = on_command
        self._off_command_off_command = off_command
        self._state_state = None
 
    @property
    def name(self):
        """Return the name of the switch."""
        return self._name_name
 
    @property
    def is_on(self):
        """Return true if switch is on."""
        return self._state_state
 
    def turn_on(self, **kwargs: Any) -> None:
        """Turn the switch on."""
        _LOGGER.debug("Turning on switch with command %s", self._on_command_on_command)
        self._data_data.update_state(self._on_command_on_command, self._state_command_state_command)
 
    def turn_off(self, **kwargs: Any) -> None:
        """Turn the switch off."""
        _LOGGER.debug("Turning off switch with command %s", self._off_command_off_command)
        self._data_data.update_state(self._off_command_off_command, self._state_command_state_command)
 
    def update(self) -> None:
        """Update the switch's state."""
        self._data_data.update()
 
        self._state_state = self._data_data.get_value(self._state_command_state_command)
        if self._state_state is None:
            _LOGGER.debug("Could not get data for %s", self._state_command_state_command)