90 lines
3.1 KiB
Python
Executable file
90 lines
3.1 KiB
Python
Executable file
from commands.helper import OTPCredentials, load_dump_pretty
|
|
from config import env_vars
|
|
from os.path import join as join_path
|
|
|
|
import click
|
|
import requests
|
|
|
|
|
|
@click.group()
|
|
def user():
|
|
pass
|
|
|
|
|
|
@user.command("files")
|
|
@click.option("--name", envvar="OTP_NAME", required=True)
|
|
@click.option("--realm", envvar="OTP_REALM", required=True)
|
|
@click.option("--seed", envvar="OTP_SEED", required=True)
|
|
def list_files(name, realm, seed):
|
|
data = OTPCredentials(name, realm, seed).get_json()
|
|
r = requests.get(
|
|
join_path(env_vars.get("UCLOUD_API_SERVER"), "user", "files"), json=data
|
|
)
|
|
print(load_dump_pretty(r.content))
|
|
|
|
|
|
@user.command("vms")
|
|
@click.option("--name", envvar="OTP_NAME", required=True)
|
|
@click.option("--realm", envvar="OTP_REALM", required=True)
|
|
@click.option("--seed", envvar="OTP_SEED", required=True)
|
|
def list_vms(name, realm, seed):
|
|
data = OTPCredentials(name, realm, seed).get_json()
|
|
r = requests.get(
|
|
join_path(env_vars.get("UCLOUD_API_SERVER"), "user", "vms"), json=data
|
|
)
|
|
print(load_dump_pretty(r.content))
|
|
|
|
|
|
@user.command("network")
|
|
@click.option("--name", envvar="OTP_NAME", required=True)
|
|
@click.option("--realm", envvar="OTP_REALM", required=True)
|
|
@click.option("--seed", envvar="OTP_SEED", required=True)
|
|
def list_networks(name, realm, seed):
|
|
data = OTPCredentials(name, realm, seed).get_json()
|
|
r = requests.get(
|
|
join_path(env_vars.get("UCLOUD_API_SERVER"), "user", "networks"), json=data
|
|
)
|
|
print(load_dump_pretty(r.content))
|
|
|
|
|
|
@user.command("add-ssh")
|
|
@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("--key-name", required=True)
|
|
@click.option("--key", required=True)
|
|
def add_ssh(name, realm, seed, key_name, key):
|
|
otp = OTPCredentials(name, realm, seed)
|
|
data = {**otp.get_json(), "key_name": key_name, "key": key}
|
|
r = requests.post(
|
|
join_path(env_vars.get("UCLOUD_API_SERVER"), "user", "add-ssh"), json=data
|
|
)
|
|
print(load_dump_pretty(r.content))
|
|
|
|
|
|
@user.command("remove-ssh")
|
|
@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("--key-name", required=True)
|
|
def remove_ssh(name, realm, seed, key_name):
|
|
otp = OTPCredentials(name, realm, seed)
|
|
data = {**otp.get_json(), "key_name": key_name}
|
|
r = requests.get(
|
|
join_path(env_vars.get("UCLOUD_API_SERVER"), "user", "remove-ssh"), json=data
|
|
)
|
|
print(load_dump_pretty(r.content))
|
|
|
|
|
|
@user.command("get-ssh")
|
|
@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("--key-name", default="")
|
|
def get_ssh(name, realm, seed, key_name):
|
|
otp = OTPCredentials(name, realm, seed)
|
|
data = {**otp.get_json(), "key_name": key_name}
|
|
r = requests.get(
|
|
join_path(env_vars.get("UCLOUD_API_SERVER"), "user", "get-ssh"), json=data
|
|
)
|
|
print(load_dump_pretty(r.content))
|