mirror of
https://github.com/OVERLORD7F/SVMU.git
synced 2025-10-01 21:52:47 +03:00
0.4-dev
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:
47
main.py
47
main.py
@@ -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 ")
|
||||
|
Reference in New Issue
Block a user