switch.py

Go to the documentation of this file.

"""Support for Tellstick switches."""
 
from __future__ import annotations
 
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 .const import (
    ATTR_DISCOVER_CONFIG,
    ATTR_DISCOVER_DEVICES,
    DATA_TELLSTICK,
    DEFAULT_SIGNAL_REPETITIONS,
)
from .entity import TellstickDevice
 
 
def setup_platform(
    hass: HomeAssistant,
    config: ConfigType,
    add_entities: AddEntitiesCallback,
    discovery_info: DiscoveryInfoType | None = None,
) -> None:
    """Set up Tellstick switches."""
    if discovery_info is None or discovery_info[ATTR_DISCOVER_DEVICES] is None:
        return
 
    # Allow platform level override, fallback to module config
    signal_repetitions = discovery_info.get(
        ATTR_DISCOVER_CONFIG, DEFAULT_SIGNAL_REPETITIONS
    )
 
    add_entities(
        [
            TellstickSwitch(hass.data[DATA_TELLSTICK][tellcore_id], signal_repetitions)
            for tellcore_id in discovery_info[ATTR_DISCOVER_DEVICES]
        ],
        True,
    )
 
 
class TellstickSwitch(TellstickDevice, SwitchEntity):
    """Representation of a Tellstick switch."""
 
    _attr_force_update = True
 
    def _parse_ha_data(self, kwargs):
        """Turn the value from HA into something useful."""
 
    def _parse_tellcore_data(self, tellcore_data):
        """Turn the value received from tellcore into something useful."""
 
    def _update_model(self, new_state, data):
        """Update the device entity state to match the arguments."""
        self._state_state_state = new_state
 
    def _send_device_command(self, requested_state, requested_data):
        """Let tellcore update the actual device to the requested state."""
        if requested_state:
            self._tellcore_device_tellcore_device.turn_on()
        else:
            self._tellcore_device_tellcore_device.turn_off()