143 lines
4.9 KiB
Python
143 lines
4.9 KiB
Python
"""Support for WLED switches."""
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from homeassistant.components.switch import SwitchEntity
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import ENTITY_CATEGORY_CONFIG
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
|
|
from .const import (
|
|
ATTR_DURATION,
|
|
ATTR_FADE,
|
|
ATTR_TARGET_BRIGHTNESS,
|
|
ATTR_UDP_PORT,
|
|
DOMAIN,
|
|
)
|
|
from .coordinator import WLEDDataUpdateCoordinator
|
|
from .helpers import wled_exception_handler
|
|
from .models import WLEDEntity
|
|
|
|
PARALLEL_UPDATES = 1
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
entry: ConfigEntry,
|
|
async_add_entities: AddEntitiesCallback,
|
|
) -> None:
|
|
"""Set up WLED switch based on a config entry."""
|
|
coordinator: WLEDDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
|
|
|
|
switches = [
|
|
WLEDNightlightSwitch(coordinator),
|
|
WLEDSyncSendSwitch(coordinator),
|
|
WLEDSyncReceiveSwitch(coordinator),
|
|
]
|
|
async_add_entities(switches)
|
|
|
|
|
|
class WLEDNightlightSwitch(WLEDEntity, SwitchEntity):
|
|
"""Defines a WLED nightlight switch."""
|
|
|
|
_attr_icon = "mdi:weather-night"
|
|
_attr_entity_category = ENTITY_CATEGORY_CONFIG
|
|
|
|
def __init__(self, coordinator: WLEDDataUpdateCoordinator) -> None:
|
|
"""Initialize WLED nightlight switch."""
|
|
super().__init__(coordinator=coordinator)
|
|
self._attr_name = f"{coordinator.data.info.name} Nightlight"
|
|
self._attr_unique_id = f"{coordinator.data.info.mac_address}_nightlight"
|
|
|
|
@property
|
|
def extra_state_attributes(self) -> dict[str, Any] | None:
|
|
"""Return the state attributes of the entity."""
|
|
return {
|
|
ATTR_DURATION: self.coordinator.data.state.nightlight.duration,
|
|
ATTR_FADE: self.coordinator.data.state.nightlight.fade,
|
|
ATTR_TARGET_BRIGHTNESS: self.coordinator.data.state.nightlight.target_brightness,
|
|
}
|
|
|
|
@property
|
|
def is_on(self) -> bool:
|
|
"""Return the state of the switch."""
|
|
return bool(self.coordinator.data.state.nightlight.on)
|
|
|
|
@wled_exception_handler
|
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
|
"""Turn off the WLED nightlight switch."""
|
|
await self.coordinator.wled.nightlight(on=False)
|
|
|
|
@wled_exception_handler
|
|
async def async_turn_on(self, **kwargs: Any) -> None:
|
|
"""Turn on the WLED nightlight switch."""
|
|
await self.coordinator.wled.nightlight(on=True)
|
|
|
|
|
|
class WLEDSyncSendSwitch(WLEDEntity, SwitchEntity):
|
|
"""Defines a WLED sync send switch."""
|
|
|
|
_attr_icon = "mdi:upload-network-outline"
|
|
_attr_entity_category = ENTITY_CATEGORY_CONFIG
|
|
|
|
def __init__(self, coordinator: WLEDDataUpdateCoordinator) -> None:
|
|
"""Initialize WLED sync send switch."""
|
|
super().__init__(coordinator=coordinator)
|
|
self._attr_name = f"{coordinator.data.info.name} Sync Send"
|
|
self._attr_unique_id = f"{coordinator.data.info.mac_address}_sync_send"
|
|
|
|
@property
|
|
def extra_state_attributes(self) -> dict[str, Any] | None:
|
|
"""Return the state attributes of the entity."""
|
|
return {ATTR_UDP_PORT: self.coordinator.data.info.udp_port}
|
|
|
|
@property
|
|
def is_on(self) -> bool:
|
|
"""Return the state of the switch."""
|
|
return bool(self.coordinator.data.state.sync.send)
|
|
|
|
@wled_exception_handler
|
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
|
"""Turn off the WLED sync send switch."""
|
|
await self.coordinator.wled.sync(send=False)
|
|
|
|
@wled_exception_handler
|
|
async def async_turn_on(self, **kwargs: Any) -> None:
|
|
"""Turn on the WLED sync send switch."""
|
|
await self.coordinator.wled.sync(send=True)
|
|
|
|
|
|
class WLEDSyncReceiveSwitch(WLEDEntity, SwitchEntity):
|
|
"""Defines a WLED sync receive switch."""
|
|
|
|
_attr_icon = "mdi:download-network-outline"
|
|
_attr_entity_category = ENTITY_CATEGORY_CONFIG
|
|
|
|
def __init__(self, coordinator: WLEDDataUpdateCoordinator) -> None:
|
|
"""Initialize WLED sync receive switch."""
|
|
super().__init__(coordinator=coordinator)
|
|
self._attr_name = f"{coordinator.data.info.name} Sync Receive"
|
|
self._attr_unique_id = f"{coordinator.data.info.mac_address}_sync_receive"
|
|
|
|
@property
|
|
def extra_state_attributes(self) -> dict[str, Any] | None:
|
|
"""Return the state attributes of the entity."""
|
|
return {ATTR_UDP_PORT: self.coordinator.data.info.udp_port}
|
|
|
|
@property
|
|
def is_on(self) -> bool:
|
|
"""Return the state of the switch."""
|
|
return bool(self.coordinator.data.state.sync.receive)
|
|
|
|
@wled_exception_handler
|
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
|
"""Turn off the WLED sync receive switch."""
|
|
await self.coordinator.wled.sync(receive=False)
|
|
|
|
@wled_exception_handler
|
|
async def async_turn_on(self, **kwargs: Any) -> None:
|
|
"""Turn on the WLED sync receive switch."""
|
|
await self.coordinator.wled.sync(receive=True)
|