much better cli

commands/vm.py

@@ -0,0 +1,75 @@
+import click
+import json
+import requests
+
+from decouple import config
+from helper import OTPCredentials
+
+
+def vm_action(action, otp, vmid):
+    data = {**otp.get_json(), "vmid": vmid}
+    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", required=True)
+@click.option("--realm", required=True)
+@click.option("--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(json.loads(r.content))
+
+
+@vm.command("start")
+@click.option("--name", required=True)
+@click.option("--realm", required=True)
+@click.option("--seed", required=True)
+@click.option("--vmid", required=True)
+def start(name, realm, seed, vmid):
+    r = vm_action("start", OTPCredentials(name, realm, seed), vmid)
+    print(json.loads(r.content))
+
+
+@vm.command("suspend")
+@click.option("--name", required=True)
+@click.option("--realm", required=True)
+@click.option("--seed", required=True)
+@click.option("--vmid", required=True)
+def suspend(name, realm, seed, vmid):
+    r = vm_action("suspend", OTPCredentials(name, realm, seed), vmid)
+    print(json.loads(r.content))
+
+
+@vm.command("resume")
+@click.option("--name", required=True)
+@click.option("--realm", required=True)
+@click.option("--seed", required=True)
+@click.option("--vmid", required=True)
+def resume(name, realm, seed, vmid):
+    r = vm_action("resume", OTPCredentials(name, realm, seed), vmid)
+    print(json.loads(r.content))
+
+
+@vm.command("shutdown")
+@click.option("--name", required=True)
+@click.option("--realm", required=True)
+@click.option("--seed", required=True)
+@click.option("--vmid", required=True)
+def shutdown(name, realm, seed, vmid):
+    r = vm_action("shutdown", OTPCredentials(name, realm, seed), vmid)
+    print(json.loads(r.content))