OVERLORD/SVMU
1
1
Fork 0
mirror of https://github.com/OVERLORD7F/SVMU.git synced 2025-10-01 21:52:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

Expanding config-related functions

Browse Source 
- New function change_vm_uuids
- New function change_data_pool
This commit is contained in:
OVERLORD7F
2025-05-28 16:59:38 +03:00
parent 3ca5404b81
commit 71a7c38c27
2 changed files with 60 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
config_data_import.py
View File

@@ -10,20 +10,26 @@ from rich.console import Console , Align
from rich.prompt import Prompt
console = Console()
def config_menu(config_relative_path):
def config_menu(base_url, api_key, config_relative_path):
cls()
config_menu_options="[gold bold][1] [grey53 italic]Show current configuration\n[/grey53 italic] \
\n[gold bold][2] [grey53 italic]Change configuraion[/grey53 italic]\n \
\n\n[green_yellow bold]ENTER - return to Main Menu"
config_menu_options="[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[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 == "1":
config_show(config_relative_path)
config_menu(config_relative_path)
config_menu(base_url, api_key, config_relative_path)
if sub_choice == "2":
config_edit(config_relative_path)
if sub_choice == "3":
change_data_pool(base_url, api_key, config_relative_path)
if sub_choice == "4":
change_vm_uuids(config_relative_path)
def config_show(config_relative_path):
cls()
@@ -47,14 +53,53 @@ def config_import(config_relative_path):
vm_list.append(value)
#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=[]
for x in vm_list:
vm_names.append(get_vm_name(base_url, api_key, x))
return base_url, api_key, data_pool_uuid, data_pool_name, vm_list, vm_names
def change_data_pool(base_url, api_key, config_relative_path):
cls()
show_data_pools(base_url, api_key)
new_data_pool_uuid = input("Type NEW Data Pool UUID: ")
config = configparser.ConfigParser()
config.read(config_relative_path)
if config.has_section('Data_Pool'):
# update data_pool_uuid
config.set('Data_Pool', 'data_pool_uuid', new_data_pool_uuid)
with open(config_relative_path, 'w') as config_file:
config.write(config_file)
else:
print("No 'Data_Pool' section in config file..")
config_show(config_relative_path)
def change_vm_uuids(config_relative_path):
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'):
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):")
x = 0
while True:
vm_input = input(">> ")
if not vm_input:
break
x += 1
config.set('VM_List', f'uuid_{x}', vm_input)
with open(config_relative_path, 'w') as configfile:
config.write(configfile)
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 config_edit(config_relative_path):
read_input = input("Create new config file? (Y / N): ")
menu_choice = str(read_input)
@@ -67,8 +112,8 @@ def config_edit(config_relative_path):
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,
@@ -88,7 +133,7 @@ def config_edit(config_relative_path):
vm_input = input(">> ")
if vm_input:
x += 1
file.write(f"UUID_{x} = {vm_input}\n")
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:")