refactoring done

This commit is contained in:
ahmadbilalkhalid 2019-12-31 19:54:08 +05:00
parent 280043659d
commit dc283251d9
13 changed files with 227 additions and 289 deletions

4
.gitignore vendored
View File

@ -2,7 +2,7 @@
.vscode
__pycache__
ucloud_cli.egg-info
uncloud_cli.egg-info
build/
dist/
dist/

View File

@ -1,27 +0,0 @@
#!/usr/bin/env python3
import click
import sys
from ucloud_cli.commands.vm import vm
from ucloud_cli.commands.user import user
from ucloud_cli.commands.host import host
from ucloud_cli.commands.image import image
from ucloud_cli.commands.network import network
from ucloud_cli.helper import exception_handler
@click.group()
def entry_point():
pass
if __name__ == "__main__":
sys.excepthook = exception_handler
entry_point.add_command(vm)
entry_point.add_command(user)
entry_point.add_command(image)
entry_point.add_command(host)
entry_point.add_command(network)
entry_point()

47
bin/uncloud-cli Executable file
View File

@ -0,0 +1,47 @@
#!/usr/bin/env python3
import click
import sys
import logging
from logging.handlers import SysLogHandler
from uncloud_cli.commands.vm import vm
from uncloud_cli.commands.user import user
from uncloud_cli.commands.host import host
from uncloud_cli.commands.image import image
from uncloud_cli.commands.network import network
from uncloud_cli.helper import exception_handler, NoTracebackStreamHandler
@click.group()
def entry_point():
pass
if __name__ == "__main__":
sys.excepthook = exception_handler
# Setting up root logger
logger = logging.getLogger()
logger.setLevel(logging.DEBUG)
syslog_handler = SysLogHandler(address='/dev/log')
syslog_handler.setLevel(logging.DEBUG)
syslog_formatter = logging.Formatter('%(pathname)s:%(lineno)d -- %(levelname)-8s %(message)s')
syslog_handler.setFormatter(syslog_formatter)
stream_handler = NoTracebackStreamHandler()
stream_handler.setLevel(logging.INFO)
stream_formatter = logging.Formatter('%(message)s')
stream_handler.setFormatter(stream_formatter)
logger.addHandler(syslog_handler)
logger.addHandler(stream_handler)
entry_point.add_command(vm)
entry_point.add_command(user)
entry_point.add_command(image)
entry_point.add_command(host)
entry_point.add_command(network)
entry_point()

View File

@ -1,4 +0,0 @@
OTP_NAME=replace_me
OTP_REALM=replace_me
OTP_SEED=replace_me
UCLOUD_API_SERVER=http://[::]:5000

View File

