Using click instead of argparse in uncloud script

uncloud/cli/commands/helper.py

@@ -0,0 +1,62 @@
+import json
+import binascii
+import click
+import requests
+
+from os.path import join as join_path
+
+from pyotp import TOTP
+from uncloud.settings import settings
+
+
+def load_dump_pretty(content):
+    if isinstance(content, bytes):
+        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(settings['client']['api_server'], *args), json=data
+    )
+    try:
+        print(load_dump_pretty(r.content))
+    except Exception:
+        print('Error occurred while getting output from api server.')
+
+
+def get_token(_, param, value):
+    if value is not None:
+        try:
+            token = TOTP(value).now()
+        except binascii.Error:
+            raise click.BadParameter('')
+        else:
+            param.name = 'token'
+            return token
+
+
+def add_otp_options(f):
+    options = [
+        click.option('--name', show_default='name mentioned in config file.', prompt=True),
+        click.option('--realm', show_default='realm mentioned in config file.', prompt=True),
+        click.option('--seed', callback=get_token, show_default='seed mentioned in config file',
+                     prompt=True)
+    ]
+
+    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

uncloud/cli/commands/host.py

@@ -0,0 +1,31 @@
+import click
+import requests
+
+from .helper import add_otp_options, make_request
+
+
+@click.group()
+def host():
+    pass
+
+
+@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')
+    }
+    make_request('host', 'create', data=kwargs)
+
+
+@host.command('list')
+def list_host():
+    make_request('host', 'list', request_method=requests.get)

uncloud/cli/commands/image.py

@@ -0,0 +1,24 @@
+import click
+import requests
+
+from .helper import make_request
+
+
+@click.group()
+def image():
+    pass
+
+
+@image.command('list')
+@click.option('--public', is_flag=True)
+def _list(public):
+    if public:
+        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', 'image_store', required=True)
+def create_from_file(**kwargs):
+    make_request('image', 'create', data=kwargs)

uncloud/cli/commands/network.py

@@ -0,0 +1,17 @@
+import click
+
+from .helper import add_otp_options, make_request
+
+
+@click.group()
+def network():
+    pass
+
+
+@network.command('create')
+@add_otp_options
+@click.option('--network-name', required=True)
+@click.option('--network-type', 'type', required=True)
+@click.option('--user', required=True, type=bool, default=False)
+def create(**kwargs):
+    make_request('network', 'create', data=kwargs)

uncloud/cli/commands/user.py

@@ -0,0 +1,48 @@
+import click
+
+from .helper import add_otp_options, make_request
+
+
+@click.group()
+def user():
+    pass
+
+
+@user.command('files')
+@add_otp_options
+def list_files(**kwargs):
+    make_request('user', 'files', data=kwargs)
+
+
+@user.command('vms')
+@add_otp_options
+def list_vms(**kwargs):
+    make_request('user', 'vms', data=kwargs)
+
+
+@user.command('networks')
+@add_otp_options
+def list_networks(**kwargs):
+    make_request('user', 'network', data=kwargs)
+
+
+@user.command('add-ssh')
+@add_otp_options
+@click.option('--key-name', required=True)
+@click.option('--key', required=True)
+def add_ssh(**kwargs):
+    make_request('user', 'add-ssh', data=kwargs)
+
+
+@user.command('remove-ssh')
+@add_otp_options
+@click.option('--key-name', required=True)
+def remove_ssh(**kwargs):
+    make_request('user', 'remove-ssh', data=kwargs)
+
+
+@user.command('get-ssh')
+@add_otp_options
+@click.option('--key-name', default='')
+def get_ssh(**kwargs):
+    make_request('user', 'get-ssh', data=kwargs)

uncloud/cli/commands/vm.py

@@ -0,0 +1,63 @@
+import click
+
+from .helper import add_otp_options, make_request, add_vm_options
+
+
+@click.group()
+def vm():
+    pass
+
+
+@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')
+    }
+    make_request('vm', kwargs.pop('action'), data=kwargs)
+
+
+@vm.command('start')
+@add_otp_options
+@add_vm_options
+def start(**kwargs):
+    make_request('vm', 'action', data=kwargs)
+
+
+@vm.command('stop')
+@add_otp_options
+@add_vm_options
+def stop(**kwargs):
+    make_request('vm', 'action', data=kwargs)
+
+
+@vm.command('delete')
+@add_otp_options
+@add_vm_options
+def delete(**kwargs):
+    make_request('vm', 'action', data=kwargs)
+
+
+@vm.command('status')
+@add_otp_options
+@click.option('--vm-name', required=True)
+def status(**kwargs):
+    make_request('vm', 'status', data=kwargs)
+
+
+@vm.command('migrate')
+@add_otp_options
+@click.option('--vm-name', required=True)
+@click.option('--destination', required=True)
+def vm_migration(**kwargs):
+    make_request('vm', 'migrate', data=kwargs)

uncloud/cli/main.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+
+import click
+
+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
+
+
+@click.group()
+def cli():
+    pass
+
+
+cli.add_command(vm)
+cli.add_command(user)
+cli.add_command(image)
+cli.add_command(host)
+cli.add_command(network)
+
+if __name__ == '__main__':
+    cli()