mirror of
https://ak-git.vectorsigma.ru/terghalin/metalcheck-cli.git
synced 2026-02-22 10:00:30 +09:00
92 lines
3.0 KiB
Python
92 lines
3.0 KiB
Python
import click
|
|
import requests
|
|
|
|
@click.group()
|
|
@click.pass_context
|
|
def virtual_machines(ctx):
|
|
"""Commands for managing Virtual Machines."""
|
|
pass
|
|
|
|
@virtual_machines.command("list")
|
|
@click.pass_context
|
|
def list_command(ctx):
|
|
"""List all virtual machines."""
|
|
base_url = ctx.obj["BASE_URL"]
|
|
list_virtual_machines(base_url)
|
|
|
|
@virtual_machines.command("add")
|
|
@click.pass_context
|
|
def add_command(ctx):
|
|
"""Add a new virtual machine."""
|
|
base_url = ctx.obj["BASE_URL"]
|
|
add_virtual_machine(base_url)
|
|
|
|
@virtual_machines.command("delete")
|
|
@click.pass_context
|
|
def delete_command(ctx):
|
|
"""Delete a virtual machine."""
|
|
base_url = ctx.obj["BASE_URL"]
|
|
delete_virtual_machine(base_url)
|
|
|
|
def list_virtual_machines(base_url):
|
|
"""List all virtual machines."""
|
|
vm_url = f"{base_url}/vm/data"
|
|
try:
|
|
response = requests.get(vm_url)
|
|
if response.status_code == 200:
|
|
virtual_machines = response.json().get("virtual_machines", [])
|
|
click.echo("\n💻 Virtual Machines:")
|
|
for vm in virtual_machines:
|
|
click.echo(
|
|
f"ID: {vm[0]}, Name: {vm[1]}, Location: {vm[2]}, "
|
|
f"CPU: {vm[3]}, Memory: {vm[4]}, Storage: {vm[5]}, "
|
|
f"Type: {vm[6]}"
|
|
)
|
|
else:
|
|
click.echo(f"Error: {response.status_code} - {response.text}")
|
|
except requests.RequestException as e:
|
|
click.echo(f"Error: {e}")
|
|
|
|
def add_virtual_machine(base_url):
|
|
"""Add a new virtual machine."""
|
|
vm_url = f"{base_url}/vm/data"
|
|
try:
|
|
name = click.prompt("Name")
|
|
location = click.prompt("Location")
|
|
cpu = click.prompt("CPU (cores)", type=int)
|
|
memory = click.prompt("Memory (GB)")
|
|
storage = click.prompt("Storage (e.g., 500GB SSD)")
|
|
vm_type = click.prompt("Type (e.g., cx-21)")
|
|
|
|
data = {
|
|
"name": name,
|
|
"location": location,
|
|
"cpu": cpu,
|
|
"memory": memory,
|
|
"storage": storage,
|
|
"type": vm_type,
|
|
}
|
|
|
|
response = requests.post(vm_url, json=data)
|
|
if response.status_code in [200, 201]:
|
|
response_data = response.json()
|
|
message = response_data.get("message", "Virtual machine added successfully!")
|
|
click.echo(f"✅ {message}")
|
|
else:
|
|
click.echo(f"Error: {response.status_code} - {response.text}")
|
|
except requests.RequestException as e:
|
|
click.echo(f"Error: {e}")
|
|
|
|
def delete_virtual_machine(base_url):
|
|
"""Delete a virtual machine."""
|
|
vm_url = f"{base_url}/vm/data"
|
|
try:
|
|
vm_id = click.prompt("Enter the ID of the virtual machine to delete", type=int)
|
|
response = requests.delete(f"{vm_url}/{vm_id}")
|
|
if response.status_code == 200:
|
|
click.echo("✅ Virtual machine deleted successfully!")
|
|
else:
|
|
click.echo(f"Error: {response.status_code} - {response.text}")
|
|
except requests.RequestException as e:
|
|
click.echo(f"Error: {e}")
|