302 lines
11 KiB
Python
302 lines
11 KiB
Python
import requests
|
|
|
|
class IaaSClient:
|
|
def __init__(self, api_key, base_url="http://localhost:5000/api"):
|
|
self.base_url = base_url
|
|
self.api_key = api_key
|
|
|
|
def _make_request(self, method, endpoint, data=None):
|
|
headers = {
|
|
'X-API-KEY': self.api_key,
|
|
'Content-Type': 'application/json'
|
|
}
|
|
url = f"{self.base_url}/{endpoint}"
|
|
|
|
try:
|
|
response = requests.request(method, url, headers=headers, json=data)
|
|
if response.status_code == 401:
|
|
print("Authentication failed. Please check your API key.")
|
|
return None
|
|
return response.json()
|
|
except requests.exceptions.RequestException as e:
|
|
print(f"API Error: {str(e)}")
|
|
return None
|
|
|
|
# Universe CRUD
|
|
def get_universes(self):
|
|
return self._make_request('GET', 'universes')
|
|
|
|
def get_universe(self, universe_id):
|
|
return self._make_request('GET', f'universes/{universe_id}')
|
|
|
|
def create_universe(self, name, description):
|
|
return self._make_request('POST', 'universes', {'name': name, 'description': description})
|
|
|
|
def update_universe(self, universe_id, data):
|
|
return self._make_request('PUT', f'universes/{universe_id}', data)
|
|
|
|
def delete_universe(self, universe_id):
|
|
return self._make_request('DELETE', f'universes/{universe_id}')
|
|
|
|
# Project CRUD
|
|
def get_projects(self):
|
|
return self._make_request('GET', 'projects')
|
|
|
|
def get_project(self, project_id):
|
|
return self._make_request('GET', f'projects/{project_id}')
|
|
|
|
def create_project(self, name, universe_id):
|
|
return self._make_request('POST', 'projects', {'name': name, 'universe_id': universe_id})
|
|
|
|
def update_project(self, project_id, data):
|
|
return self._make_request('PUT', f'projects/{project_id}', data)
|
|
|
|
def delete_project(self, project_id):
|
|
return self._make_request('DELETE', f'projects/{project_id}')
|
|
|
|
# virtual_data_centers CRUD
|
|
def get_virtual_data_centers(self):
|
|
return self._make_request('GET', 'virtual_data_centers')
|
|
|
|
def get_virtual_data_center(self, virtual_data_center_id):
|
|
return self._make_request('GET', f'virtual_data_centers/{virtual_data_center_id}')
|
|
|
|
def create_virtual_data_center(self, name, project_id):
|
|
return self._make_request('POST', 'virtual_data_centers', {'name': name, 'project_id': project_id})
|
|
|
|
def update_virtual_data_center(self, virtual_data_center_id, data):
|
|
return self._make_request('PUT', f'virtual_data_centers/{virtual_data_center_id}', data)
|
|
|
|
def delete_virtual_data_center(self, virtual_data_center_id):
|
|
return self._make_request('DELETE', f'virtual_data_centers/{virtual_data_center_id}')
|
|
|
|
# Network CRUD
|
|
def get_networks(self):
|
|
return self._make_request('GET', 'networks')
|
|
|
|
def get_network(self, network_id):
|
|
return self._make_request('GET', f'networks/{network_id}')
|
|
|
|
def create_network(self, name, virtual_data_center_id):
|
|
return self._make_request('POST', 'networks', {'name': name, 'virtual_data_center_id': virtual_data_center_id})
|
|
|
|
def update_network(self, network_id, data):
|
|
return self._make_request('PUT', f'networks/{network_id}', data)
|
|
|
|
def delete_network(self, network_id):
|
|
return self._make_request('DELETE', f'networks/{network_id}')
|
|
|
|
# Network Port CRUD
|
|
def get_network_ports(self):
|
|
return self._make_request('GET', 'network_ports')
|
|
|
|
def get_network_port(self, network_port_id):
|
|
return self._make_request('GET', f'network_ports/{network_port_id}')
|
|
|
|
def create_network_port(self, data):
|
|
return self._make_request('POST', 'network_ports', data)
|
|
|
|
def update_network_port(self, network_port_id, data):
|
|
return self._make_request('PUT', f'network_ports/{network_port_id}', data)
|
|
|
|
def delete_network_port(self, network_port_id):
|
|
return self._make_request('DELETE', f'network_ports/{network_port_id}')
|
|
|
|
def get_network_ports_by_network(self,network_id):
|
|
return self._make_request('GET', f'network_ports/by_network/{network_id}')
|
|
|
|
|
|
# Region CRUD
|
|
def get_regions(self):
|
|
return self._make_request('GET', 'regions')
|
|
|
|
def get_region(self, region_id):
|
|
return self._make_request('GET', f'regions/{region_id}')
|
|
|
|
def create_region(self, name, description):
|
|
return self._make_request('POST', 'regions', {'name': name, 'description': description})
|
|
|
|
def update_region(self, region_id, data):
|
|
return self._make_request('PUT', f'regions/{region_id}', data)
|
|
|
|
def delete_region(self, region_id):
|
|
return self._make_request('DELETE', f'regions/{region_id}')
|
|
|
|
# Region Access CRUD
|
|
def get_region_accesses(self):
|
|
return self._make_request('GET', 'region_access')
|
|
|
|
def get_region_access(self, region_access_id):
|
|
return self._make_request('GET', f'region_access/{region_access_id}')
|
|
|
|
def create_region_access(self, data):
|
|
return self._make_request('POST', 'region_access', data)
|
|
|
|
def update_region_access(self, region_access_id, data):
|
|
return self._make_request('PUT', f'region_access/{region_access_id}', data)
|
|
|
|
def delete_region_access(self, region_access_id):
|
|
return self._make_request('DELETE', f'region_access/{region_access_id}')
|
|
|
|
# Workload Host CRUD
|
|
def get_workload_hosts(self):
|
|
return self._make_request('GET', 'workload_hosts')
|
|
|
|
def get_workload_host(self, workload_host_id):
|
|
return self._make_request('GET', f'workload_hosts/{workload_host_id}')
|
|
|
|
def create_workload_host(self, data):
|
|
return self._make_request('POST', 'workload_hosts', data)
|
|
|
|
def update_workload_host(self, workload_host_id, data):
|
|
return self._make_request('PUT', f'workload_hosts/{workload_host_id}', data)
|
|
|
|
def delete_workload_host(self, workload_host_id):
|
|
return self._make_request('DELETE', f'workload_hosts/{workload_host_id}')
|
|
|
|
# Image CRUD
|
|
def get_images(self):
|
|
return self._make_request('GET', 'images')
|
|
|
|
def get_image(self, image_id):
|
|
return self._make_request('GET', f'images/{image_id}')
|
|
|
|
def create_image(self, data):
|
|
print(":creating image")
|
|
return self._make_request('POST', 'images', data)
|
|
|
|
def update_image(self, image_id, data):
|
|
return self._make_request('PUT', f'images/{image_id}', data)
|
|
|
|
def delete_image(self, image_id):
|
|
return self._make_request('DELETE', f'images/{image_id}')
|
|
|
|
# Container CRUD
|
|
def get_containers(self):
|
|
return self._make_request('GET', 'workloads/containers')
|
|
|
|
def get_container(self, container_id):
|
|
return self._make_request('GET', f'workloads/containers/{container_id}')
|
|
|
|
def create_container(self, data):
|
|
# TODO - If the create failes, i.e not enough hosts, we need to catch andhandle the 400 response
|
|
return self._make_request('POST', 'workloads/containers', data)
|
|
|
|
def update_container(self, container_id, data):
|
|
return self._make_request('PUT', f'workloads/containers/{container_id}', data)
|
|
|
|
def delete_container(self, container_id):
|
|
return self._make_request('DELETE', f'workloads/containers/{container_id}')
|
|
|
|
# Virtual Machine CRUD
|
|
def get_virtual_machines(self):
|
|
return self._make_request('GET', 'workloads/virtual_machines')
|
|
|
|
def get_virtual_machine(self, virtual_machine_id):
|
|
return self._make_request('GET', f'workloads/virtual_machines/{virtual_machine_id}')
|
|
|
|
def create_virtual_machine(self, data):
|
|
# TODO - If the create failes, i.e not enough hosts, we need to catch andhandle the 400 response
|
|
return self._make_request('POST', 'workloads/virtual_machines', data)
|
|
|
|
def update_virtual_machine(self, virtual_machine_id, data):
|
|
return self._make_request('PUT', f'workloads/virtual_machines/{virtual_machine_id}', data)
|
|
|
|
def delete_virtual_machine(self, virtual_machine_id):
|
|
return self._make_request('DELETE', f'workloads/virtual_machines/{virtual_machine_id}')
|
|
|
|
# Workload CRUD
|
|
def get_workloads(self):
|
|
return self._make_request('GET', 'workloads')
|
|
|
|
def get_workload(self, workload_id):
|
|
return self._make_request('GET', f'workloads/{workload_id}')
|
|
|
|
def create_workload(self, data):
|
|
return self._make_request('POST', 'workloads', data)
|
|
|
|
def update_workload(self, workload_id, data):
|
|
return self._make_request('PUT', f'workloads/{workload_id}', data)
|
|
|
|
def delete_workload(self, workload_id):
|
|
return self._make_request('DELETE', f'workloads/{workload_id}')
|
|
|
|
|
|
|
|
# Workload Host Resources
|
|
def get_pooled_resources_for_host(self, host_id):
|
|
"""
|
|
Fetch pooled resources for a specific workload host.
|
|
"""
|
|
return self._make_request('GET', f'workload-hosts/{host_id}/pooled-resources')
|
|
|
|
def get_fixed_resources_for_host(self, host_id):
|
|
"""
|
|
Fetch fixed resources for a specific workload host.
|
|
"""
|
|
return self._make_request('GET', f'workload-hosts/{host_id}/fixed-resources')
|
|
|
|
def get_ovs_bridges_for_host(self, host_id):
|
|
"""
|
|
Fetch OVS bridges for a specific workload host.
|
|
"""
|
|
return self._make_request('GET', f'workload-hosts/{host_id}/ovs-bridges')
|
|
|
|
|
|
|
|
# Workloads for Host
|
|
def get_workloads_for_host(self, host_id):
|
|
"""
|
|
Fetch all workloads (containers and virtual machines) for a specific host.
|
|
"""
|
|
# Fetch all containers and virtual machines
|
|
containers = self.get_containers() or []
|
|
virtual_machines = self.get_virtual_machines() or []
|
|
|
|
# Filter workloads by host_id
|
|
workloads = []
|
|
for container in containers:
|
|
if container.get('workload_host_id') == host_id:
|
|
container['workload_type'] = 'container' # Add workload type
|
|
workloads.append(container)
|
|
|
|
for vm in virtual_machines:
|
|
if vm.get('workload_host_id') == host_id:
|
|
vm['workload_type'] = 'virtual_machine' # Add workload type
|
|
workloads.append(vm)
|
|
|
|
return workloads
|
|
|
|
|
|
|
|
def add_label(self, host_id, key, value, admin_only_view=False, admin_only_set=False):
|
|
"""Add a new label to a workload host"""
|
|
endpoint = f'workload_hosts/{host_id}/labels'
|
|
data = {
|
|
'label_key': key,
|
|
'label_value': value,
|
|
'admin_only_view': admin_only_view,
|
|
'admin_only_set': admin_only_set
|
|
}
|
|
return self._make_request('POST', endpoint, data)
|
|
|
|
def remove_label(self, host_id, key):
|
|
"""Remove a label from a workload host"""
|
|
endpoint = f'workload_hosts/{host_id}/labels'
|
|
data = {
|
|
'label_key': key
|
|
}
|
|
return self._make_request('DELETE', endpoint, data)
|
|
|
|
def get_labels_for_host(self, host_id):
|
|
"""
|
|
Fetch all labels for a specific workload host
|
|
|
|
Args:
|
|
host_id (str/UUID): The ID of the workload host
|
|
|
|
Returns:
|
|
list: List of label dictionaries or None if error occurs
|
|
"""
|
|
endpoint = f'workload_hosts/{host_id}/labels'
|
|
return self._make_request('GET', endpoint) |