switch.py

Go to the documentation of this file.

"""Support for EufyHome switches."""
 
from __future__ import annotations
 
from typing import Any
 
import lakeside
 
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
 
 
def setup_platform(
    hass: HomeAssistant,
    config: ConfigType,
    add_entities: AddEntitiesCallback,
    discovery_info: DiscoveryInfoType | None = None,
) -> None:
    """Set up EufyHome switches."""
    if discovery_info is None:
        return
    add_entities([EufyHomeSwitch(discovery_info)], True)
 
 
class EufyHomeSwitch(SwitchEntity):
    """Representation of a EufyHome switch."""
 
    def __init__(self, device):
        """Initialize the light."""
 
        self._state_state = None
        self._name_name = device["name"]
        self._address_address = device["address"]
        self._code_code = device["code"]
        self._type_type = device["type"]
        self._switch_switch = lakeside.switch(self._address_address, self._code_code, self._type_type)
        self._switch_switch.connect()
 
    def update(self) -> None:
        """Synchronise state from the switch."""
        self._switch_switch.update()
        self._state_state = self._switch_switch.power
 
    @property
    def unique_id(self):
        """Return the ID of this light."""
        return self._address_address
 
    @property
    def name(self):
        """Return the name of the device if any."""
        return self._name_name
 
    @property
    def is_on(self):
        """Return true if device is on."""
        return self._state_state
 
    def turn_on(self, **kwargs: Any) -> None:
        """Turn the specified switch on."""
        try:
            self._switch_switch.set_state(True)
        except BrokenPipeError:
            self._switch_switch.connect()
            self._switch_switch.set_state(power=True)
 
    def turn_off(self, **kwargs: Any) -> None:
        """Turn the specified switch off."""
        try:
            self._switch_switch.set_state(False)
        except BrokenPipeError:
            self._switch_switch.connect()
            self._switch_switch.set_state(False)