From 54328f4e4af27afd02f8509084b74fba521ba40f Mon Sep 17 00:00:00 2001 From: OVERLORD7F Date: Thu, 15 May 2025 15:41:33 +0300 Subject: [PATCH] LEETS GOOOOOOOO main is less then 100 lines !!!! Code cleanup, all functions moved from main. Small changes in main menu vm_info_short moved to domain_api --- config_data_import.py | 27 +++++++++- domain_api.py | 20 ++++++++ main.py | 115 +++++++++++------------------------------- vm_info_short.py | 20 -------- 4 files changed, 75 insertions(+), 107 deletions(-) diff --git a/config_data_import.py b/config_data_import.py index 6379d55..55cdfcd 100644 --- a/config_data_import.py +++ b/config_data_import.py @@ -11,10 +11,33 @@ def import_vm_uuid(config_relative_path): return vm_uuids def import_threelines(config_relative_path): - threelines = [] + threelines = [None] * 3 with open(config_relative_path, "r") as f: - all_lines = f.readlines() + all_lines = f.readlines() + if len(all_lines) < 3: + raise ValueError("Check config. Receiving less than 3 lines!") threelines[0] = all_lines[0].strip('\n') threelines[1] = "jwt " + all_lines[1].strip('\n') #actual format for api_key. That was realy obvious DACOM >:C threelines[2] = all_lines[2].strip('\n') return threelines + +def config_edit(config_relative_path): + read_input=input("Create new config file? (Y / N): ") + menu_choice=str(read_input) + if menu_choice == "Y" or menu_choice == "y": + base_url = input("Type SpaceVM Controller IP: ") + api_key = input("Type your API Key: ") + data_pool_uuid = input("Type Data pool uuid you wish to use: ") + lines = [base_url, api_key, data_pool_uuid] + with open(config_relative_path, "w+") as file: + for line in lines: + file.write(line + '\n') + + print("Type VM-UUID (input ENTER to stop)") + with open(config_relative_path, "a") as file: #appends new content at the end without modifying the existing data + vm_input="test" + while (vm_input != ""): + vm_input = input(">> ") + file.write(vm_input + '\n') + print("UUIDs has been written in config") + print("\nConfiguration completed!") \ No newline at end of file diff --git a/domain_api.py b/domain_api.py index a72039f..8c5369e 100644 --- a/domain_api.py +++ b/domain_api.py @@ -111,6 +111,26 @@ def vm_info(base_url , api_key , vm_uuids): get_disk_info(domain_all_content) +import requests + +def vm_info_short(base_url , api_key): #output data pool info + url= f"http://{base_url}//api/domains/" + response = requests.get(url , headers={'Authorization' : api_key}) + + if response.status_code == 200: + vm_info_short = response.json() + results_vm_info_short = vm_info_short['results'] + + print(f"\nShort VM overview | Total: {vm_info_short['count']}") + print("=" * 43) + for x in results_vm_info_short: + print(f" VM: {x['verbose_name']}") + print(f"UUID: {x['id']}") + print("-" * 43) + + else: + print(f"Failed to retrieve data {response.status_code}") + def create_and_attach_disk(base_url , api_key , vm_id, data_pool_uuid, vdisk_size, preallocation): domain_name=get_domain_info(base_url , api_key , vm_id) disk_name=domain_name["verbose_name"]+"_"+secrets.token_hex(5) #generates unique hex id. this method can generate ~million unique ids diff --git a/main.py b/main.py index 5ff5888..11af9c1 100644 --- a/main.py +++ b/main.py @@ -1,95 +1,56 @@ -import sys import os - from config_data_import import * - from cluster_api import * from domain_api import * from data_pools_api import * -from vm_info_short import * - -power_state = ["Unknown" , "Off" , "Suspend" , "On"] #3 - on; 2 - suspend; 1 - off; 0 - unknown #config.txt in the same directory with main.py -base_dir = os.getcwd() # Use the current directory as fallback -config_relative_path = os.path.join(base_dir, 'config.txt') - - -def config_edit(): - read_input=input("Create new config file? (Y / N): ") - menu_choice=str(read_input) - if menu_choice == "Y" or menu_choice == "y": - base_url = input("Type SpaceVM Controller IP: ") - api_key = input("Type your API Key: ") - data_pool_uuid = input("Type Data pool uuid you wish to use: ") - lines = [base_url, api_key, data_pool_uuid] - with open(config_relative_path, "w+") as file: - for line in lines: - file.write(line + '\n') - - print("Type VM-UUID (input ENTER to stop)") - with open(config_relative_path, "a") as file: #appends new content at the end without modifying the existing data - vm_input="test" - while (vm_input != ""): - vm_input = input(">> ") - file.write(vm_input + '\n') - print("UUIDs has been written in config") - print("\nConfiguration completed!") -if os.path.exists(config_relative_path) and os.path.getsize(config_relative_path) > 0: #check if file exists and not empty +config_relative_path = os.path.join(os.getcwd() , 'config.txt') +print(config_relative_path) +#check if config exists and not empty +if os.path.exists(config_relative_path) and os.path.getsize(config_relative_path) > 0: pass #do nothing else: print("Config file was not found or empty.. ") - config_edit() + config_edit(config_relative_path) -#importing API-KEY / IP / DATA POOL UUID from config -threelines = import_threelines(config_relative_path) -print(f"3 lines: {threelines}") -base_url=threelines[0] -api_key=threelines[1] -data_pool_uuid=threelines[2] - - -''' -with open(config_relative_path, "r") as f: - all_lines = f.readlines() - base_url = all_lines[0].strip('\n') - api_key = "jwt " + all_lines[1].strip('\n') #actual format for api_key. That was realy obvious DACOM >:C - data_pool_uuid = all_lines[2].strip('\n') -''' - -#importing VM-UUIDs +#importing API-KEY / IP / DATA POOL UUID / VM-UUIDs from config +#base_url=threelines[0] api_key=threelines[1] data_pool_uuid=threelines[2] +base_url, api_key, data_pool_uuid = import_threelines(config_relative_path) vm_uuids = import_vm_uuid(config_relative_path) -print(f"vm uuids: {vm_uuids}") -#so-called INT MAIN -menu_choice=0 +menu_choice=0 while(menu_choice != ""): #main menu loop - read_input=input("\nUitility Main Menu: \n1) Manage utility config \n2) Enter disk edit mode \n3) Show breif cluster overview \n4) Show VM info \n>>> ") - menu_choice=str(read_input) + print("\nUitility Main Menu:") + print("1) Manage utility config") + print("2) Enter disk edit mode") + print("3) Show breif cluster overview") + print("4) Show VM info") + print("5) Show data pools") + print("6) Show VMs Name / UUID") + menu_choice=str(input("\n>>> ")) if menu_choice == "1": print("\033[H\033[2J", end="") # clears cmd screen, but saves scrollback buffer print("1) Show current configuration \n2) Change configuraion") - read_input=input(">> ") - menu_choice=int(read_input) - if menu_choice == 1: + sub_choice=str(input("\n>>> ")) + + if sub_choice == "1": print("Current configuration:\n") with open(config_relative_path, "r") as f: print(f.read()) - if menu_choice == 2: - config_edit() + if sub_choice == "2": + config_edit(config_relative_path) + if menu_choice == "2": print("\033[H\033[2J", end="") print("Select option: \n 1) Delete vDisk by UUID \n 2) Delete ALL vDisks on selected Virtual Machine \n 3) Create Disk \n 4) Prepare VMs for Courses™") - read_input=input(">> ") - menu_choice=int(read_input) - - if menu_choice == 1: + sub_choice=str(input("\n>>> ")) + if sub_choice == "1": read_input=input("Input vDisk uuid to delete: ") vdisk_uuid=str(read_input) delete_disk(base_url , api_key , vdisk_uuid) - - if menu_choice == 2: + 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 domain_all_content = get_domain_all_content(base_url , api_key , vm_uuids[select_uuids]) @@ -97,26 +58,20 @@ while(menu_choice != ""): #main menu loop for x in disk_uuids: delete_disk(base_url , api_key , x) print("All attached vDisks has been deleted!") - - if menu_choice == 3: + if sub_choice == "3": vdisk_size=str(input("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 create_and_attach_disk(vm_uuids[select_uuids] , data_pool_uuid, vdisk_size, "falloc") - if menu_choice == 4: + if sub_choice == "4": print("#" * 5 , "Preparing VMs for Courses" , "#" * 5) 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) domain_all_content = get_domain_all_content(base_url , api_key , domain_uuid) + vm_info(base_url , api_key , domain_uuid) if domain_info: - print("=" * 14 , "Virtual Machine Info" , "=" * 15) - print(f"\t VM: {domain_info['verbose_name']}") - print(f"\t Power State: {power_state[domain_info['user_power_state']]}") #translating status code to "pretty name" - print(f"\t vDisks: {domain_info['vdisks_count']}") - print("-" * 19 , "vDisks Info" , "-" * 19) - get_disk_info(domain_all_content) disk_uuids = get_disk_uuids(base_url , api_key , domain_all_content) for y in disk_uuids: delete_disk(base_url , api_key , y) @@ -132,22 +87,12 @@ while(menu_choice != ""): #main menu loop if menu_choice == "3": cluster_info(base_url , api_key) - if menu_choice == "4": print("\033[H\033[2J", end="") for x in vm_uuids: vm_info(base_url , api_key , x) - if menu_choice == "5": data_pools(base_url , api_key) - if menu_choice == "6": vm_info_short(base_url , api_key) - -print("Exiting Utility..") -sys.exit() - - - - - +print("Exiting Utility..") \ No newline at end of file diff --git a/vm_info_short.py b/vm_info_short.py index f9b4d10..a318db6 100644 --- a/vm_info_short.py +++ b/vm_info_short.py @@ -1,22 +1,2 @@ #from main import base_url , api_key , requests -import requests - -def vm_info_short(base_url , api_key): #output data pool info - url= f"http://{base_url}//api/domains/" - response = requests.get(url , headers={'Authorization' : api_key}) - - if response.status_code == 200: - vm_info_short = response.json() - results_vm_info_short = vm_info_short['results'] - - print("\nShort VM overview") - print(f"\nVM total: {vm_info_short['count']}") - print("-" * 41) - for x in results_vm_info_short: - print(" "*16,f"VM {x['verbose_name']}") - print(f"UID: {x['id']}") - print("-" * 41) - - else: - print(f"Failed to retrieve data {response.status_code}") \ No newline at end of file