Initial commit

modules/menu/dynamic/get_code/__init__.py

@@ -0,0 +1 @@
+from .submenu import get_code_submenu_item

modules/menu/dynamic/get_code/generator.py

@@ -0,0 +1,22 @@
+import os
+from cursesmenu import CursesMenu
+from cursesmenu.items import FunctionItem
+
+from modules.config import sessions
+from modules.config.models import SessionConfig
+from ...callbacks.get_code import get_code_callback
+
+
+def generate_get_code_menu() -> CursesMenu:
+    submenu = CursesMenu(
+        title='Sessions'
+    )
+    for session_name in sessions.keys():
+        if os.path.isfile(f'{session_name}.session'):
+            submenu.items.append(FunctionItem(
+                f'{session_name} - {SessionConfig(**sessions[session_name])}',
+                function=get_code_callback,
+                args=[session_name]
+            ))
+
+    return submenu

modules/menu/dynamic/get_code/submenu.py

@@ -0,0 +1,15 @@
+from ...menu import menu
+from cursesmenu import CursesMenu
+from modules.custom_items.DynamicSubmenu import DynamicSubmenuItem
+from .generator import generate_get_code_menu
+
+
+get_code_submenu = CursesMenu(
+    title='Sessions'
+)
+
+get_code_submenu_item = DynamicSubmenuItem(
+    text='Get confirmation code',
+    generator=generate_get_code_menu,
+    menu=menu
+)