23 lines
662 B
Python
23 lines
662 B
Python
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
|