@ -1,14 +1,12 @@
import os
from setuptools import setup, find_packages
with open("README.md", "r") as fh:
long_description = fh.read()
setup(name='ucloud_cli',
setup(name='uncloud_cli',
version='0.1',
description='A utility to interact with ucloud server.',
url='https://code.ungleich.ch/ucloud/ucloud-cli',
description='A utility to interact with uncloud server.',
url='https://code.ungleich.ch/uncloud/uncloud-cli',
long_description=long_description,
long_description_content_type='text/markdown',
classifiers=[
@ -26,6 +24,5 @@ setup(name='ucloud_cli',
'pyotp',
'click'
],
scripts=['bin/ucloud-cli'],
data_files=[(os.path.expanduser('~/ucloud/'), ['conf/ucloud-cli.conf'])],
scripts=['bin/uncloud-cli'],
zip_safe=False)

View File

@ -1,16 +1,12 @@
import json
import binascii
import click
import requests
from os.path import join as join_path
from pyotp import TOTP
class OTPCredentials:
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):
return {"name": self.name, "realm": self.realm, "token": TOTP(self.seed).now()}
from uncloud_cli.config import config, config_file
def load_dump_pretty(content):
@ -18,3 +14,54 @@ def load_dump_pretty(content):
content = content.decode("utf-8")
parsed = json.loads(content)
return json.dumps(parsed, indent=4, sort_keys=True)
def make_request(*args, data=None, request_method=requests.post):
r = request_method(
join_path(config['client']['api_server'], *args), json=data
)
print(load_dump_pretty(r.content))
def get_token(ctx, param, value):
if value is not None:
try:
token = TOTP(value).now()
except binascii.Error:
raise click.BadParameter('Please enter the correct seed in {}'.format(config_file))
else:
param.name = 'token'
return token
def add_otp_options(f):
options = [
click.option(
"--name", required=True, default=config['client']['name'],
show_default='name mentioned in {}'.format(config_file)
),
click.option(
"--realm", required=True, default=config['client']['realm'],
show_default='realm mentioned in {}'.format(config_file)
),
click.option(
"--seed", required=True, default=config['client']['seed'],
callback=get_token, show_default='seed mentioned in {}'.format(config_file)
)
]
for opt in reversed(options):
f = opt(f)
return f
def add_vm_options(f):
options = [
click.option('--vm-name', required=True),
click.option('--action', required=True, default=f.__name__)
]
for opt in reversed(options):
f = opt(f)
return f

View File

@ -1,9 +1,7 @@
import click
import requests
from .helper import OTPCredentials, load_dump_pretty
from uncloud_cli.config import env_vars
from os.path import join as join_path
from .helper import add_otp_options, make_request
@click.group()
@ -11,28 +9,23 @@ def host():
pass
@host.command("create")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@click.option("--hostname", required=True)
@click.option("--cpu", required=True, type=int)
@click.option("--ram", required=True)
@click.option("--os-ssd", required=True)
@click.option("--hdd", default=list(), multiple=True)
def create(name, realm, seed, hostname, cpu, ram, os_ssd, hdd):
data = {
**OTPCredentials(name, realm, seed).get_json(),
"hostname": hostname,
"specs": {"cpu": cpu, "ram": ram, "os-ssd": os_ssd, "hdd": hdd},
@host.command('create')
@add_otp_options
@click.option('--hostname', required=True)
@click.option('--cpu', required=True, type=int)
@click.option('--ram', required=True)
@click.option('--os-ssd', required=True)
@click.option('--hdd', default=list(), multiple=True)
def create(**kwargs):
kwargs['specs'] = {
'cpu': kwargs.pop('cpu'),
'ram': kwargs.pop('ram'),
'os-ssd': kwargs.pop('os_ssd'),
'hdd': kwargs.pop('hdd')
}
r = requests.post(
join_path(env_vars.get("UCLOUD_API_SERVER"), "host", "create"), json=data
)
print(load_dump_pretty(r.content))
make_request('host', 'create', data=kwargs)
@host.command("list")
@host.command('list')
def list_host():
r = requests.get(join_path(env_vars.get("UCLOUD_API_SERVER"), "host", "list"))
print(load_dump_pretty(r.content))
make_request('host', 'list', request_method=requests.get)

View File

@ -1,10 +1,8 @@
from uncloud_cli.commands.helper import load_dump_pretty
from uncloud_cli.config import env_vars
from os.path import join as join_path
import click
import requests
from uncloud_cli.commands.helper import make_request
@click.group()
def image():
@ -15,17 +13,12 @@ def image():
@click.option("--public", is_flag=True)
def _list(public):
if public:
r = requests.get(join_path(env_vars.get("UCLOUD_API_SERVER"), "image", "list-public"))
print(load_dump_pretty(r.content))
make_request('image', 'list-public', request_method=requests.get)
@image.command("create-from-file")
@click.option("--name", required=True)
@click.option("--uuid", required=True)
@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(
join_path(env_vars.get("UCLOUD_API_SERVER"), "image", "create"), json=data
)
print(load_dump_pretty(r.content))
@click.option("--image-store-name", 'image_store', required=True)
def create_from_file(**kwargs):
make_request('image', 'create', data=kwargs)

View File

@ -1,6 +1,4 @@
from uncloud_cli.commands.helper import load_dump_pretty, OTPCredentials
from uncloud_cli.config import env_vars
from os.path import join as join_path
from uncloud_cli.commands.helper import add_otp_options, make_request
import click
import requests
@ -12,20 +10,9 @@ def network():
@network.command("create")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@add_otp_options
@click.option("--network-name", required=True)
@click.option("--network-type", required=True)
@click.option("--network-type", 'type', required=True)
@click.option("--user", required=True, type=bool, default=False)
def create(name, realm, seed, network_name, network_type, user):
data = {
**OTPCredentials(name, realm, seed).get_json(),
"network_name": network_name,
"type": network_type,
"user": user,
}
r = requests.post(
join_path(env_vars.get("UCLOUD_API_SERVER"), "network", "create"), json=data
)
print(load_dump_pretty(r.content))
def create(**kwargs):
make_request('network', 'create', data=kwargs)

View File

@ -1,9 +1,5 @@
from uncloud_cli.commands.helper import OTPCredentials, load_dump_pretty
from uncloud_cli.config import env_vars
from os.path import join as join_path
import click
import requests
from uncloud_cli.commands.helper import add_otp_options, make_request
@click.group()
@ -12,79 +8,40 @@ def user():
@user.command("files")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
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))
@add_otp_options
def list_files(**kwargs):
make_request('user', 'files', data=kwargs)
@user.command("vms")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
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))
@add_otp_options
def list_vms(**kwargs):
make_request('user', 'vms', data=kwargs)
@user.command("networks")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
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))
@add_otp_options
def list_networks(**kwargs):
make_request('user', 'network', data=kwargs)
@user.command("add-ssh")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@add_otp_options
@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))
def add_ssh(**kwargs):
make_request('user', 'add-ssh', data=kwargs)
@user.command("remove-ssh")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@add_otp_options
@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))
def remove_ssh(**kwargs):
make_request('user', 'remove-ssh', data=kwargs)
@user.command("get-ssh")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@add_otp_options
@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))
def get_ssh(**kwargs):
make_request('user', 'get-ssh', data=kwargs)

