120 lines
3.9 KiB
Python
120 lines
3.9 KiB
Python
"""Support for Roborock switch."""
|
|
from collections.abc import Callable, Coroutine
|
|
from dataclasses import dataclass
|
|
import logging
|
|
from typing import Any
|
|
|
|
from roborock.roborock_typing import RoborockCommand
|
|
|
|
from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import EntityCategory
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
from homeassistant.util import slugify
|
|
|
|
from .const import DOMAIN
|
|
from .coordinator import RoborockDataUpdateCoordinator
|
|
from .device import RoborockEntity
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
@dataclass
|
|
class RoborockSwitchDescriptionMixin:
|
|
"""Define an entity description mixin for switch entities."""
|
|
|
|
# Gets the status of the switch
|
|
get_value: Callable[[RoborockEntity], Coroutine[Any, Any, dict]]
|
|
# Evaluate the result of get_value to determine a bool
|
|
evaluate_value: Callable[[dict], bool]
|
|
# Sets the status of the switch
|
|
set_command: Callable[[RoborockEntity, bool], Coroutine[Any, Any, dict]]
|
|
|
|
|
|
@dataclass
|
|
class RoborockSwitchDescription(
|
|
SwitchEntityDescription, RoborockSwitchDescriptionMixin
|
|
):
|
|
"""Class to describe an Roborock switch entity."""
|
|
|
|
|
|
SWITCH_DESCRIPTIONS: list[RoborockSwitchDescription] = [
|
|
RoborockSwitchDescription(
|
|
set_command=lambda entity, value: entity.send(
|
|
RoborockCommand.SET_CHILD_LOCK_STATUS, {"lock_status": 1 if value else 0}
|
|
),
|
|
get_value=lambda data: data.send(RoborockCommand.GET_CHILD_LOCK_STATUS),
|
|
evaluate_value=lambda data: data["lock_status"] == 1,
|
|
key="child_lock",
|
|
translation_key="child_lock",
|
|
icon="mdi:account-lock",
|
|
entity_category=EntityCategory.CONFIG,
|
|
),
|
|
RoborockSwitchDescription(
|
|
set_command=lambda entity, value: entity.send(
|
|
RoborockCommand.SET_FLOW_LED_STATUS, {"status": 1 if value else 0}
|
|
),
|
|
get_value=lambda data: data.send(RoborockCommand.GET_FLOW_LED_STATUS),
|
|
evaluate_value=lambda data: data["status"] == 1,
|
|
key="status_indicator",
|
|
translation_key="status_indicator",
|
|
icon="mdi:alarm-light-outline",
|
|
entity_category=EntityCategory.CONFIG,
|
|
),
|
|
]
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
config_entry: ConfigEntry,
|
|
async_add_entities: AddEntitiesCallback,
|
|
) -> None:
|
|
"""Set up Roborock switch platform."""
|
|
|
|
coordinators: dict[str, RoborockDataUpdateCoordinator] = hass.data[DOMAIN][
|
|
config_entry.entry_id
|
|
]
|
|
async_add_entities(
|
|
(
|
|
RoborockSwitchEntity(
|
|
f"{description.key}_{slugify(device_id)}",
|
|
coordinator,
|
|
description,
|
|
)
|
|
for device_id, coordinator in coordinators.items()
|
|
for description in SWITCH_DESCRIPTIONS
|
|
),
|
|
True,
|
|
)
|
|
|
|
|
|
class RoborockSwitchEntity(RoborockEntity, SwitchEntity):
|
|
"""A class to let you turn functionality on Roborock devices on and off."""
|
|
|
|
entity_description: RoborockSwitchDescription
|
|
|
|
def __init__(
|
|
self,
|
|
unique_id: str,
|
|
coordinator: RoborockDataUpdateCoordinator,
|
|
entity_description: RoborockSwitchDescription,
|
|
) -> None:
|
|
"""Create a switch entity."""
|
|
self.entity_description = entity_description
|
|
super().__init__(unique_id, coordinator.device_info, coordinator.api)
|
|
|
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
|
"""Turn off the switch."""
|
|
await self.entity_description.set_command(self, False)
|
|
|
|
async def async_turn_on(self, **kwargs: Any) -> None:
|
|
"""Turn on the switch."""
|
|
await self.entity_description.set_command(self, True)
|
|
|
|
async def async_update(self) -> None:
|
|
"""Update switch."""
|
|
self._attr_is_on = self.entity_description.evaluate_value(
|
|
await self.entity_description.get_value(self)
|
|
)
|