Added config profiles #23 (Massive changes in caonfig_data_import)
Removed example config #24
Fixed #25
Added current node display #26
Tested ISO auto-mount: Currently attached iso will be swapped (if any present)
Unified promts styling & other menu improvements
This commit is contained in:
OVERLORD7F
2025-09-18 15:59:44 +03:00
parent 6c272e46ab
commit 29af815ad3
4 changed files with 360 additions and 163 deletions

47
main.py
View File

@@ -7,43 +7,60 @@ from data_pools_api import *
from disk_edit_mode import *
from rich.panel import Panel
from rich.console import Console , Align
SVMU_ver="0.35-dev"
SVMU_ver="0.4-dev"
config_relative_path = os.path.join(os.getcwd() , 'SpaceVM_Utility.conf') #config in the same directory with main.py
print("Reading config from:", os.path.abspath(config_relative_path))
if not os.path.exists(config_relative_path):
print(f"Config file not found: {config_relative_path}")
menu_choice=0
# Initialize console and clear screen
console = Console()
os.system('cls' if os.name=='nt' else 'clear')
# Initialize config path and ensure it points to a valid profile
config_relative_path = get_default_config_path()
# Get initial configuration and show startup screen
skip_startup_splash = get_skip_startup_splash(config_relative_path)
show_startup_logo(skip_startup_splash, SVMU_ver) #shows startup splash (ASCII art)
menu_choice=0
while(menu_choice != ""): #main menu loop
check_config(config_relative_path)
profile_name = os.path.basename(os.path.splitext(config_relative_path)[0]) #converting full path to pretty file name
skip_startup_splash, base_url, api_key, data_pool_uuid, data_pool_name, vm_uuids, vm_names, disk1_size, disk2_size, disk3_size, disk_interface, preallocation, iso_uuid, iso_name = config_import(config_relative_path) #importing API-KEY / IP / DATA POOL UUID / VM-UUIDs from config
menu_options=f"[gold bold][1] [grey53 italic]Manage utility config\n[/grey53 italic] \
vm_pretty_names = ', '.join(vm_names)
menu_options=f"[gold bold][1] [grey53 italic]Utiliy Configuration / Profiles\n[/grey53 italic] \
\n[gold bold][2] [grey53 italic]Enter disk edit mode[/grey53 italic]\n \
\n[gold bold][3] [grey53 italic]Show breif cluster overview[/grey53 italic]\n \
\n[gold bold][4] [grey53 italic]Enter VM menu[/grey53 italic]\n \
\n[gold bold][5] [grey53 italic]Show data pools[/grey53 italic]\n \
\n\n[green_yellow bold]ENTER - exit Utility[/]\n\n \
[underline bold grey53]Currently imported config:[/]\n \
[bold grey53]Connected to Controller: [bright_yellow]{base_url}[/]\n Selected Data Pool: [bright_yellow]{data_pool_name}[/]\n Selected VMs:\n [bright_yellow]{vm_names}[/]\n Auto-mount ISO: [bright_cyan]{iso_name}[/]"
[underline bold grey53]Current profile:[/] [bold green]{profile_name}[/]\n \
[bold grey53]Connected to Controller: [bright_yellow]{base_url}[/]\n Selected Data Pool: [bright_yellow]{data_pool_name}[/]\n Selected VMs:\n [bright_yellow]{vm_pretty_names}[/]\n Auto-mount ISO: [bright_cyan]{iso_name}[/]"
menu_options=Align.center(menu_options, vertical="middle")
menu_subtitle = "[blue bold][link=https://github.com/OVERLORD7F/SVMU]:wrench: Project_GitHub[/link] [yellow]| [magenta bold][link=https://spacevm.ru/docs/]:books: SpaceVM_Docs[/link] [yellow]| [red bold][link=https://comptek.ru]:briefcase: Comptek[/link]"
console.print(Panel(menu_options, title=f"[bold magenta]SpaceVM Utility {SVMU_ver} - Main Menu" , subtitle = menu_subtitle, subtitle_align="right" , style="yellow" , width=150 , padding = 2))
menu_choice=str(input("\n>>> "))
menu_choice = console.input("[bold yellow]\n>>> [/]")
if menu_choice == "1":
config_menu(base_url, api_key, config_relative_path)
result = config_menu(base_url, api_key, config_relative_path)
# Check if we got a new profile path
if isinstance(result, str):
# Update config path and reload
config_relative_path = result
os.system('cls' if os.name=='nt' else 'clear')
continue
# Check if we need to reload config
elif isinstance(result, bool) and result:
# Clear screen and continue to reload config
os.system('cls' if os.name=='nt' else 'clear')
continue
if menu_choice == "2":
disk_edit_mode(base_url , api_key , data_pool_uuid , vm_uuids, disk1_size, disk2_size, disk3_size, disk_interface, preallocation, iso_uuid)
os.system('cls' if os.name=='nt' else 'clear')
if menu_choice == "3":
cluster_info(base_url , api_key)
os.system('cls' if os.name=='nt' else 'clear')
if menu_choice == "4":
vm_menu(base_url , api_key, vm_uuids, config_relative_path)
os.system('cls' if os.name=='nt' else 'clear')
if menu_choice == "5":
show_data_pools(base_url , api_key)
os.system('cls' if os.name=='nt' else 'clear') #clears screen before looping back to main menu
os.system('cls' if os.name=='nt' else 'clear')
console.print("[red bold]Exiting Utility ")