1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
"""Controller holding and managing HUE resources of type `scene`."""
from typing import TYPE_CHECKING, Any
from aiohue.v2.models.feature import DimmingFeaturePut, RecallAction, RecallFeature
from aiohue.v2.models.resource import ResourceTypes
from aiohue.v2.models.room import Room
from aiohue.v2.models.scene import Scene, ScenePut
from aiohue.v2.models.smart_scene import (
SmartScene,
SmartScenePut,
SmartSceneRecall,
SmartSceneRecallAction,
)
from aiohue.v2.models.zone import Zone
from .base import BaseResourcesController, GroupedControllerBase
if TYPE_CHECKING:
from aiohue.v2 import HueBridgeV2
SCENE_TYPES = Scene | SmartScene
class RegularScenesController(BaseResourcesController[type[Scene]]):
"""Controller holding and managing HUE resources of type `scene`."""
item_type = ResourceTypes.SCENE
item_cls = Scene
allow_parser_error = True
async def recall(
self,
id: str,
dynamic: bool = False,
duration: int | None = None,
brightness: float | None = None,
) -> None:
"""Turn on / recall scene."""
action = RecallAction.DYNAMIC_PALETTE if dynamic else RecallAction.ACTIVE
update_obj = ScenePut(recall=RecallFeature(action=action, duration=duration))
if brightness is not None:
update_obj.recall.dimming = DimmingFeaturePut(brightness=brightness)
await self.update(id, update_obj)
def get_group(self, id: str) -> Room | Zone:
"""Get group attached to given scene id."""
scene = self[id]
return next(x for x in self._bridge.groups if x.id == scene.group.rid)
class SmartScenesController(BaseResourcesController[type[SmartScene]]):
"""Controller holding and managing HUE resources of type `smart_scene`."""
item_type = ResourceTypes.SMART_SCENE
item_cls = SmartScene
allow_parser_error = True
async def recall(
self, id: str, action: SmartSceneRecallAction = SmartSceneRecallAction.ACTIVATE
) -> None:
"""Turn on / recall scene."""
update_obj = SmartScenePut(recall=SmartSceneRecall(action=action))
await self.update(id, update_obj)
def get_group(self, id: str) -> Room | Zone:
"""Get group attached to given scene id."""
scene = self[id]
return next(x for x in self._bridge.groups if x.id == scene.group.rid)
class ScenesController(GroupedControllerBase[SCENE_TYPES]):
"""Controller grouping resources of all scene resources."""
def __init__(self, bridge: "HueBridgeV2") -> None:
"""Initialize instance."""
self.scene = RegularScenesController(bridge)
self.smart_scene = SmartScenesController(bridge)
super().__init__(
bridge,
[
self.scene,
self.smart_scene,
],
)
async def recall(self, id: str, *args: Any, **kwargs: Any) -> None:
"""Turn on / recall scene."""
scene = self[id]
# forward call to correct controller
# this method is here for convenience (and backwards compatibility)
if isinstance(scene, SmartScene):
await self.smart_scene.recall(id, *args, **kwargs)
return
await self.scene.recall(id, *args, **kwargs)
def get_group(self, id: str) -> Room | Zone:
"""Get group attached to given scene id."""
scene = self[id]
return next(x for x in self._bridge.groups if x.id == scene.group.rid)
|