View File

@ -1,19 +1,7 @@
import click
import json
import requests
import subprocess as sp
from uncloud_cli.commands.helper import OTPCredentials, load_dump_pretty
from uncloud_cli.config import env_vars
from os.path import join as join_path
def vm_command(command, otp, vm_name, **kwargs):
data = {**otp.get_json(), "vm_name": vm_name, "action": command, **kwargs}
r = requests.post(
join_path(env_vars.get("UCLOUD_API_SERVER"), "vm", "action"), json=data
)
return r
from uncloud_cli.commands.helper import add_otp_options, make_request, add_vm_options
@click.group()
@ -21,121 +9,56 @@ def vm():
pass
@vm.command("create")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@click.option("--vm-name", required=True)
@click.option("--cpu", required=True, type=int)
@click.option("--ram", required=True)
@click.option("--os-ssd", required=True)
@click.option("--hdd", default=list(), multiple=True)
@click.option("--image", required=True)
@click.option("--network", default=list(), multiple=True)
def create(name, realm, seed, vm_name, cpu, ram, os_ssd, hdd, image, network):
data = {
**OTPCredentials(name, realm, seed).get_json(),
"vm_name": vm_name,
"specs": {"cpu": cpu, "ram": ram, "os-ssd": os_ssd, "hdd": hdd},
"network": network,
"image": image,
@vm.command('create')
@add_otp_options
@add_vm_options
@click.option('--cpu', required=True, type=int)
@click.option('--ram', required=True)
@click.option('--os-ssd', required=True)
@click.option('--hdd', default=list(), multiple=True)
@click.option('--image', required=True)
@click.option('--network', default=list(), multiple=True)
def create(**kwargs):
kwargs['specs'] = {
'cpu': kwargs.pop('cpu'),
'ram': kwargs.pop('ram'),
'os-ssd': kwargs.pop('os_ssd'),
'hdd': kwargs.pop('hdd')
}
r = requests.post(
join_path(env_vars.get("UCLOUD_API_SERVER"), "vm", "create"), json=data
)
print(load_dump_pretty(r.content))
make_request('vm', kwargs.pop('action'), data=kwargs)
@vm.command("start")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@click.option("--vm-name", required=True)
@click.option("--in_support_of")
def start(name, realm, seed, vm_name, in_support_of):
r = vm_command(
"start", OTPCredentials(name, realm, seed), vm_name, in_support_of=in_support_of
)
print(load_dump_pretty(r.content))
@vm.command('start')
@add_otp_options
@add_vm_options
def start(**kwargs):
make_request('vm', 'action', data=kwargs)
@vm.command("stop")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@click.option("--vm-name", required=True)
@click.option("--in_support_of")
def stop(name, realm, seed, vm_name, in_support_of):
r = vm_command(
"stop", OTPCredentials(name, realm, seed), vm_name, in_support_of=in_support_of
)
print(load_dump_pretty(r.content))
@vm.command('stop')
@add_otp_options
@add_vm_options
def stop(**kwargs):
make_request('vm', 'action', data=kwargs)
@vm.command("delete")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@click.option("--vm-name", required=True)
@click.option("--in_support_of")
def delete(name, realm, seed, vm_name, in_support_of):
r = vm_command(
"delete",
OTPCredentials(name, realm, seed),
vm_name,
in_support_of=in_support_of,
)
print(load_dump_pretty(r.content))
@vm.command('delete')
@add_otp_options
@add_vm_options
def delete(**kwargs):
make_request('vm', 'action', data=kwargs)
@vm.command("status")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@click.option("--vm-name", required=True)
@click.option("--in_support_of")
def status(name, realm, seed, vm_name, in_support_of):
otp = OTPCredentials(name, realm, seed)
data = {**otp.get_json(), "vm_name": vm_name, "in_support_of": in_support_of}
r = requests.get(join_path(env_vars.get("UCLOUD_API_SERVER"), "vm", "status"), json=data)
print(load_dump_pretty(r.content))
@vm.command('status')
@add_otp_options
@click.option('--vm-name', required=True)
def status(**kwargs):
make_request('vm', 'status', data=kwargs)
@vm.command("migrate")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@add_otp_options
@click.option("--vm-name", required=True)
@click.option("--destination", required=True)
@click.option("--in_support_of")
def vm_migration(name, realm, seed, vm_name, destination, in_support_of):
otp = OTPCredentials(name, realm, seed)
data = {
**otp.get_json(),
"vm_name": vm_name,
"destination": destination,
"in_support_of": in_support_of,
}
r = requests.post(
join_path(env_vars.get("UCLOUD_API_SERVER"), "vm", "migrate"), json=data
)
print(load_dump_pretty(r.content))
@vm.command("ssh")
@click.option("--name", required=True, default=env_vars.get("OTP_NAME"))
@click.option("--realm", required=True, default=env_vars.get("OTP_REALM"))
@click.option("--seed", required=True, default=env_vars.get("OTP_SEED"))
@click.option("--vm-name", required=True)
@click.option("--in_support_of")
def ssh(name, realm, seed, vm_name, in_support_of):
otp = OTPCredentials(name, realm, seed)
data = {**otp.get_json(), "vm_name": vm_name, "in_support_of": in_support_of}
r = requests.get(join_path(env_vars.get("UCLOUD_API_SERVER"), "vm", "status"), json=data)
try:
_json = json.loads(r.content)
sp.run(['ssh', '-o', 'ConnectTimeout=10',
'root@{}'.format(_json['ip'][0])])
except Exception as err:
print("Some error occurred while accessing VM."
"Make sure VM is running", err)
def vm_migration(**kwargs):
make_request('vm', 'migrate', data=kwargs)

View File

@ -1,9 +1,10 @@
import sys
from os.path import expanduser
from decouple import Config, RepositoryEnv
import configparser
import os
config_file = os.path.expanduser('~/uncloud/uncloud.conf')
try:
env_vars = Config(RepositoryEnv(expanduser("~/uncloud/uncloud-cli.conf")))
config = configparser.ConfigParser()
config.read(config_file)
except Exception as err:
print(err)
sys.exit(1)
sys.exit(err)

View File

@ -1,5 +1,7 @@
import os
import sys
import logging
import click
def exception_handler(exception_type, exception, traceback):
@ -7,3 +9,25 @@ def exception_handler(exception_type, exception, traceback):
sys.__excepthook__(exception_type, exception, traceback)
else:
print("%s: %s" % (exception_type.__name__, exception))
class NoTracebackStreamHandler(logging.StreamHandler):
def handle(self, record):
info, cache = record.exc_info, record.exc_text
record.exc_info, record.exc_text = None, None
if record.levelname in ["WARNING", "WARN"]:
click.echo(click.style('', fg='yellow', bold=True, reset=False), nl=False)
elif record.levelname == "ERROR":
click.echo(click.style('', fg='red', bold=True, reset=False), nl=False)
elif record.levelname == "INFO":
click.echo(click.style('', fg='green', bold=True, reset=False), nl=False)
elif record.levelname == "CRITICAL":
click.echo(click.style('', fg='cyan', bold=True, reset=False), nl=False)
try:
super().handle(record)
finally:
record.exc_info = info
record.exc_text = cache
click.echo(click.style('', 'reset'), nl=False)