Initial commit
This commit is contained in:
22
modules/menu/dynamic/get_code/generator.py
Normal file
22
modules/menu/dynamic/get_code/generator.py
Normal file
@@ -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
|
||||
Reference in New Issue
Block a user