92 lines
No EOL
3.2 KiB
Python
Executable file
92 lines
No EOL
3.2 KiB
Python
Executable file
import click
|
|
import json
|
|
import requests
|
|
|
|
from decouple import config
|
|
from .helper import OTPCredentials, load_dump_pretty
|
|
|
|
|
|
def vm_command(command, otp, uuid):
|
|
data = {**otp.get_json(), "uuid": uuid, "action": command}
|
|
r = requests.post(f"{config('UCLOUD_API_SERVER')}/vm/action", json=data)
|
|
return r
|
|
|
|
|
|
@click.group()
|
|
def vm():
|
|
pass
|
|
|
|
|
|
@vm.command("create")
|
|
@click.option("--name", envvar="OTP_NAME", required=True)
|
|
@click.option("--realm", envvar="OTP_REALM", required=True)
|
|
@click.option("--seed", envvar="OTP_SEED", required=True)
|
|
@click.option("--specs", required=True)
|
|
@click.option("--image_uuid", required=True)
|
|
def create(name, realm, seed, specs, image_uuid):
|
|
with open(specs, "r") as specs_f:
|
|
specs = json.loads(specs_f.read())
|
|
data = {
|
|
**OTPCredentials(name, realm, seed).get_json(),
|
|
"specs": specs,
|
|
"image_uuid": image_uuid,
|
|
}
|
|
r = requests.post(f"{config('UCLOUD_API_SERVER')}/vm/create", json=data)
|
|
print(load_dump_pretty(r.content))
|
|
|
|
|
|
@vm.command("start")
|
|
@click.option("--name", envvar="OTP_NAME", required=True)
|
|
@click.option("--realm", envvar="OTP_REALM", required=True)
|
|
@click.option("--seed", envvar="OTP_SEED", required=True)
|
|
@click.option("--uuid", required=True)
|
|
def start(name, realm, seed, uuid):
|
|
r = vm_command("start", OTPCredentials(name, realm, seed), uuid)
|
|
print(load_dump_pretty(r.content))
|
|
|
|
|
|
@vm.command("stop")
|
|
@click.option("--name", envvar="OTP_NAME", required=True)
|
|
@click.option("--realm", envvar="OTP_REALM", required=True)
|
|
@click.option("--seed", envvar="OTP_SEED", required=True)
|
|
@click.option("--uuid", required=True)
|
|
def stop(name, realm, seed, uuid):
|
|
r = vm_command("stop", OTPCredentials(name, realm, seed), uuid)
|
|
print(load_dump_pretty(r.content))
|
|
|
|
|
|
@vm.command("delete")
|
|
@click.option("--name", envvar="OTP_NAME", required=True)
|
|
@click.option("--realm", envvar="OTP_REALM", required=True)
|
|
@click.option("--seed", envvar="OTP_SEED", required=True)
|
|
@click.option("--uuid", required=True)
|
|
def delete(name, realm, seed, uuid):
|
|
r = vm_command("delete", OTPCredentials(name, realm, seed), uuid)
|
|
print(load_dump_pretty(r.content))
|
|
|
|
|
|
@vm.command("status")
|
|
@click.option("--name", envvar="OTP_NAME", required=True)
|
|
@click.option("--realm", envvar="OTP_REALM", required=True)
|
|
@click.option("--seed", envvar="OTP_SEED", required=True)
|
|
@click.option("--uuid", required=True)
|
|
def status(name, realm, seed, uuid):
|
|
data = {"name": name, "realm": realm, "seed": seed, "uuid": uuid}
|
|
r = requests.get(f"{config('UCLOUD_API_SERVER')}/vm/status", json=data)
|
|
print(load_dump_pretty(r.content))
|
|
|
|
@vm.command("migrate")
|
|
@click.option("--name", envvar="OTP_NAME", required=True)
|
|
@click.option("--realm", envvar="OTP_REALM", required=True)
|
|
@click.option("--seed", envvar="OTP_SEED", required=True)
|
|
@click.option("--uuid", required=True)
|
|
@click.option("--destination", required=True)
|
|
def vm_migration(name, realm, seed, uuid, destination):
|
|
otp = OTPCredentials(name, realm, seed)
|
|
data = {
|
|
**otp.get_json(),
|
|
"uuid": uuid,
|
|
"destination": destination
|
|
}
|
|
r = requests.post(f"{config('UCLOUD_API_SERVER')}/vm/migrate", json=data)
|
|
print(load_dump_pretty(r.content)) |