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))