51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
"""Support for Duotecno switches."""
|
|
from typing import Any
|
|
|
|
from duotecno.unit import SwitchUnit
|
|
|
|
from homeassistant.components.switch import SwitchEntity
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
|
|
from .const import DOMAIN
|
|
from .entity import DuotecnoEntity
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
entry: ConfigEntry,
|
|
async_add_entities: AddEntitiesCallback,
|
|
) -> None:
|
|
"""Set up Velbus switch based on config_entry."""
|
|
cntrl = hass.data[DOMAIN][entry.entry_id]
|
|
async_add_entities(
|
|
DuotecnoSwitch(channel) for channel in cntrl.get_units("SwitchUnit")
|
|
)
|
|
|
|
|
|
class DuotecnoSwitch(DuotecnoEntity, SwitchEntity):
|
|
"""Representation of a switch."""
|
|
|
|
_unit: SwitchUnit
|
|
|
|
@property
|
|
def is_on(self) -> bool:
|
|
"""Return true if the switch is on."""
|
|
return self._unit.is_on()
|
|
|
|
async def async_turn_on(self, **kwargs: Any) -> None:
|
|
"""Instruct the switch to turn on."""
|
|
try:
|
|
await self._unit.turn_on()
|
|
except OSError as err:
|
|
raise HomeAssistantError("Transmit for the turn_on packet failed") from err
|
|
|
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
|
"""Instruct the switch to turn off."""
|
|
try:
|
|
await self._unit.turn_off()
|
|
except OSError as err:
|
|
raise HomeAssistantError("Transmit for the turn_off packet failed") from err
|