From a091d75443e799d7c7ed4d585863697181079720 Mon Sep 17 00:00:00 2001 From: OVERLORD7F Date: Wed, 7 May 2025 16:20:36 +0300 Subject: [PATCH] Beggining of changes in file structure --- cluster_api.py | 3 + domain_api.py | 7 ++ main.py | 250 +++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 260 insertions(+) create mode 100644 cluster_api.py create mode 100644 domain_api.py create mode 100644 main.py diff --git a/cluster_api.py b/cluster_api.py new file mode 100644 index 0000000..3eaed63 --- /dev/null +++ b/cluster_api.py @@ -0,0 +1,3 @@ + +#from main import base_url , api_key , requests + diff --git a/domain_api.py b/domain_api.py new file mode 100644 index 0000000..6b89a8b --- /dev/null +++ b/domain_api.py @@ -0,0 +1,7 @@ +# functions for working with domain-api + + +#import requests +#import secrets #for generating unique names +#from main import base_url , api_key , requests , power_state +#from main import config_relative_path diff --git a/main.py b/main.py new file mode 100644 index 0000000..e844625 --- /dev/null +++ b/main.py @@ -0,0 +1,250 @@ +import requests +import sys +import secrets +from cluster_api import cluster_info +#from domain_api import * # functions foк working with domain-api + +power_state = ["Unknown" , "Off" , "Suspend" , "On"] #3 - on; 2 - suspend; 1 - off; 0 - unknown +config_relative_path = "Y:\\py\\config.txt" # absolute path to cluster config file + + +#importing API-KEY / IP / DATA POOL UUID from config +with open(config_relative_path, "r") as f: # using '\' (instead of '\\') throws syntax warning + 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') + + +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') + + +def vm_info(vm_uuids): + domain_uuid = vm_uuids + domain_info = get_domain_info(domain_uuid) + domain_all_content = get_domain_all_content(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) + +#get domain info "http://10.2.1.52/api/domains/uuid OR /domains/{id}/all-content/" +def get_domain_info(domain_uuid): + url= f"http://{base_url}/api/domains/{domain_uuid}" + response = requests.get(url , headers={'Authorization' : api_key}) + + if response.status_code == 200: #200 - OK + domain_data = response.json() + return domain_data #returns as dictionary! + else: + print(f"Failed to retrieve data {response.status_code}") + + +def get_domain_all_content(domain_uuid): + url= f"http://{base_url}/api/domains/{domain_uuid}/all-content" + response = requests.get(url , headers={'Authorization' : api_key}) + if response.status_code == 200: #200 - OK + domain_all_data = response.json() + return domain_all_data #returns as dictionary! + else: + print(f"Failed to retrieve data {response.status_code}") + + +def get_disk_uuids(domain_all_content): + #domain_all_content (type - dictionary) + #returns VMs vdisk uuids (type - list) + try: + # check for "vdisks" field in recieved json response + if 'vdisks' not in domain_all_content: + raise KeyError("No 'vdisks' field in recieved data") + # Get list of all vdisks + disks = domain_all_content['vdisks'] + # Extracting UUID for each disk + vdisk_uuid = [disk['id'] for disk in disks] + vdisk_size = [] + return vdisk_uuid + except KeyError as e: + print(f"ERROR: {e}") + return [] + except TypeError: + print("ERROR: unexpected data format") + return [] + +def get_disk_info(domain_all_content): + # check for "vdisks" field in recieved json response + if 'vdisks' not in domain_all_content: + print("No 'vdisks' field in recieved data") + return + + # get vdisk list + disks = domain_all_content['vdisks'] + + # check for disks + if not disks: + print("No 'disks' field in recieved data. \nProbably VM does not have any attached disks?") + return + + # Print info for each disk + for disk in disks: + # check for requiered fileds + if 'id' in disk and 'verbose_name' in disk and 'size' in disk: + print(f"Name: {disk['verbose_name']}") + print(f"UUID: {disk['id']}") + print(f"Size: {disk['size']} GB") + print("-" * 51) + else: + print("ERROR: failed to retrieve vdisk data.") + +def delete_disk(vdisk_uuid): + url = f"http://{base_url}/api/vdisks/{vdisk_uuid}/remove/" + headers={ + "Authorization" : api_key, + "Content-Type" : "application/json", + } + payload= { + "force": False, + "guaranteed": False, + "clean_type": "zero", + "clean_count": 1 + } + response = requests.post(url , headers=headers, json=payload) + if response.status_code == 200: + print(f"vDisk {vdisk_uuid} successfully deleted") + return True + else: + print(f"ERROR deleting disk {vdisk_uuid} :\n {response.status_code} - {response.text}") + return False + +def create_and_attach_disk(vm_id, data_pool_uuid, vdisk_size, preallocation): + domain_name=get_domain_info(vm_id) + disk_name=domain_name["verbose_name"]+"_"+secrets.token_hex(5) #generates unique hex id. this method can generate ~million unique ids + url = f"http://{base_url}/api/domains/{vm_id}/create-attach-vdisk/" + headers={ + "Authorization" : api_key, + "Content-Type" : "application/json", + } + payload= { + "verbose_name": disk_name, + "preallocation": preallocation, + "size": vdisk_size, + "datapool": data_pool_uuid, + "target_bus": "virtio", + } + response = requests.post(url , headers=headers, json=payload) + if response.status_code == 200: + print(f"\nvDisk {disk_name} - {vdisk_size}GB has been created") + return True + else: + print(f"ERROR creating vDisk :\n {response.status_code} - {response.text}") + return False + + +#importing VM-UUIDs +vm_uuids = [] +with open(config_relative_path, "r") as f: + for i in range(3): # ignoring 2 first lines (IP, API-KEY) + next(f) + for line in f: + line = line.strip('\n') + if line: # checks if line is empty (EOF). ESSENTIAL, DO NOT REMOVE + vm_uuids.append(line) + +#so-called INT MAIN +menu_choice=0 +while(menu_choice != ""): #main menu loop + read_input=input("\nUitility Main Menu: \n1) Edit config \n2) Enter disk edit mode \n3) Show breif cluster overview \n4) Show VM info \n>>> ") + menu_choice=str(read_input) + + if menu_choice == "1": + config_edit() + if menu_choice == "2": + print("\033[H\033[2J", end="") # clears cmd screen, but saves scrollback buffer + 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: + read_input=input("Input vDisk uuid to delete: ") + vdisk_uuid=str(read_input) + delete_disk(vdisk_uuid) + if menu_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 + print(f"actual selected uuid = {select_uuids}") + print(vm_uuids[select_uuids]) + domain_all_content = get_domain_all_content(vm_uuids[select_uuids]) + disk_uuids = get_disk_uuids(domain_all_content) + for x in disk_uuids: + delete_disk(x) + print("All attached vDisks has been deleted!") + if menu_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 + print(f"actual selected uuid = {select_uuids}") + print(vm_uuids[select_uuids]) + create_and_attach_disk(vm_uuids[select_uuids] , data_pool_uuid, vdisk_size, "falloc") + if menu_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(domain_uuid) + domain_all_content = get_domain_all_content(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(domain_all_content) + for y in disk_uuids: + delete_disk(y) + print("All attached vDisks has been deleted!") + for z in vm_uuids: # only for creating disks + domain_uuid = z.strip('\n') + domain_info = get_domain_info(domain_uuid) + domain_all_content = get_domain_all_content(domain_uuid) + if domain_info: + create_and_attach_disk(domain_uuid , data_pool_uuid, 10, "falloc") + create_and_attach_disk(domain_uuid , data_pool_uuid, 20, "falloc") + create_and_attach_disk(domain_uuid , data_pool_uuid, 20, "falloc") + + if menu_choice == "3": + cluster_info() + + if menu_choice == "4": + print("\033[H\033[2J", end="") + print(f"vm uuids - {vm_uuids}") + for x in vm_uuids: + vm_info(x) + + +print("Exiting Utility..") +sys.exit() + + + + +