mirror of
https://github.com/OVERLORD7F/SVMU.git
synced 2025-10-01 21:52:47 +03:00
- 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)
This commit is contained in:
OVERLORD7F
@@ -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}")
|
||||
137
domain_api.py
137
domain_api.py
@@ -1,7 +1,136 @@
|
||||
# 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
|
||||
import requests
|
||||
import secrets #for generating unique names
|
||||
#from main import power_state
|
||||
|
||||
|
||||
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
|
||||
186
main.py
186
main.py
@@ -1,20 +1,15 @@
|
||||
import requests
|
||||
import sys
|
||||
import secrets
|
||||
from cluster_api import cluster_info
|
||||
#from domain_api import * # functions foк working with domain-api
|
||||
import os
|
||||
from cluster_api import *
|
||||
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
|
||||
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')
|
||||
|
||||
#config_relative_path = "Y:\\py\\SpaceVM_VM_Utility\\config.txt"
|
||||
|
||||
def config_edit():
|
||||
read_input=input("Create new config file? (Y / N): ")
|
||||
@@ -28,7 +23,6 @@ def config_edit():
|
||||
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"
|
||||
@@ -36,130 +30,16 @@ def config_edit():
|
||||
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
|
||||
|
||||
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 = []
|
||||
@@ -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])
|
||||
disk_uuids = get_disk_uuids(domain_all_content)
|
||||
domain_all_content = get_domain_all_content(base_url , api_key , vm_uuids[select_uuids])
|
||||
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_all_content = get_domain_all_content(domain_uuid)
|
||||
domain_info = get_domain_info(base_url , api_key , 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_all_content = get_domain_all_content(domain_uuid)
|
||||
domain_info = get_domain_info(base_url , api_key , 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(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, 10, "falloc")
|
||||
create_and_attach_disk(base_url , api_key , 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..")
|
||||
|
||||
Reference in New Issue
Block a user