switch.py

"""TapHome switch integration."""
import typing

from homeassistant.components.switch import DOMAIN, SwitchEntity
from homeassistant.const import CONF_SWITCHES
from homeassistant.core import HomeAssistant

from .add_entry_request import AddEntryRequest
from .const import TAPHOME_PLATFORM
from .coordinator import *
from .taphome_entity import *
from .taphome_sdk import *


class SwitchConfigEntry(TapHomeConfigEntry):
    def __init__(self, device_config: dict):
        super().__init__(device_config)
        self._device_class = self.get_optional("device_class", None)

    @property
    def device_class(self):
        return self._device_class


class TapHomeSwitch(TapHomeEntity[SwitchState], SwitchEntity):
    """Representation of an switch"""

    def __init__(
        self,
        core_id: str,
        config_entry: SwitchConfigEntry,
        coordinator: TapHomeDataUpdateCoordinator,
        switch_service: SwitchService,
    ):
        super().__init__(core_id, config_entry, DOMAIN, coordinator, SwitchState)
        self.switch_service = switch_service
        self._device_class = config_entry.device_class

    @property
    def device_class(self):
        """Return the class of this device, from component DEVICE_CLASSES."""
        return self._device_class

    @property
    def is_on(self):
        """Returns if the switch entity is on or not."""
        if not self.taphome_state is None:
            return self.taphome_state.switch_state == SwitchStates.ON

    async def async_turn_on(self, **kwargs):
        """Turn device on."""
        await self.async_turn(SwitchStates.ON)

    async def async_turn_off(self):
        """Turn device off."""
        await self.async_turn(SwitchStates.OFF)

    async def async_turn(self, switch_state: SwitchStates):
        async with UpdateTapHomeState(self) as state:
            await self.switch_service.async_turn(switch_state, self.taphome_device)
            state.switch_state = switch_state


def setup_platform(
    hass: HomeAssistant,
    config,
    add_entities,
    discovery_info=None,
) -> None:
    """Set up the switch platform."""
    add_entry_requests: typing.List[AddEntryRequest] = hass.data[TAPHOME_PLATFORM][
        CONF_SWITCHES
    ]
    switches = []
    for add_entry_request in add_entry_requests:
        switch_service = SwitchService(add_entry_request.tapHome_api_service)
        switch = TapHomeSwitch(
            add_entry_request.core_id,
            add_entry_request.config_entry,
            add_entry_request.coordinator,
            switch_service,
        )
        switches.append(switch)

    add_entities(switches)