add(app): create python application
This commit is contained in:
89
cli/commands/vm.py
Normal file
89
cli/commands/vm.py
Normal file
@@ -0,0 +1,89 @@
|
||||
import click
|
||||
import requests
|
||||
|
||||
BASE_URL = "http://localhost:8000/vm" # Backend URL for Virtual Machines API
|
||||
|
||||
@click.group()
|
||||
def virtual_machines():
|
||||
"""Commands for managing Virtual Machines."""
|
||||
pass
|
||||
|
||||
@virtual_machines.command("list")
|
||||
def list_command():
|
||||
handle_command("list")
|
||||
|
||||
@virtual_machines.command("add")
|
||||
def add_command():
|
||||
handle_command("add")
|
||||
|
||||
@virtual_machines.command("delete")
|
||||
def delete_command():
|
||||
handle_command("delete")
|
||||
|
||||
def handle_command(action):
|
||||
"""Handle commands related to Virtual Machines."""
|
||||
if action == "list":
|
||||
list_virtual_machines()
|
||||
elif action == "add":
|
||||
add_virtual_machine()
|
||||
elif action == "delete":
|
||||
delete_virtual_machine()
|
||||
|
||||
def list_virtual_machines():
|
||||
"""List all virtual machines."""
|
||||
try:
|
||||
response = requests.get(f"{BASE_URL}/data")
|
||||
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():
|
||||
"""Add a new virtual machine."""
|
||||
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(f"{BASE_URL}/data", 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():
|
||||
"""Delete a virtual machine."""
|
||||
try:
|
||||
vm_id = click.prompt("Enter the ID of the virtual machine to delete", type=int)
|
||||
response = requests.delete(f"{BASE_URL}/data/{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}")
|
||||
Reference in New Issue
Block a user