Updating profile info works

modules/menu/dynamic/get_code/generator.py

@@ -1,22 +0,0 @@
-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

@@ -1,7 +1,8 @@
 from ...menu import menu
 from cursesmenu import CursesMenu
 from modules.custom_items.DynamicSubmenu import DynamicSubmenuItem
-from .generator import generate_get_code_menu
+from ..common_generators.valid_sessions_generator import generate_sessions_list
+from ...callbacks.get_code import get_code_callback
 
 
 get_code_submenu = CursesMenu(
@@ -10,6 +11,7 @@ get_code_submenu = CursesMenu(
 
 get_code_submenu_item = DynamicSubmenuItem(
     text='Get confirmation code',
-    generator=generate_get_code_menu,
+    generator=generate_sessions_list,
+    args=['Get code for', get_code_callback],
     menu=menu
 )