diff --git a/pyth-api1.py b/pyth-api1.py deleted file mode 100644 index 1d12f34..0000000 --- a/pyth-api1.py +++ /dev/null @@ -1,273 +0,0 @@ -import requests -import sys -import secrets #for generating unique names - - -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') - - -#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) - -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 cluster_info(): #output short clusters overview - url= f"http://{base_url}/api/clusters" - response = requests.get(url , headers={'Authorization' : api_key}) - - if response.status_code == 200: - cluster_info = response.json() - results_cluster_info = cluster_info['results'] - - print("\nShort clusters overview:") - print(f"\nClusters total: {cluster_info['count']}") - print("-" * 51) - for x in results_cluster_info: - - print(f"\nCluster Name: {x['verbose_name']} ({x['status']})") - print(f"Nodes: {x['nodes_count']}") - print(f"Total CPU: {x['cpu_count']} Cores || CPU Usage: {round(x['cpu_used_percent_user'] , 2)}%")#output is rounded by 2 - print(f"Total RAM: {int(x['memory_count']/1024)}GB || RAM Usage: {round(x['mem_used_percent_user'] , 2)}%") #RAM pretty output = mb-to-gb + set 'int' to remove .0 - - print("-" * 51) - else: - print(f"Failed to retrieve data {response.status_code}") - -#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 - - -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) - -#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() - - - - -