uncloud-cli/commands/host.py

import json

from commands.helper import OTPCredentials, load_dump_pretty
from decouple import config

import click
import requests


@click.group()
def host():
    pass


@host.command("add")
@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("--hostname", required=True)
def add_host(name, realm, seed, specs, hostname):
    with open(specs, "r") as specs_f:
        specs = json.loads(specs_f.read())
        data = {
            **OTPCredentials(name, realm, seed).get_json(),
            "specs": specs,
            "hostname": hostname,
        }
        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("{}/host/list".format(config('UCLOUD_API_SERVER')))
    print(load_dump_pretty(r.content))