Merge pull request #108 from iKonoTelecomunicaciones/107-delay-node

feat(Delay node): ✨ Create a delay node

menuflow/flow.py

@@ -9,6 +9,7 @@
 from .middlewares import ASRMiddleware, HTTPMiddleware, IRMMiddleware, LLMMiddleware, TTMMiddleware
 from .nodes import (
     CheckTime,
+    Delay,
     Email,
     HTTPRequest,
     Input,
@@ -184,6 +185,10 @@ def node(
             node_initialized = Subroutine(
                 subroutine_node_data=node_data, room=room, default_variables=self.flow_variables
             )
+        elif node_data.get("type") == "delay":
+            node_initialized = Delay(
+                delay_node_data=node_data, room=room, default_variables=self.flow_variables
+            )
         else:
             return
 

menuflow/nodes/__init__.py

@@ -1,5 +1,6 @@
 from .base import Base, convert_to_bool
 from .check_time import CheckTime
+from .delay import Delay
 from .email import Email
 from .http_request import HTTPRequest
 from .input import Input

menuflow/nodes/delay.py

@@ -0,0 +1,29 @@
+from asyncio import sleep
+from typing import Dict
+
+from ..repository import Delay as DelayModel
+from ..room import Room
+from .base import Base
+
+
+class Delay(Base):
+    def __init__(self, delay_node_data: DelayModel, room: Room, default_variables: Dict) -> None:
+        Base.__init__(self, room=room, default_variables=default_variables)
+        self.log = self.log.getChild(delay_node_data.get("id"))
+        self.content: Dict = delay_node_data
+
+    @property
+    def time(self) -> int:
+        self.log.debug(f"Waiting {self.content.get('time', 0)} seconds...")
+        return self.render_data(data=self.content.get("time", 0))
+
+    @property
+    async def o_connection(self) -> str:
+        o_connection = await self.get_o_connection()
+        return self.render_data(data=o_connection)
+
+    async def run(self):
+        self.log.debug(f"Room {self.room.room_id} enters delay node {self.id}")
+        await sleep(self.time)
+        o_connection = await self.o_connection
+        await self.room.update_menu(node_id=o_connection, state=None)

menuflow/repository/__init__.py

@@ -4,6 +4,7 @@
 from .nodes import (
     Case,
     CheckTime,
+    Delay,
     Email,
     HTTPRequest,
     InactivityOptions,

menuflow/repository/nodes/__init__.py

@@ -1,4 +1,5 @@
 from .check_time import CheckTime
+from .delay import Delay
 from .email import Email
 from .http_request import HTTPRequest
 from .input import InactivityOptions, Input

menuflow/repository/nodes/delay.py

@@ -0,0 +1,21 @@
+from __future__ import annotations
+
+from attr import dataclass, ib
+
+from ..flow_object import FlowObject
+
+
+@dataclass
+class Delay(FlowObject):
+    """
+    ## Delay
+    Delay the time (in seconds).
+
+    - id: delay1
+      type: delay
+      time: 10 # seconds
+      o_connection: m2
+    """
+
+    time: int = ib(default=None)
+    o_connection: str = ib(default=None)

menuflow/utils/types.py

@@ -15,6 +15,7 @@ class Nodes(SerializableEnum):
     set_vars = "set_vars"
     subroutine = "subroutine"
     switch = "switch"
+    delay = "delay"
 
 
 class Middlewares(SerializableEnum):