- Small code cleanup. config_data_import <400 lines!

- Tested new version 6.5.8
- Added elapsed timer #31 (Courses only)
- Corrected version label in menus
- New splash screens (highly requested feature)
- Moved spash screens files to ./assets directory
- Correct colors for menu  options

README.md

@@ -9,7 +9,7 @@ Written in python, uses [SpaceVM API](https://spacevm.ru/docs/6.5/api/) to colle
 
 >[!NOTE]
 >_This utility is focused on managing virtual disks_<br>
->_Works with SpaceVM 6.5.5 / 6.5.6 / 6.5.7_ <br>
+>_Works with SpaceVM 6.5.5 / 6.5.6 / 6.5.7 / 6.5.8_ <br>
 > [:file_folder:_Repo Mirror Available Here_:clipboard:](https://gt.7fproject.com/OVERLORD/SVMU)
 
 # Requirements

config_data_import.py

@@ -9,62 +9,62 @@ from rich.panel import Panel
 from rich.console import Console , Align
 from rich.prompt import Prompt
 console = Console()
-
 def config_menu(base_url, api_key, config_relative_path):
-        cls()
-        config_menu_options="[gold bold][1] [grey53 italic]Show current profile configuration\n[/]\
-\n[gold bold][2] [grey53 italic]Setup new profile[/]\n \
-\n[gold bold][3] [grey53 italic]Switch profile[/]\n\
-\n[gold bold][4] [grey53 italic]Delete profile[/]\n\
-\n[gold bold][5] [grey53 italic]Set default profile[/]\n\
-\n[gold bold][6] [grey53 italic]Change selected data pool[/]\n\
-\n[gold bold][7] [grey53 italic]Change selected VMs[/]\n\
-\n[gold bold][8] [grey53 italic]Change ISO UUID (auto-mount)[/]\n\
-\n[gold bold][9] [grey53 italic]Skip start-up splash[/]\n\
-\n\n[green_yellow bold]ENTER - return to Main Menu[/]"
-        config_menu_options=Align.center(config_menu_options, vertical="middle")
-        console = Console()
-        console.print(Panel(config_menu_options, title="[gold bold]SpaceVM Utility - Utility Configuration" , border_style="magenta" , width=150 , padding = 2))
-        sub_choice = console.input("[bold yellow]\n>>> [/]")
-        needs_reload = False
-        
-        if sub_choice == "1":
-            config_show(config_relative_path)
-            return config_menu(base_url, api_key, config_relative_path)
-        
-        if sub_choice == "2":
-            new_path = create_new_profile()
-            if new_path:
-                return new_path
-            
-        if sub_choice == "3":
-            new_profile_path = switch_profile()
-            if new_profile_path:  # If we got a new profile path
-                return new_profile_path  # Return it to main.py
-            
-        if sub_choice == "4":
-            delete_profile(config_relative_path)
+    cls()
+    menu = (
+        "[gold1 bold][1][/] [grey53 italic]Show current profile configuration[/grey53 italic]\n\n"
+        "[gold1 bold][2][/] [grey53 italic]Setup new profile[/]\n\n"
+        "[gold1 bold][3][/] [grey53 italic]Switch profile[/]\n\n"
+        "[gold1 bold][4][/] [grey53 italic]Delete profile[/]\n\n"
+        "[gold1 bold][5][/] [grey53 italic]Set default profile[/]\n\n"
+        "[gold1 bold][6][/] [grey53 italic]Change selected data pool[/]\n\n"
+        "[gold1 bold][7][/] [grey53 italic]Change selected VMs[/]\n\n"
+        "[gold1 bold][8][/] [grey53 italic]Change ISO UUID (auto-mount)[/]\n\n"
+        "[gold1 bold][9][/] [grey53 italic]Skip start-up splash[/]\n\n"
+        "[green_yellow bold]ENTER - return to Main Menu"
+    )
+    console.print(Panel(Align.center(menu, vertical="middle"), title="[gold bold]SpaceVM Utility - Utility Configuration", border_style="magenta", width=150, padding=2))
+    choice = console.input("[bold yellow]\n>>> [/]")
+    new_profile = None #used for controlling profile switch / updating current profile
+    needs_reload = False
 
-        if sub_choice == "5":
-            set_default_profile()
+    if choice == "1":
+        config_show(config_relative_path)
+        return new_profile, needs_reload
+    if choice == "2":
+        result = create_new_profile() # create_new_profile may return a profile path (to switch to)
+        if result:
+            new_profile = result
+        return new_profile, needs_reload
+    if choice == "3":
+        result = switch_profile() # switch_profile returns the selected profile path
+        if result:
+            new_profile = result
+        return new_profile, needs_reload
+    if choice == "4":
+        delete_profile(config_relative_path)
+        return new_profile, needs_reload
+    if choice == "5":
+        set_default_profile()
+        return new_profile, needs_reload
+    if choice == "6":
+        change_data_pool(base_url, api_key, config_relative_path)
+        needs_reload = True
+        return new_profile, needs_reload
+    if choice == "7":
+        change_vm_uuids(config_relative_path, base_url, api_key)
+        needs_reload = True
+        return new_profile, needs_reload
+    if choice == "8":
+        change_iso_uuid(config_relative_path)
+        needs_reload = True
+        return new_profile, needs_reload
+    if choice == "9":
+        change_startup_option(config_relative_path)
+        needs_reload = True
+        return new_profile, needs_reload
 
-        if sub_choice == "6":
-            change_data_pool(base_url, api_key, config_relative_path)
-            needs_reload = True
-
-        if sub_choice == "7":
-            change_vm_uuids(config_relative_path, base_url, api_key)
-            needs_reload = True
-
-        if sub_choice == "8":
-            change_iso_uuid(config_relative_path)
-            needs_reload = True
-
-        if sub_choice == "9":
-            change_startup_option(config_relative_path)
-            needs_reload = True            
-            
-        return needs_reload
+    return new_profile, needs_reload
 
 def config_show(config_relative_path):
     cls()
@@ -77,7 +77,6 @@ def config_show(config_relative_path):
 def config_import(config_relative_path):
     config = configparser.ConfigParser()
     config.read(config_relative_path)
-
     skip_startup_splash = config.get('General', 'skip_startup_splash')
     base_url = config.get('General', 'controller_ip')
     api_key = "jwt " + config.get('General', 'api_key') #That was realy obvious DACOM >:C
@@ -88,8 +87,7 @@ def config_import(config_relative_path):
         for key, value in config['VM_List'].items():
             vm_list.append(value)
     
-    #importing VM_Options
-    if config.has_section('VM_Options'):
+    if config.has_section('VM_Options'): #importing VM_Options
         iso_uuid = config.get('VM_Options' , 'iso_uuid')
         disk_interface = config.get('VM_Options' , 'disk_interface')    
         preallocation = config.get('VM_Options' , 'preallocation')
@@ -101,35 +99,22 @@ def config_import(config_relative_path):
         preallocation = "falloc"
         iso_name= "none"
         config = configparser.ConfigParser() #writing default values to config
-        config["VM_Options"] = {
-            "disk_interface": "virtio",
-            "preallocation": "falloc",
-            "iso_uuid": "none",
-        }        
+        config["VM_Options"] = {"disk_interface": "virtio", "preallocation": "falloc", "iso_uuid": "none"}        
         with open(config_relative_path, "a") as configfile: # appending to existing config file
             config.write(configfile) 
 
-    #importing disk sizes for SpaceVM courses 
-    if config.has_section('Courses-Space-VM'):
-        disk1_size = config.get('Courses-Space-VM', 'disk1')
-        disk2_size = config.get('Courses-Space-VM', 'disk2') 
-        disk3_size = config.get('Courses-Space-VM', 'disk3')
+    if config.has_section('Courses-Space-VM'): #importing disk sizes for SpaceVM courses
+        disk1_size, disk2_size, disk3_size = config.get('Courses-Space-VM', 'disk1'), config.get('Courses-Space-VM', 'disk2'), config.get('Courses-Space-VM', 'disk3')
     else:
         console.print("[bold yellow]Applying default values to Disk sizes for Courses")
         disk1_size, disk2_size, disk3_size = 10, 20, 20 #applying default values for courses
         config = configparser.ConfigParser() #writing default values to config
-        config["Courses-Space-VM"] = {
-            "disk1": 10,
-            "disk2": 20,
-            "disk3": 20,
-        }
+        config["Courses-Space-VM"] = {"disk1": 10, "disk2": 20,"disk3": 20}
         with open(config_relative_path, "a") as configfile: # appending to existing config file
             config.write(configfile)        
 
-    #get pretty name for selected data pool
-    data_pool_name = get_data_pool_name(base_url , api_key , data_pool_uuid)
-    #get pretty name for selected VMs
-    vm_names=[]
+    data_pool_name = get_data_pool_name(base_url , api_key , data_pool_uuid)  #get pretty name for selected data pool
+    vm_names=[]     #get pretty name for selected VMs
     for x in vm_list:
         vm_names.append(get_vm_name(base_url, api_key, x))
     
@@ -137,23 +122,16 @@ def config_import(config_relative_path):
 
 def change_startup_option(config_relative_path):
     cls()
-    # Ask the user once and normalize the answer
     new_value = Prompt.ask("[yellow bold]Skip start-up splash ?[/]", choices=["Y", "N"], default="N", case_sensitive=False)
     if new_value.lower() == "y":
         startup_option = "yes"
     else:
         startup_option = "no"
-
     config = configparser.ConfigParser()
     config.read(config_relative_path)
-    # Ensure General section exists so we can write the option
-    if not config.has_section('General'):
-        config.add_section('General')
-
     config.set('General', 'skip_startup_splash', startup_option)
     with open(config_relative_path, 'w') as config_file:
         config.write(config_file)
-
     console.print(f"[green bold]Option set to: {new_value}")
 
 def change_data_pool(base_url, api_key, config_relative_path): #change selected data pool in config
@@ -181,13 +159,12 @@ def change_iso_uuid(config_relative_path):
             config.write(config_file)
     else:
         print("No 'VM_Options' section in config file..")
-    config_show(config_relative_path)
+    #config_show(config_relative_path)
 
 def change_vm_uuids(config_relative_path, base_url, api_key): #change selected VM uuids in config
     config = configparser.ConfigParser()
     config.read(config_relative_path)
-    # Remove old VM_List section if it exists, then add a fresh one
-    if config.has_section('VM_List'):
+    if config.has_section('VM_List'): # Remove old VM_List section if it exists, then add a new one
         config.remove_section('VM_List')
     config.add_section('VM_List')
     cls()
@@ -197,8 +174,7 @@ def change_vm_uuids(config_relative_path, base_url, api_key): #change selected V
         vm_input = console.input("[bold yellow]>> [/]" )
         if not vm_input:
             break
-        # validate only the entered VM UUID via get_vm_name
-        vm_name = get_vm_name(base_url, "jwt " + api_key, vm_input)
+        vm_name = get_vm_name(base_url, "jwt " + api_key, vm_input) # validate entered VM UUID via get_vm_name
         if not vm_name:
             console.print("[red bold]Invalid VM UUID (not found)")
             continue
@@ -207,18 +183,13 @@ def change_vm_uuids(config_relative_path, base_url, api_key): #change selected V
 
     with open(config_relative_path, 'w') as configfile:
         config.write(configfile)
-
-    console.print("[green bold]VM UUIDs have been updated in config :pencil:")
+        console.print("[green bold]VM UUIDs have been updated in config :pencil:")
     Prompt.ask("[green_yellow bold]Press ENTER to proceed.. :right_arrow_curving_down:")
     config_show(config_relative_path)
 
-
-def check_config(config_relative_path):
-    """Check if config exists and is valid"""
-    # Only check if the file is empty and needs to be removed
-    if os.path.exists(config_relative_path) and os.path.getsize(config_relative_path) == 0:
+def check_config(config_relative_path): # Check if config exists and is valid
+    if os.path.exists(config_relative_path) and os.path.getsize(config_relative_path) == 0: # Only check if the file is empty
         console.print("[red bold]Config file is empty!")
-        os.remove(config_relative_path)  # Remove empty file
 
 def cls():
     os.system('cls' if os.name=='nt' else 'clear')
@@ -235,15 +206,13 @@ def check_ping(base_url):
     else:
         return False
 
-def create_profiles_dir():
-    """Create profiles directory if it doesn't exist"""
+def create_profiles_dir(): #creates ./profiles directory if it does not exist
     profiles_dir = os.path.join(os.getcwd(), 'profiles')
     if not os.path.exists(profiles_dir):
         os.makedirs(profiles_dir)
     return profiles_dir
 
-def get_available_profiles():
-    """Get list of available profile names"""
+def get_available_profiles(): #Get list of available profile names
     profiles_dir = create_profiles_dir()
     profiles = []
     for filename in os.listdir(profiles_dir):
@@ -251,14 +220,11 @@ def get_available_profiles():
             profiles.append(filename[:-5])  # Remove .conf extension
     return profiles
 
-def create_new_profile():
-    """Create a new profile configuration"""
+def create_new_profile(): # Create new profile. Returns profile_path
     cls()
     profiles_dir = create_profiles_dir()
-    
     profile_name = Prompt.ask("[yellow bold]Enter new profile name")
     profile_path = os.path.join(profiles_dir, f"{profile_name}.conf")
-    
     if os.path.exists(profile_path):
         console.print("[red bold]Profile already exists!")
         return
@@ -275,11 +241,7 @@ def create_new_profile():
     data_pool_uuid = console.input("[bold yellow]Type Data Pool UUID you wish to use: [/]")
 
     config = configparser.ConfigParser()
-    config["General"] = {
-        "controller_ip": base_url,
-        "api_key": api_key,
-        "skip_startup_splash": "no",
-    }
+    config["General"] = {"controller_ip": base_url, "api_key": api_key, "skip_startup_splash": "no"}
     config["Data_Pool"] = {"data_pool_uuid": data_pool_uuid}
 
     with open(profile_path, "w") as configfile:
@@ -293,8 +255,7 @@ def create_new_profile():
             vm_input = console.input("[bold yellow]>> [/]")
             if not vm_input:
                 break
-            # validate only the vm_input by fetching VM name
-            vm_name = get_vm_name(base_url, "jwt " + api_key, vm_input)
+            vm_name = get_vm_name(base_url, "jwt " + api_key, vm_input) # validate entered VM UUID via get_vm_name
             if not vm_name:
                 console.print("[red bold]Invalid VM UUID (not found)")
                 continue
@@ -302,13 +263,10 @@ def create_new_profile():
             file.write(f"uuid_{x} = {vm_input}\n")
 
     console.print("[green bold]Configuration completed! :white_check_mark:")
-    # Prompt to switch to the newly created profile now
     switch_now = Prompt.ask("[yellow bold]Switch to new profile now?", choices=["Y", "N"] , default="Y", case_sensitive=False)
     if switch_now.lower() == "y" or switch_now.lower() == "Y":
-        # Also ask if user wants to set this profile as default
         set_default = Prompt.ask("[yellow bold]Set new profile as default?", choices=["Y", "N"], default="N", case_sensitive=False)
         if set_default.lower() == "y" or set_default.lower() == "Y":
-            # Use existing set_default_profile logic: mark this profile as default
             profiles_dir = create_profiles_dir()
             profiles = get_available_profiles()
             for p in profiles:
@@ -320,8 +278,7 @@ def create_new_profile():
                     with open(p_path, 'w') as f:
                         cfg.write(f)
 
-            # set new profile as default
-            cfg = configparser.ConfigParser()
+            cfg = configparser.ConfigParser() # set new profile as default
             cfg.read(profile_path)
             if not cfg.has_section('General'):
                 cfg.add_section('General')
@@ -330,11 +287,9 @@ def create_new_profile():
                 cfg.write(f)
         return profile_path
 
-def switch_profile():
-    """Switch to a different profile."""
+def switch_profile(): # switch to different profile
     cls()
     profiles = get_available_profiles()
-
     if not profiles:
         console.print("[red bold]No profiles found!")
         return
@@ -342,26 +297,20 @@ def switch_profile():
     console.print("[yellow bold]Available profiles:")
     for i, profile in enumerate(profiles, 1):
         console.print(f"[grey53]{i}. {profile}")
-
     choice = Prompt.ask("[yellow bold]Select profile number: ", choices=[str(i) for i in range(1, len(profiles) + 1)])
     selected_profile = profiles[int(choice) - 1]
 
-    # Get the path for the selected profile
-    selected_profile_path = os.path.join(create_profiles_dir(), f"{selected_profile}.conf")
-
+    selected_profile_path = os.path.join(create_profiles_dir(), f"{selected_profile}.conf") # Get the path for the selected profile
     if not os.path.exists(selected_profile_path):
         console.print(f"[red bold]Profile '{selected_profile}' does not exist!")
         return
 
-    # Return the new profile path to update in main.py
-    console.print(f"[green bold]Switched to profile: {selected_profile}")
+    console.print(f"[green bold]Switched to profile: {selected_profile}") # Return the new profile path to update in main.py
     return selected_profile_path  # Return the new path instead of True
 
-def delete_profile(current_profile_path):
-    """Delete an existing profile"""
+def delete_profile(current_profile_path): #Delete existing profile
     cls()
     profiles = get_available_profiles()
-    
     if not profiles:
         console.print("[red bold]No profiles found!")
         return
@@ -372,11 +321,10 @@ def delete_profile(current_profile_path):
         
     choice = Prompt.ask("[yellow bold]Select profile to delete", choices=[str(i) for i in range(1, len(profiles)+1)])
     selected_profile = profiles[int(choice)-1]
-
     selected_profile_path = os.path.join(create_profiles_dir(), f"{selected_profile}.conf")
 
     if os.path.normpath(selected_profile_path) == os.path.normpath(current_profile_path):
-        console.print("[red bold]Cannot delete the currently active profile!")
+        console.print("[red bold]Cannot delete currently active profile!")
         Prompt.ask("[green_yellow bold]Press ENTER to return.. :right_arrow_curving_down:")
         return
     
@@ -385,11 +333,9 @@ def delete_profile(current_profile_path):
         os.remove(os.path.join(os.getcwd(), 'profiles', f"{selected_profile}.conf"))
         console.print(f"[green bold]Profile {selected_profile} deleted!")
 
-def set_default_profile():
-    """Set the selected profile as the default profile."""
+def set_default_profile(): # Set selected profile as default
     cls()
     profiles = get_available_profiles()
-
     if not profiles:
         console.print("[red bold]No profiles found!")
         return
@@ -400,9 +346,8 @@ def set_default_profile():
 
     choice = Prompt.ask("[yellow bold]Select profile number to set as default", choices=[str(i) for i in range(1, len(profiles) + 1)])
     selected_profile = profiles[int(choice) - 1]
-
-    # Set all other profiles' default flag to false
-    profiles_dir = create_profiles_dir()
+   
+    profiles_dir = create_profiles_dir() #Fuck you there could be only ONE default ! (all other profiles deafult flags set to false)
     for profile in profiles:
         profile_path = os.path.join(profiles_dir, f"{profile}.conf")
         config = configparser.ConfigParser()
@@ -411,9 +356,8 @@ def set_default_profile():
             config.set('General', 'load_by_default', 'false')
             with open(profile_path, 'w') as f:
                 config.write(f)
-
-    # Set the selected profile as default
-    selected_profile_path = os.path.join(profiles_dir, f"{selected_profile}.conf")
+ 
+    selected_profile_path = os.path.join(profiles_dir, f"{selected_profile}.conf") # Set the selected profile as default
     config = configparser.ConfigParser()
     config.read(selected_profile_path)
     if not config.has_section('General'):
@@ -424,8 +368,7 @@ def set_default_profile():
 
     console.print(f"[green bold]Profile '{selected_profile}' set as default!")
 
-def get_default_config_path():
-    """Retrieve the path of the default profile configuration or prompt the user to select one."""
+def get_default_config_path(): # Retrieve the path of the default profile. If none, promt user to select one
     profiles_dir = create_profiles_dir()
     profiles = get_available_profiles()
 
@@ -437,25 +380,20 @@ def get_default_config_path():
             if config.getboolean('General', 'load_by_default'):
                 return profile_path
 
-    # If no default profile is found, prompt the user to select one
-    if profiles:
+    if profiles: # no default profile is found, prompt to select one
         console.print("[yellow bold]No default profile found. Please select a profile to load:")
         for i, profile in enumerate(profiles, 1):
             console.print(f"[grey53]{i}. {profile}")
         choice = Prompt.ask("[yellow bold]Select profile number", choices=[str(i) for i in range(1, len(profiles) + 1)])
         selected_profile = profiles[int(choice) - 1]
         return os.path.join(profiles_dir, f"{selected_profile}.conf")
-
     console.print("[red bold]No profiles available. Please create a new profile.")
     create_new_profile()
 
-    # Refresh the profiles list and directly return the newly created profile
-    profiles = get_available_profiles()
+    profiles = get_available_profiles() # Refresh the profiles list and return new one
     if profiles:
         new_profile_path = os.path.join(profiles_dir, f"{profiles[-1]}.conf")
         console.print(f"[green bold]Loaded newly created profile: {profiles[-1]}")
         return new_profile_path
 
-    raise RuntimeError("Failed to create or load a profile.")
-
-# config_relative_path will be set when passed from main.py
+    raise RuntimeError("Failed to create or load a profile.")

disk_edit_mode.py

@@ -1,4 +1,4 @@
-import os, sys
+import os, sys, time
 import requests
 from domain_api import *
 from rich.prompt import Prompt
@@ -7,10 +7,10 @@ from rich.panel import Panel
 
 def 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')
-        diks_edit_menu_options="[gold bold][1] [grey53 italic]Delete vDisk by UUID\n[/grey53 italic] \
-\n[gold bold][2] [grey53 italic]Delete ALL vDisks on selected Virtual Machine[/grey53 italic]\n \
-\n[gold bold][3] [grey53 italic]Create Disk[/grey53 italic]\n \
-\n[gold bold][4] [grey53 italic]Prepare VMs for Courses™[/grey53 italic]\n \
+        diks_edit_menu_options="[gold1 bold][1] [grey53 italic]Delete vDisk by UUID\n[/grey53 italic] \
+\n[gold1 bold][2] [grey53 italic]Delete ALL vDisks on selected Virtual Machine[/grey53 italic]\n \
+\n[gold1 bold][3] [grey53 italic]Create Disk[/grey53 italic]\n \
+\n[gold1 bold][4] [grey53 italic]Prepare VMs for Courses™[/grey53 italic]\n \
 \n\n[green_yellow bold]ENTER - return to Main Menu"
         diks_edit_menu_options = Align.center(diks_edit_menu_options, vertical="middle")
         console = Console()
@@ -44,6 +44,7 @@ def disk_edit_mode(base_url , api_key , data_pool_uuid , vm_uuids, disk1_size, d
             for y in vm_uuids: #power-on check
                 domain_uuid = y.strip('\n')
                 vm_check_power(base_url , api_key , domain_uuid)
+            start_time = time.perf_counter() #after power-on check passed, starting timer
             for x in vm_uuids: # only for removing disks
                 domain_uuid = x.strip('\n')
                 domain_info = get_domain_info(base_url , api_key , domain_uuid)
@@ -70,7 +71,7 @@ def disk_edit_mode(base_url , api_key , data_pool_uuid , vm_uuids, disk1_size, d
                         console.print("[grey53 italic]iso_uuid was not specified. Skipping ISO auto-mount..[/]")
                     else:
                         attach_iso(base_url, api_key, domain_uuid, iso_uuid)
-                        
-            console.print("[bold green]\nDone. Happy virtualization :thumbs_up::thumbs_up:")
+            end_time = time.perf_counter() # timer end           
+            console.print(f"[bold gold1]\nOperation completed. Elapsed time {end_time - start_time:.1f} seconds \n[bold green]Happy virtualization :thumbs_up::thumbs_up:")
             Prompt.ask("[green_yellow bold]ENTER - return to Main Menu.. :right_arrow_curving_down:")
         os.system('cls' if os.name=='nt' else 'clear')

domain_api.py

@@ -317,9 +317,9 @@ def attach_iso(base_url, api_key, vm_id, iso_uuid):
 
 def vm_menu(base_url, api_key, vm_uuids, config_relative_path):
         os.system('cls' if os.name=='nt' else 'clear') 
-        config_menu_options="[gold bold][1] [grey53 italic]Show VM info \n    (for selected VMs in config)[/grey53 italic]\n \
-\n[gold bold][2] [grey53 italic]Show VMs Name / UUID[/grey53 italic]\n \
-\n[gold bold][3] [grey53 italic]Select VMs by tag / UUID[/grey53 italic]\n \
+        config_menu_options="[gold1 bold][1] [grey53 italic]Show VM info \n    (for selected VMs in config)[/grey53 italic]\n \
+\n[gold1 bold][2] [grey53 italic]Show VMs Name / UUID[/grey53 italic]\n \
+\n[gold1 bold][3] [grey53 italic]Select VMs by tag / UUID[/grey53 italic]\n \
 \n\n[green_yellow bold]ENTER - return to Main Menu[/]"
         config_menu_options=Align.center(config_menu_options, vertical="middle")
         console = Console()

main.py

@@ -7,29 +7,24 @@ from data_pools_api import *
 from disk_edit_mode import *
 from rich.panel import Panel
 from rich.console import Console , Align
-SVMU_ver="0.4-dev"
+SVMU_ver="v0.4.4" # pls dont forget to change before uploading release (lol)
 
-# Initialize console and clear screen
-console = Console()
+console = Console() # Initialize console and clear screen
 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)
+config_relative_path = get_default_config_path() # Get config path + necessery checks
+skip_startup_splash = get_skip_startup_splash(config_relative_path) #check profile for skip startup. Profile is not imported at this point (!)
+show_startup_logo(skip_startup_splash, SVMU_ver) #shows / skips startup splash (ASCII art)
 
 menu_choice=0
 while(menu_choice != ""):    #main menu loop
     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
     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 \
+    menu_options=f"[gold1 bold][1] [grey53 italic]Utiliy Configuration / Profiles\n[/grey53 italic] \
+\n[gold1 bold][2] [grey53 italic]Enter disk edit mode[/grey53 italic]\n \
+\n[gold1 bold][3] [grey53 italic]Show breif cluster overview[/grey53 italic]\n \
+\n[gold1 bold][4] [grey53 italic]Enter VM menu[/grey53 italic]\n \
+\n[gold1 bold][5] [grey53 italic]Show data pools[/grey53 italic]\n \
 \n\n[green_yellow bold]ENTER - exit Utility[/]\n\n \
 [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}[/]"
@@ -39,16 +34,14 @@ while(menu_choice != ""):    #main menu loop
     menu_choice = console.input("[bold yellow]\n>>> [/]")
     
     if menu_choice == "1":
-        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
+        new_profile, needs_reload = config_menu(base_url, api_key, config_relative_path)
+        # If a new profile path was returned, switch to it
+        if new_profile:
+            config_relative_path = new_profile
             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
+        # If menu indicated that config changes require a reload, go back to main menu
+        if needs_reload:
             os.system('cls' if os.name=='nt' else 'clear')
             continue
     if menu_choice == "2":

splash_screen.py

@@ -22,9 +22,10 @@ def get_skip_startup_splash(config_path):
 def show_startup_logo(skip_startup_splash, SVMU_ver):
     if skip_startup_splash == "yes":
         return
-    splash_file = os.path.join(os.path.dirname(__file__), "splash-screens.txt")
+    splash_dir = os.path.dirname(__file__) # Assume splash file lives in the assets/ directory
+    splash_file = os.path.join(splash_dir, "assets", "splash-screens.txt")
     if not os.path.exists(splash_file):
-        console.print("[bold red]Splash screens file not found![/bold red]")
+        console.print(f"[bold red]No splash file fouond at: {splash_file}[/bold red]")
         return
 
     with open(splash_file, "r", encoding="utf-8") as f: