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

View File

@@ -12,90 +12,61 @@ console = Console()
def config_menu(base_url, api_key, config_relative_path):
cls()
config_menu_options="[gold bold][0] [grey53 italic]Show example configuration\n[/]\
\n[gold bold][1] [grey53 italic]Show current configuration\n[/] \
\n[gold bold][2] [grey53 italic]Setup new config file[/]\n \
\n[gold bold][3] [grey53 italic]Change selected data pool[/]\n\
\n[gold bold][4] [grey53 italic]Change selected VMs[/]\n\
\n[gold bold][5] [grey53 italic]Change ISO UUID (auto-mount)[/]\n\
\n[gold bold][6] [grey53 italic]Skip start-up splash[/]\n\
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=str(input("\n>>> "))
if sub_choice == "0":
config_show_example()
sub_choice = console.input("[bold yellow]\n>>> [/]")
needs_reload = False
if sub_choice == "1":
config_show(config_relative_path)
config_menu(base_url, api_key, config_relative_path)
return config_menu(base_url, api_key, config_relative_path)
if sub_choice == "2":
config_edit(config_relative_path)
new_path = create_new_profile()
if new_path:
return new_path
if sub_choice == "3":
change_data_pool(base_url, api_key, config_relative_path)
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":
change_vm_uuids(config_relative_path)
delete_profile(config_relative_path)
if sub_choice == "5":
change_iso_uuid(config_relative_path)
set_default_profile()
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)
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)
def config_show_example():
conf_example= """
[General]
skip_startup_splash = no
#Master Controller IP of your cluster
#Has to be accessible for a machine, which will be executing this Utility
controller_ip = 10.20.30.44
#Integration API Key. how to get your key:
# ( https://spacevm.ru/docs/latest/base/operator_guide/security/users/#_14 )
# do not specify JWT tag with your key!
api_key =
[Data_Pool]
#Data pool which will be used for utility operations
#(Targeted storage for new vDisks)
data_pool_uuid =
[VM_List]
#Selected VMs which will be used for utility operations
#How to find UUID:
#List all available VMs in Utility Main Menu (Option 6)
#Use https://spacevm.ru/docs/latest/cli/space/vm/info/ or copy UUID from web panel
uuid_1 =
uuid_2 =
[VM_Options]
#Select interface which will be used in virtual disk creation.
#Available options: virtio / ide / scsi / sata
disk_interface = virtio
#Select allocation type for virtual disks
#Available options: none / falloc / full / metadata
preallocation = falloc
#Specify uuid of iso you wish to automatically mount to Virtual Machines during operations (Courses)
#This step is skipped if "none" provided
iso_uuid = none
[Courses-Space-VM]
#Set vDisk size for "Prepare VMs for Courses™" option
disk1 = 10
disk2 = 20
disk3 = 20
"""
cls()
console.rule(title = "Example config file", align="center", style="yellow")
console.print(conf_example)
console.rule(style="yellow")
Prompt.ask("[green_yellow bold]ENTER - return to Utility Configuration.. :right_arrow_curving_down:")
needs_reload = True
return needs_reload
def config_show(config_relative_path):
cls()
console.rule(title = "Current configuration" , align="center" , style="yellow")
with open(config_relative_path, "r") as f:
@@ -112,7 +83,6 @@ def config_import(config_relative_path):
api_key = "jwt " + config.get('General', 'api_key') #That was realy obvious DACOM >:C
data_pool_uuid = config.get('Data_Pool', 'data_pool_uuid')
vm_list = []
if 'VM_List' in config:
for key, value in config['VM_List'].items():
@@ -168,11 +138,15 @@ def config_import(config_relative_path):
def change_startup_option(config_relative_path):
cls()
#console.print("[yellow bold]Skip start-up splash ?")
new_value = Prompt.ask("[yellow bold]Skip start-up splash ?[/]", choices=["Y", "N"], default="N")
new_value = Prompt.ask("[yellow bold]Skip start-up splash ?[/]", choices=["Y", "N"], default="N", case_sensitive=False)
if new_value == "Y" or new_value == "y":
startup_option = "yes"
if new_value == "N" or new_value == "n":
startup_option = "no"
config = configparser.ConfigParser()
config.read(config_relative_path)
if config.has_section('General'):
config.set('General', 'skip_startup_splash', new_value)
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}")
@@ -182,7 +156,7 @@ def change_startup_option(config_relative_path):
def change_data_pool(base_url, api_key, config_relative_path): #change selected data pool in config
cls()
show_data_pools(base_url, api_key)
new_data_pool_uuid = input("Type NEW Data Pool UUID: ")
new_data_pool_uuid = console.input("[bold yellow]Type NEW Data Pool UUID: [/]")
config = configparser.ConfigParser()
config.read(config_relative_path)
if config.has_section('Data_Pool'):
@@ -195,7 +169,7 @@ def change_data_pool(base_url, api_key, config_relative_path): #change selected
def change_iso_uuid(config_relative_path):
cls()
new_iso_uuid = input("Type ISO UUID: ")
new_iso_uuid = console.input("[bold yellow]Type ISO UUID: [/]")
config = configparser.ConfigParser()
config.read(config_relative_path)
if config.has_section('VM_Options'):
@@ -214,10 +188,10 @@ def change_vm_uuids(config_relative_path): #change selected VM uuids in config
config.remove_section('VM_List')
config.add_section('VM_List')
cls()
console.print("[yellow bold]Type new VM UUIDs one by one (input ENTER to stop):")
console.print("[yellow bold]Type new VM UUIDs one by one [red bold](ENTER to stop)[/] ")
x = 0
while True:
vm_input = input(">> ")
vm_input = console.input("[bold yellow]>> [/]" )
if not vm_input:
break
x += 1
@@ -231,62 +205,12 @@ def change_vm_uuids(config_relative_path): #change selected VM uuids in config
config_show(config_relative_path)
def config_edit(config_relative_path):
read_input = Prompt.ask("[bold yellow]Create new config file?[/]", choices=["Y", "N"], default="N")
menu_choice = str(read_input)
if menu_choice == "Y" or menu_choice == "y":
base_url = input("Type SpaceVM Controller IP: ")
while check_ping(base_url) != True:
base_url = console.input("[bold red]No response.\nCheck and type SpaceVM Controller IP again: [/]")
api_key = input("Type your API Key: ")
while check_api_key(base_url, "jwt " + api_key) != 200:
api_key = console.input("[bold red]Check and type SpaceVM Controller API Key again: [/]")
show_data_pools(base_url, "jwt " + api_key)
data_pool_uuid = input("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["Data_Pool"] = {"data_pool_uuid": data_pool_uuid}
#disk_interface=input("Specify preffered disk interface (virtio / ide / scsi / sata): ")
#preallocation=input("Specify allocation type for virtual disks (none / falloc / full / metadata): ")
#iso_uuid=input("Specify ISO uuid you wish to auto-mount during operations(none - skip this step): ")
#config["VM_Options"] = {
# "disk_interface": disk_interface,
# "preallocation": preallocation,
# "iso_uuid": iso_uuid
#}
with open(config_relative_path, "w") as configfile: #writing everything from above to config file
config.write(configfile)
print("Type VM UUIDs one by one (input ENTER to stop)")
with open(config_relative_path, "a") as file:
file.write("[VM_List]\n") #manually writing section for VMs
vm_input = []
x = 0
while vm_input != "":
vm_input = input(">> ")
if vm_input:
x += 1
file.write(f"uuid_{x} = {vm_input}\n")
console.print("[green bold]VM UUIDs have been written in config :pencil:")
console.print("[green bold]Configuration completed ! :white_check_mark:")
Prompt.ask("[green_yellow bold]Press ENTER to proceed.. :right_arrow_curving_down:")
cls()
def check_config(config_relative_path):
if os.path.exists(config_relative_path) and os.path.getsize(config_relative_path) > 0: #check if config exists and not empty
pass #do nothing
else:
console.print("[yellow bold]Config file was not found or empty.. ")
config_edit(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:
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')
@@ -301,4 +225,224 @@ def check_ping(base_url):
if status == 0:
return True
else:
return False
return False
def create_profiles_dir():
"""Create profiles directory if it doesn't 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"""
profiles_dir = create_profiles_dir()
profiles = []
for filename in os.listdir(profiles_dir):
if filename.endswith('.conf'):
profiles.append(filename[:-5]) # Remove .conf extension
return profiles
def create_new_profile():
"""Create a new profile configuration"""
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
base_url = console.input("[bold yellow]Type SpaceVM Controller IP: [/]")
while check_ping(base_url) != True:
base_url = console.input("[bold red]No response.\nCheck and type SpaceVM Controller IP again: [/]")
api_key = console.input("[bold yellow]Type your API Key: [/]")
while check_api_key(base_url, "jwt " + api_key) != 200:
api_key = console.input("[bold red]Check and type SpaceVM Controller API Key again: [/]")
show_data_pools(base_url, "jwt " + api_key)
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["Data_Pool"] = {"data_pool_uuid": data_pool_uuid}
with open(profile_path, "w") as configfile:
config.write(configfile)
console.print("[yellow bold]Type VM UUIDs one by one (input ENTER to stop):")
with open(profile_path, "a") as file:
file.write("[VM_List]\n")
x = 0
while True:
vm_input = console.input("[bold yellow]>> [/]")
if not vm_input:
break
x += 1
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:
p_path = os.path.join(profiles_dir, f"{p}.conf")
cfg = configparser.ConfigParser()
cfg.read(p_path)
if cfg.has_section('General'):
cfg.set('General', 'load_by_default', 'false')
with open(p_path, 'w') as f:
cfg.write(f)
# set new profile as default
cfg = configparser.ConfigParser()
cfg.read(profile_path)
if not cfg.has_section('General'):
cfg.add_section('General')
cfg.set('General', 'load_by_default', 'true')
with open(profile_path, 'w') as f:
cfg.write(f)
return profile_path
def switch_profile():
"""Switch to a different profile."""
cls()
profiles = get_available_profiles()
if not profiles:
console.print("[red bold]No profiles found!")
return
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")
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}")
return selected_profile_path # Return the new path instead of True
def delete_profile(current_profile_path):
"""Delete an existing profile"""
cls()
profiles = get_available_profiles()
if not profiles:
console.print("[red bold]No profiles found!")
return
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 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!")
Prompt.ask("[green_yellow bold]Press ENTER to return.. :right_arrow_curving_down:")
return
confirm = Prompt.ask(f"[red bold]Are you sure you want to delete {selected_profile}?", choices=["Y", "N"] , default="Y" , case_sensitive=False)
if confirm.upper() == "Y":
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."""
cls()
profiles = get_available_profiles()
if not profiles:
console.print("[red bold]No profiles found!")
return
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 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()
for profile in profiles:
profile_path = os.path.join(profiles_dir, f"{profile}.conf")
config = configparser.ConfigParser()
config.read(profile_path)
if config.has_section('General'):
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")
config = configparser.ConfigParser()
config.read(selected_profile_path)
if not config.has_section('General'):
config.add_section('General')
config.set('General', 'load_by_default', 'true')
with open(selected_profile_path, 'w') as f:
config.write(f)
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."""
profiles_dir = create_profiles_dir()
profiles = get_available_profiles()
for profile in profiles:
profile_path = os.path.join(profiles_dir, f"{profile}.conf")
config = configparser.ConfigParser()
config.read(profile_path)
if config.has_section('General') and config.has_option('General', 'load_by_default'):
if config.getboolean('General', 'load_by_default'):
return profile_path
# If no default profile is found, prompt the user to select one
if profiles:
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()
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

View File

@@ -1,8 +1,9 @@
import os
import os, sys
import requests
from domain_api import *
from rich.prompt import Prompt
from rich.console import Console , Align
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')
@@ -14,15 +15,15 @@ def disk_edit_mode(base_url , api_key , data_pool_uuid , vm_uuids, disk1_size, d
diks_edit_menu_options = Align.center(diks_edit_menu_options, vertical="middle")
console = Console()
console.print(Panel(diks_edit_menu_options, title="[bold red]Disk Edit Mode" , border_style="magenta" , width=150 , padding = 2))
sub_choice=str(input("\n>>> "))
sub_choice = console.input("[bold yellow]\n>>> [/]")
if sub_choice == "1":
read_input=input("Input vDisk uuid to delete: ")
read_input = console.input("[bold yellow]Input vDisk uuid to delete: [/]" )
vdisk_uuid=str(read_input)
delete_disk(base_url , api_key , vdisk_uuid)
if sub_choice == "2":
print(vm_uuids)
select_uuids=int(input("Select VM to delete disks from. \n Type VM uuid index number (from list above) to select: ")) - 1
select_uuids = int(console.input("[bold yellow]Select VM to delete disks from. \n Type VM uuid index number (from list above) to select: [/]")) - 1
vm_check_power(base_url , api_key , vm_uuids[select_uuids]) #power on check
domain_all_content = get_domain_all_content(base_url , api_key , vm_uuids[select_uuids])
disk_uuids = get_disk_uuids(base_url , api_key , domain_all_content)
@@ -31,9 +32,9 @@ def disk_edit_mode(base_url , api_key , data_pool_uuid , vm_uuids, disk1_size, d
console.print("[bold red]All attached vDisks has been deleted!")
if sub_choice == "3":
vdisk_size=str(input("Enter disk size (GB): "))
vdisk_size = str(console.input("[bold yellow]Enter disk size (GB): [/]"))
print(vm_uuids)
select_uuids=int(input("Select VM to attach new disk. \n Type VM uuid index number (from list above) to select: ")) - 1
select_uuids = int(console.input("[bold yellow]Select VM to attach new disk. \n Type VM uuid index number (from list above) to select: [/]")) - 1
print(f"{vm_uuids[select_uuids]} - {data_pool_uuid} - {vdisk_size} ")
create_and_attach_disk(base_url , api_key , vm_uuids[select_uuids] , data_pool_uuid , vdisk_size , disk_interface, preallocation)
@@ -51,7 +52,9 @@ def disk_edit_mode(base_url , api_key , data_pool_uuid , vm_uuids, disk1_size, d
if domain_info:
disk_uuids = get_disk_uuids(base_url , api_key , domain_all_content)
for y in disk_uuids:
delete_disk(base_url , api_key , y)
if not delete_disk(base_url , api_key , y): #if delete_disk returns False - aborting
console.print("[bold red] Aborting further operations.")
sys.exit(1)
for z in vm_uuids: # only for creating disks
domain_uuid = z.strip('\n')
vm_name = get_vm_name(base_url, api_key, domain_uuid)

View File

@@ -1,9 +1,9 @@
# functions for working with domain-api
import requests
import requests, json
import secrets #for generating unique names
import os
import configparser
from config_data_import import *
#from config_data_import import * hopefully removes circular import and warnings with get_iso_name / get_vm_name
from rich.console import Console , Align
from rich.columns import Columns
from rich.panel import Panel
@@ -76,7 +76,13 @@ def delete_disk(base_url , api_key , vdisk_uuid):
console.print(f"[grey53 italic]{vdisk_uuid}[/] :wastebasket:")
return True
else:
print(f"ERROR deleting disk {vdisk_uuid} :\n {response.status_code} - {response.text}")
#print(f"ERROR deleting disk {vdisk_uuid} :\n {response.status_code} - {response.text}")
error_data = json.loads(response.text)
#decoding unicode response-string
error_detail = error_data.get("errors", [{}])[0].get("detail", "")
decoded_error = error_detail.encode('utf-8').decode('utf-8')
console.print(f"[bold red]ERROR {response.status_code} deleting disk {vdisk_uuid} :cross_mark:")
console.print(f"[bold red]{decoded_error}")
return False
@@ -135,12 +141,39 @@ def get_vm_name(base_url, api_key, vm_uuids):
return (f"{vm_name['verbose_name']}")
def get_vm_node_name(domain_all_content):
"""Return the node verbose_name from domain_all_content if present, else None.
Expected domain_all_content format:
"node": {"id": "<uuid>", "verbose_name": "string"}
"""
try:
if isinstance(domain_all_content, dict):
node = domain_all_content.get('node')
if isinstance(node, dict):
return node.get('verbose_name')
except Exception:
# Defensive: any unexpected structure returns None
pass
return None
def vm_info(base_url, api_key, vm_uuids):
domain_info = get_domain_info(base_url, api_key, vm_uuids)
domain_all_content = get_domain_all_content(base_url, api_key, vm_uuids)
if domain_info:
console = Console()
vm_info_lines = f"[bold]Power State:[/] [bold red]{power_state[domain_info['user_power_state']]}[/bold red] \n[bold]vDisks:[/] {domain_info['vdisks_count']}"
# Get the node verbose name from domain_all_content (if present)
vm_node = get_vm_node_name(domain_all_content)
# fallback display value
vm_node = vm_node if vm_node else "Unknown"
vm_info_lines = (
f"[bold]Power State:[/] [bold red]{power_state[domain_info['user_power_state']]}[/bold red]"
f" \n[bold]Node:[/] [bold bright_cyan]{vm_node}[/]"
f" \n[bold]vDisks:[/] {domain_info.get('vdisks_count', 'N/A')}"
)
vm_info_renderable = Panel(vm_info_lines, title=f"[bold magenta]{domain_info['verbose_name']}" , expand=False , border_style="yellow")
vm_info_renderable=Align.center(vm_info_renderable, vertical="middle")
print("\n")
@@ -220,7 +253,7 @@ def select_vm_by_tags(base_url, api_key, config_relative_path):
url = f"http://{base_url}/api/domains/"
response = requests.get(url, headers={'Authorization': api_key})
if response.status_code == 200:
verbose_name_input = input("Specify tag: ")
verbose_name_input = console.input("[bold yellow]Specify tag: [/]")
output_renderables = []
vm_info_short = response.json()
y= vm_info_short
@@ -238,8 +271,8 @@ def select_vm_by_tags(base_url, api_key, config_relative_path):
console.rule(style="grey53")
if vm_id_list: # promt to write found VM UUIDs to config
write_to_config = Confirm.ask("[bold yellow]Write these VM UUIDs to config file?")
if write_to_config:
write_to_config = Prompt.ask("[bold yellow]Write these VM UUIDs to config file?" , choices=["Y" , "N"], default="Y", case_sensitive=False)
if write_to_config == "y" or write_to_config == "Y":
config = configparser.ConfigParser()
config.read(config_relative_path)
# Remove old VM_List section if it exists, then add a fresh one
@@ -291,7 +324,7 @@ def vm_menu(base_url, api_key, vm_uuids, config_relative_path):
config_menu_options=Align.center(config_menu_options, vertical="middle")
console = Console()
console.print(Panel(config_menu_options, title="[gold bold]Show VM info" , border_style="magenta" , width=150 , padding = 2))
sub_choice=str(input("\n>>> "))
sub_choice = console.input("[bold yellow]\n>>> [/]")
if sub_choice == "1":
os.system('cls' if os.name=='nt' else 'clear')
for x in vm_uuids:

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 ")