uncloud/uncloud-cli
2
0
Fork 0
Code Issues 3 Pull requests Projects Releases Wiki Activity

a

Browse source
This commit is contained in:
ahmadbilalkhalid 2019-09-13 17:33:19 +05:00
commit c90dacabe5
7 changed files with 43 additions and 46 deletions
Download patch file Download diff file Expand all files Collapse all files

15
commands/helper.py
View file

@ -1,21 +1,18 @@
import click
import json
from dataclasses import dataclass
from pyotp import TOTP
@dataclass
class OTPCredentials:
name: str
realm: str
seed: str
def __init__(self, name, realm, seed):
self.name = name # type: str
self.realm = realm # type: str
self.seed = seed # type: str
def get_json(self):
r = {"name": self.name, "realm": self.realm, "token": TOTP(self.seed).now()}
return r
return {"name": self.name, "realm": self.realm, "token": TOTP(self.seed).now()}
def load_dump_pretty(content):
parsed = json.loads(content)
return json.dumps(parsed, indent=4, sort_keys=True)
return json.dumps(parsed, indent=4, sort_keys=True)

13
commands/host.py
View file

@ -1,9 +1,10 @@
import click
import json
import requests
from commands.helper import OTPCredentials, load_dump_pretty
from decouple import config
from .helper import OTPCredentials, load_dump_pretty
import click
import requests
@click.group()
@ -25,11 +26,11 @@ def add_host(name, realm, seed, specs, hostname):
"specs": specs,
"hostname": hostname,
}
r = requests.post(f"{config('UCLOUD_API_SERVER')}/host/create", json=data)
r = requests.post("{}/host/create".format(config('UCLOUD_API_SERVER')), json=data)
print(load_dump_pretty(r.content))
@host.command("list")
def list_host():
r = requests.get(f"{config('UCLOUD_API_SERVER')}/host/list")
print(load_dump_pretty(r.content))
r = requests.get("{}/host/list".format(config('UCLOUD_API_SERVER')))
print(load_dump_pretty(r.content))

12
commands/image.py
View file

@ -1,10 +1,8 @@
import click
import json
import requests
from commands.helper import load_dump_pretty
from decouple import config
from .helper import OTPCredentials, load_dump_pretty
import click
import requests
@click.group()
def image():
@ -16,7 +14,7 @@ def image():
@click.option("--private", is_flag=True)
def _list(public, private):
if public:
r = requests.get(f"{config('UCLOUD_API_SERVER')}/image/list-public")
r = requests.get("{}/image/list-public".format(config('UCLOUD_API_SERVER')))
print(load_dump_pretty(r.content))
@ -26,5 +24,5 @@ def _list(public, private):
@click.option("--image_store_name", required=True)
def create_from_file(name, uuid, image_store_name):
data = {"name": name, "uuid": uuid, "image_store": image_store_name}
r = requests.post(f"{config('UCLOUD_API_SERVER')}/image/create", json=data)
r = requests.post("{}/image/create".format(config('UCLOUD_API_SERVER')), json=data)
print(load_dump_pretty(r.content))

13
commands/user.py
View file

@ -1,9 +1,8 @@
import click
import json
import requests
from commands.helper import OTPCredentials, load_dump_pretty
from decouple import config
from .helper import OTPCredentials, load_dump_pretty
import click
import requests
@click.group()
@ -17,7 +16,7 @@ def user():
@click.option("--seed", envvar="OTP_SEED", required=True)
def list_files(name, realm, seed):
data = OTPCredentials(name, realm, seed).get_json()
r = requests.get(f"{config('UCLOUD_API_SERVER')}/user/files", json=data)
r = requests.get("{}/user/files".format(config('UCLOUD_API_SERVER')), json=data)
print(load_dump_pretty(r.content))
@ -27,5 +26,5 @@ def list_files(name, realm, seed):
@click.option("--seed", envvar="OTP_SEED", required=True)
def list_vms(name, realm, seed):
data = OTPCredentials(name, realm, seed).get_json()
r = requests.get(f"{config('UCLOUD_API_SERVER')}/user/vms", json=data)
r = requests.get("{}/user/vms".format(config('UCLOUD_API_SERVER')), json=data)
print(load_dump_pretty(r.content))

18
commands/vm.py
View file

@ -1,14 +1,15 @@
import click
import json
import requests
from commands.helper import OTPCredentials, load_dump_pretty
from decouple import config
from .helper import OTPCredentials, load_dump_pretty
import click
import requests
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)
r = requests.post("{}/vm/action".format(config('UCLOUD_API_SERVER')), json=data)
return r
@ -31,7 +32,7 @@ def create(name, realm, seed, specs, image_uuid):
"specs": specs,
"image_uuid": image_uuid,
}
r = requests.post(f"{config('UCLOUD_API_SERVER')}/vm/create", json=data)
r = requests.post("{}/vm/create".format(config('UCLOUD_API_SERVER')), json=data)
print(load_dump_pretty(r.content))
@ -72,9 +73,10 @@ def delete(name, realm, seed, uuid):
@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)
r = requests.get("{}/vm/status".format(config('UCLOUD_API_SERVER')), 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)
@ -88,5 +90,5 @@ def vm_migration(name, realm, seed, uuid, destination):
"uuid": uuid,
"destination": destination
}
r = requests.post(f"{config('UCLOUD_API_SERVER')}/vm/migrate", json=data)
print(load_dump_pretty(r.content))
r = requests.post("{}/vm/migrate".format(config('UCLOUD_API_SERVER')), json=data)
print(load_dump_pretty(r.content))