- Moved api-related functions to sepate modules

- Now config file has to be placed in the same directory as main.py
- Checks for config file (if exist / if empty)

cluster_api.py

@@ -1,3 +1,26 @@
 
 #from main import base_url , api_key , requests
+import requests
 
+def cluster_info(base_url , api_key): #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}")  

domain_api.py

@@ -1,7 +1,136 @@
 # functions for working with domain-api
 
 
-#import requests
+import requests
-#import secrets #for generating unique names
+import secrets #for generating unique names
-#from main import base_url , api_key , requests , power_state 
+#from main import power_state 
-#from main import config_relative_path
+
+
+power_state = ["Unknown" , "Off" , "Suspend" , "On"] #3 - on; 2 - suspend; 1 - off; 0 - unknown
+
+
+def get_domain_info(base_url , api_key , 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(base_url, api_key, 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(base_url , api_key , 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 delete_disk(base_url , api_key , 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 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 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:
+        print("\n" , "=" * 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)
+
+
+
+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
+    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  

main.py

@@ -1,20 +1,15 @@
-import requests
 import sys
-import secrets
+import os
-from cluster_api import cluster_info
+from cluster_api import *
-#from domain_api import * # functions foк working with domain-api
+from domain_api import * 
 
 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  
 
+#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')
 
-#importing API-KEY / IP / DATA POOL UUID from config
+#config_relative_path = "Y:\\py\\SpaceVM_VM_Utility\\config.txt" 
-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): ") 
@@ -27,140 +22,25 @@ def config_edit():
         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')        
+                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        
-
 
+if os.path.exists(config_relative_path) and os.path.getsize(config_relative_path) > 0: #check if file exists and not empty
+#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')
+else:
+    print("Config file was not found or empty.. ")
+    config_edit()
+   
 #importing VM-UUIDs
 vm_uuids = [] 
 with open(config_relative_path, "r") as f:
@@ -171,6 +51,7 @@ with open(config_relative_path, "r") as f:
         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
@@ -187,16 +68,16 @@ while(menu_choice != ""):    #main menu loop
         if menu_choice == 1:
             read_input=input("Input vDisk uuid to delete: ")
             vdisk_uuid=str(read_input)
-            delete_disk(vdisk_uuid)      
+            delete_disk(base_url , api_key , 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])
+            domain_all_content = get_domain_all_content(base_url , api_key , vm_uuids[select_uuids])
-            disk_uuids = get_disk_uuids(domain_all_content)
+            disk_uuids = get_disk_uuids(base_url , api_key , domain_all_content)
             for x in disk_uuids:
-                delete_disk(x)
+                delete_disk(base_url , api_key , x)
             print("All attached vDisks has been deleted!")
         if menu_choice == 3:
             vdisk_size=str(input("Enter disk size (GB): "))
@@ -209,8 +90,8 @@ while(menu_choice != ""):    #main menu loop
             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_info = get_domain_info(base_url , api_key , domain_uuid)
-                domain_all_content = get_domain_all_content(domain_uuid)
+                domain_all_content = get_domain_all_content(base_url , api_key , domain_uuid)
                 if domain_info:
                     print("=" * 14 , "Virtual Machine Info" , "=" * 15)
                     print(f"\t VM: {domain_info['verbose_name']}")
@@ -218,27 +99,26 @@ while(menu_choice != ""):    #main menu loop
                     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)
+                    disk_uuids = get_disk_uuids(base_url , api_key , domain_all_content)
                     for y in disk_uuids:
-                        delete_disk(y)
+                        delete_disk(base_url , api_key , 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_info = get_domain_info(base_url , api_key , domain_uuid)
-                domain_all_content = get_domain_all_content(domain_uuid)
+                domain_all_content = get_domain_all_content(base_url , api_key , domain_uuid)
                 if domain_info:
-                    create_and_attach_disk(domain_uuid , data_pool_uuid, 10, "falloc")
+                    create_and_attach_disk(base_url , api_key , domain_uuid , data_pool_uuid, 10, "falloc")
-                    create_and_attach_disk(domain_uuid , data_pool_uuid, 20, "falloc")
+                    create_and_attach_disk(base_url , api_key , domain_uuid , data_pool_uuid, 20, "falloc")
-                    create_and_attach_disk(domain_uuid , data_pool_uuid, 20, "falloc")
+                    create_and_attach_disk(base_url , api_key , domain_uuid , data_pool_uuid, 20, "falloc")
     
     if menu_choice == "3":
-        cluster_info()
+        cluster_info(base_url , api_key)
     
     if menu_choice == "4":
         print("\033[H\033[2J", end="")
-        print(f"vm uuids - {vm_uuids}")
         for x in vm_uuids:
-            vm_info(x)
+            vm_info(base_url , api_key , x)
 
 
 print("Exiting Utility..")