scene.py

"""
Support for HomeSeer Events.
"""

from .hoomseer import HomeseerEntity
from homeassistant.components.scene import Scene

from .const import CONF_ALLOW_EVENTS, DATA_CLIENT, _LOGGER, DOMAIN

DEPENDENCIES = ["homeseer"]


async def async_setup_entry(hass, config_entry, async_add_entities):
    """Set up HomeSeer events as Home Assistant scenes."""
    scenes = []
    homeseer = hass.data[DOMAIN][DATA_CLIENT][config_entry.entry_id]

    allow_events = config_entry.options.get(CONF_ALLOW_EVENTS)

    if allow_events:
        for event in homeseer.events:
            dev = HSScene(event)
            scenes.append(dev)
            _LOGGER.info(f"Added HomeSeer event: {dev.name}")

        async_add_entities(scenes)


class HSScene(Scene):
    """Representation of a HomeSeer event."""

    def __init__(self, event):
        self._event = event
        self._group = self._event.group
        self._name = self._event.name
        self._scene_name = f"{self._group} {self._name}"

    @property
    def name(self):
        """Return the name of the scene."""
        return self._scene_name

    async def async_activate(self):
        """Activate the scene."""
        await self._event.run()