"""Support for Vera scenes.""" import logging from homeassistant.components.scene import Scene from homeassistant.util import slugify from . import VERA_CONTROLLER, VERA_ID_FORMAT, VERA_SCENES _LOGGER = logging.getLogger(__name__) def setup_platform(hass, config, add_entities, discovery_info=None): """Set up the Vera scenes.""" add_entities( [ VeraScene(scene, hass.data[VERA_CONTROLLER]) for scene in hass.data[VERA_SCENES] ], True, ) class VeraScene(Scene): """Representation of a Vera scene entity.""" def __init__(self, vera_scene, controller): """Initialize the scene.""" self.vera_scene = vera_scene self.controller = controller self._name = self.vera_scene.name # Append device id to prevent name clashes in HA. self.vera_id = VERA_ID_FORMAT.format( slugify(vera_scene.name), vera_scene.scene_id ) def update(self): """Update the scene status.""" self.vera_scene.refresh() def activate(self): """Activate the scene.""" self.vera_scene.activate() @property def name(self): """Return the name of the scene.""" return self._name @property def device_state_attributes(self): """Return the state attributes of the scene.""" return {"vera_scene_id": self.vera_scene.vera_scene_id}