64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
"""Support for Tuya scenes."""
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from tuya_iot import TuyaHomeManager, TuyaScene
|
|
|
|
from homeassistant.components.scene import Scene
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity import DeviceInfo
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
|
|
from . import HomeAssistantTuyaData
|
|
from .const import DOMAIN
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
|
|
) -> None:
|
|
"""Set up Tuya scenes."""
|
|
hass_data: HomeAssistantTuyaData = hass.data[DOMAIN][entry.entry_id]
|
|
scenes = await hass.async_add_executor_job(hass_data.home_manager.query_scenes)
|
|
async_add_entities(
|
|
TuyaSceneEntity(hass_data.home_manager, scene) for scene in scenes
|
|
)
|
|
|
|
|
|
class TuyaSceneEntity(Scene):
|
|
"""Tuya Scene Remote."""
|
|
|
|
_should_poll = False
|
|
|
|
def __init__(self, home_manager: TuyaHomeManager, scene: TuyaScene) -> None:
|
|
"""Init Tuya Scene."""
|
|
super().__init__()
|
|
self._attr_unique_id = f"tys{scene.scene_id}"
|
|
self.home_manager = home_manager
|
|
self.scene = scene
|
|
|
|
@property
|
|
def name(self) -> str | None:
|
|
"""Return Tuya scene name."""
|
|
return self.scene.name
|
|
|
|
@property
|
|
def device_info(self) -> DeviceInfo:
|
|
"""Return a device description for device registry."""
|
|
return DeviceInfo(
|
|
identifiers={(DOMAIN, f"{self.unique_id}")},
|
|
manufacturer="tuya",
|
|
name=self.scene.name,
|
|
model="Tuya Scene",
|
|
)
|
|
|
|
@property
|
|
def available(self) -> bool:
|
|
"""Return if the scene is enabled."""
|
|
return self.scene.enabled
|
|
|
|
def activate(self, **kwargs: Any) -> None:
|
|
"""Activate the scene."""
|
|
self.home_manager.trigger_scene(self.scene.home_id, self.scene.scene_id)
|