mirror of https://github.com/home-assistant/core
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
"""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 = 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()
|