"""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 . import ( ATTR_DISCOVER_CONFIG, ATTR_DISCOVER_DEVICES, DATA_TELLSTICK, DEFAULT_SIGNAL_REPETITIONS, 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.""" 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 = 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.turn_on() else: self._tellcore_device.turn_off() @property def force_update(self) -> bool: """Will trigger anytime the state property is updated.""" return True