diff --git a/pyvlx/node_updater.py b/pyvlx/node_updater.py index 58fb4de4..d3bd8395 100644 --- a/pyvlx/node_updater.py +++ b/pyvlx/node_updater.py @@ -3,7 +3,8 @@ FrameGetAllNodesInformationNotification, FrameNodeStatePositionChangedNotification) from .opening_device import OpeningDevice -from .parameter import Position +from .lightening_device import LighteningDevice +from .parameter import Intensity, Position class NodeUpdater(): @@ -22,6 +23,9 @@ async def process_frame(self, frame): if isinstance(node, OpeningDevice): node.position = Position(frame.current_position) await node.after_update() + elif isinstance(node, LighteningDevice): + node.intensity = Intensity(frame.current_position) + await node.after_update() elif isinstance(frame, FrameGetAllNodesInformationNotification): if frame.node_id not in self.pyvlx.nodes: return @@ -29,3 +33,6 @@ async def process_frame(self, frame): if isinstance(node, OpeningDevice): node.position = Position(frame.current_position) await node.after_update() + elif isinstance(node, LighteningDevice): + node.intensity = Intensity(frame.current_position) + await node.after_update()