Synnefo » ./kamaki

#!/usr/bin/env python

# Copyright 2011 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

"""

To add a command create a new class and add a 'command' decorator. The class

must have a 'main' method which will contain the code to be executed.

Optionally a command can implement an 'update_parser' class method in order

to add command line arguments, or modify the OptionParser in any way.

The name of the class is important and it will determine the name and grouping

of the command. This behavior can be overriden with the 'group' and 'name'

decorator arguments:

    @command(api='compute')

    class server_list(object):

        # This command will be named 'list' under group 'server'

        ...

    @command(api='compute', name='ls')

    class server_list(object):

        # This command will be named 'ls' under group 'server'

        ...

The docstring of a command class will be used as the command description in

help messages, unless overriden with the 'description' decorator argument.

The syntax of a command will be generated dynamically based on the signature

of the 'main' method, unless overriden with the 'syntax' decorator argument:

    def main(self, server_id, network=None):

        # This syntax of this command will be: '<server id> [network]'

        ...

The order of commands is important, it will be preserved in the help output.

"""

import inspect

import logging

import os

from base64 import b64encode

from grp import getgrgid

from optparse import OptionParser

from os.path import abspath, basename, exists

from pwd import getpwuid

from sys import argv, exit, stdout

from kamaki import clients

from kamaki.config import Config, ConfigError

from kamaki.utils import OrderedDict, print_addresses, print_dict, print_items

log = logging.getLogger('kamaki')

_commands = OrderedDict()

def command(api=None, group=None, name=None, description=None, syntax=None):

    """Class decorator that registers a class as a CLI command."""

    def decorator(cls):

        grp, sep, cmd = cls.__name__.partition('_')

        if not sep:

            grp, cmd = None, cls.__name__

        cls.api = api

        cls.group = group or grp

        cls.name = name or cmd

        cls.description = description or cls.__doc__

        cls.syntax = syntax

        if cls.syntax is None:

            # Generate a syntax string based on main's arguments

            spec = inspect.getargspec(cls.main.im_func)

            args = spec.args[1:]

            n = len(args) - len(spec.defaults or ())

            required = ' '.join('<%s>' % x.replace('_', ' ') for x in args[:n])

            optional = ' '.join('[%s]' % x.replace('_', ' ') for x in args[n:])

            cls.syntax = ' '.join(x for x in [required, optional] if x)

            if spec.varargs:

                cls.syntax += ' <%s ...>' % spec.varargs

        if cls.group not in _commands:

            _commands[cls.group] = OrderedDict()

        _commands[cls.group][cls.name] = cls

        return cls

    return decorator

@command()

class config_list(object):

    """list configuration options"""

    @classmethod

    def update_parser(cls, parser):

        parser.add_option('-a', dest='all', action='store_true',

                default=False, help='include empty values')

    def main(self):

        for key, val in sorted(self.config.items()):

            if not val and not self.options.all:

                continue

            print '%s=%s' % (key, val)

@command()

class config_get(object):

    """get a configuration option"""

    def main(self, key):

        val = self.config.get(key)

        if val is not None:

            print val

@command()

class config_set(object):

    """set a configuration option"""

    def main(self, key, val):

        self.config.set(key, val)

@command()

class config_del(object):

    """delete a configuration option"""

    def main(self, key):

        self.config.delete(key)

@command(api='compute')

class server_list(object):

    """list servers"""

    @classmethod

    def update_parser(cls, parser):

        parser.add_option('-l', dest='detail', action='store_true',

                default=False, help='show detailed output')

    def main(self):

        servers = self.client.list_servers(self.options.detail)

        print_items(servers)

@command(api='compute')

class server_info(object):

    """get server details"""

    def main(self, server_id):

        server = self.client.get_server_details(int(server_id))

        print_dict(server)

@command(api='compute')

class server_create(object):

    """create server"""

    @classmethod

    def update_parser(cls, parser):

        parser.add_option('--personality', dest='personalities',

                action='append', default=[],

                metavar='PATH[,SERVER PATH[,OWNER[,GROUP,[MODE]]]]',

                help='add a personality file')

        parser.epilog = "If missing, optional personality values will be " \

                "filled based on the file at PATH if missing."

    def main(self, name, flavor_id, image_id):

        personalities = []

        for personality in self.options.personalities:

            p = personality.split(',')

            p.extend([None] * (5 - len(p)))     # Fill missing fields with None

            path = p[0]

            if not path:

                log.error("Invalid personality argument '%s'", p)

                return 1

            if not exists(path):

                log.error("File %s does not exist", path)

                return 1

            with open(path) as f:

                contents = b64encode(f.read())

            st = os.stat(path)

            personalities.append({

                'path': p[1] or abspath(path),

                'owner': p[2] or getpwuid(st.st_uid).pw_name,

                'group': p[3] or getgrgid(st.st_gid).gr_name,

                'mode': int(p[4]) if p[4] else 0x7777 & st.st_mode,

                'contents': contents})

        reply = self.client.create_server(name, int(flavor_id), image_id,

                personalities)

        print_dict(reply)

@command(api='compute')

class server_rename(object):

    """update server name"""

    def main(self, server_id, new_name):

        self.client.update_server_name(int(server_id), new_name)

@command(api='compute')

class server_delete(object):

    """delete server"""

    def main(self, server_id):

        self.client.delete_server(int(server_id))

@command(api='compute')

class server_reboot(object):

    """reboot server"""

    @classmethod

    def update_parser(cls, parser):

        parser.add_option('-f', dest='hard', action='store_true',

                default=False, help='perform a hard reboot')

    def main(self, server_id):

        self.client.reboot_server(int(server_id), self.options.hard)

@command(api='cyclades')

class server_start(object):

    """start server"""

    def main(self, server_id):

        self.client.start_server(int(server_id))

@command(api='cyclades')

class server_shutdown(object):

    """shutdown server"""

    def main(self, server_id):

        self.client.shutdown_server(int(server_id))

@command(api='cyclades')

class server_console(object):

    """get a VNC console"""

    def main(self, server_id):

        reply = self.client.get_server_console(int(server_id))

        print_dict(reply)

@command(api='cyclades')

class server_firewall(object):

    """set the firewall profile"""

    def main(self, server_id, profile):

        self.client.set_firewall_profile(int(server_id), profile)

@command(api='cyclades')

class server_addr(object):

    """list server addresses"""

    def main(self, server_id, network=None):

        reply = self.client.list_server_addresses(int(server_id), network)

        margin = max(len(x['name']) for x in reply)

        print_addresses(reply, margin)

@command(api='compute')

class server_meta(object):

    """get server metadata"""

    def main(self, server_id, key=None):

        reply = self.client.get_server_metadata(int(server_id), key)

        print_dict(reply)

@command(api='compute')

class server_addmeta(object):

    """add server metadata"""

    def main(self, server_id, key, val):

        reply = self.client.create_server_metadata(int(server_id), key, val)

        print_dict(reply)

@command(api='compute')

class server_setmeta(object):

    """update server metadata"""

    def main(self, server_id, key, val):

        metadata = {key: val}

        reply = self.client.update_server_metadata(int(server_id), **metadata)

        print_dict(reply)

@command(api='compute')

class server_delmeta(object):

    """delete server metadata"""

    def main(self, server_id, key):

        self.client.delete_server_metadata(int(server_id), key)

@command(api='cyclades')

class server_stats(object):

    """get server statistics"""

    def main(self, server_id):

        reply = self.client.get_server_stats(int(server_id))

        print_dict(reply, exclude=('serverRef',))

@command(api='compute')

class flavor_list(object):

    """list flavors"""

    @classmethod

    def update_parser(cls, parser):

        parser.add_option('-l', dest='detail', action='store_true',

                default=False, help='show detailed output')

    def main(self):

        flavors = self.client.list_flavors(self.options.detail)

        print_items(flavors)

@command(api='compute')

class flavor_info(object):

    """get flavor details"""

    def main(self, flavor_id):

        flavor = self.client.get_flavor_details(int(flavor_id))

        print_dict(flavor)

@command(api='compute')

class image_list(object):

    """list images"""

    @classmethod

    def update_parser(cls, parser):

        parser.add_option('-l', dest='detail', action='store_true',

                default=False, help='show detailed output')

    def main(self):

        images = self.client.list_images(self.options.detail)

        print_items(images)

@command(api='compute')

class image_info(object):

    """get image details"""

    def main(self, image_id):

        image = self.client.get_image_details(image_id)

        print_dict(image)

@command(api='compute')

class image_delete(object):

    """delete image"""

    def main(self, image_id):

        self.client.delete_image(image_id)

@command(api='compute')

class image_meta(object):

    """get image metadata"""

    def main(self, image_id, key=None):

        reply = self.client.get_image_metadata(image_id, key)

        print_dict(reply)

@command(api='compute')

class image_addmeta(object):

    """add image metadata"""

    def main(self, image_id, key, val):

        reply = self.client.create_image_metadata(image_id, key, val)

        print_dict(reply)

@command(api='compute')

class image_setmeta(object):

    """update image metadata"""

    def main(self, image_id, key, val):

        metadata = {key: val}

        reply = self.client.update_image_metadata(image_id, **metadata)

        print_dict(reply)

@command(api='compute')

class image_delmeta(object):

    """delete image metadata"""

    def main(self, image_id, key):

        self.client.delete_image_metadata(image_id, key)

@command(api='cyclades')

class network_list(object):

    """list networks"""

    @classmethod

    def update_parser(cls, parser):

        parser.add_option('-l', dest='detail', action='store_true',

                default=False, help='show detailed output')

    def main(self):

        networks = self.client.list_networks(self.options.detail)

        print_items(networks)

@command(api='cyclades')

class network_create(object):

    """create a network"""

    def main(self, name):

        reply = self.client.create_network(name)

        print_dict(reply)

@command(api='cyclades')

class network_info(object):

    """get network details"""

    def main(self, network_id):

        network = self.client.get_network_details(network_id)

        print_dict(network)

@command(api='cyclades')

class network_rename(object):

    """update network name"""

    def main(self, network_id, new_name):

        self.client.update_network_name(network_id, new_name)

@command(api='cyclades')

class network_delete(object):

    """delete a network"""

    def main(self, network_id):

        self.client.delete_network(network_id)

@command(api='cyclades')

class network_connect(object):

    """connect a server to a network"""

    def main(self, server_id, network_id):

        self.client.connect_server(server_id, network_id)

@command(api='cyclades')

class network_disconnect(object):

    """disconnect a server from a network"""

    def main(self, server_id, network_id):

        self.client.disconnect_server(server_id, network_id)

@command(api='image')

class glance_list(object):

    """list images"""

    @classmethod

    def update_parser(cls, parser):

        parser.add_option('-l', dest='detail', action='store_true',

                default=False, help='show detailed output')

        parser.add_option('--container-format', dest='container_format',

                metavar='FORMAT', help='filter by container format')

        parser.add_option('--disk-format', dest='disk_format',

                metavar='FORMAT', help='filter by disk format')

        parser.add_option('--name', dest='name', metavar='NAME',

                help='filter by name')

        parser.add_option('--size-min', dest='size_min', metavar='BYTES',

                help='filter by minimum size')

        parser.add_option('--size-max', dest='size_max', metavar='BYTES',

                help='filter by maximum size')

        parser.add_option('--status', dest='status', metavar='STATUS',

                help='filter by status')

        parser.add_option('--order', dest='order', metavar='FIELD',

                help='order by FIELD (use a - prefix to reverse order)')

    def main(self):

        filters = {}

        for filter in ('container_format', 'disk_format', 'name', 'size_min',

                       'size_max', 'status'):

            val = getattr(self.options, filter, None)

            if val is not None:

                filters[filter] = val

        order = self.options.order or ''

        images = self.client.list_public(self.options.detail, filters=filters,

                                         order=order)

        print_items(images, title=('name',))

@command(api='image')

class glance_meta(object):

    """get image metadata"""

    def main(self, image_id):

        image = self.client.get_meta(image_id)

        print_dict(image)

@command(api='image')

class glance_register(object):

    """register an image"""

    @classmethod

    def update_parser(cls, parser):

        parser.add_option('--checksum', dest='checksum', metavar='CHECKSUM',

                help='set image checksum')

        parser.add_option('--container-format', dest='container_format',

                metavar='FORMAT', help='set container format')

        parser.add_option('--disk-format', dest='disk_format',

                metavar='FORMAT', help='set disk format')

        parser.add_option('--id', dest='id',

                metavar='ID', help='set image ID')

        parser.add_option('--owner', dest='owner',

                metavar='USER', help='set image owner (admin only)')

        parser.add_option('--property', dest='properties', action='append',

                metavar='KEY=VAL',

                help='add a property (can be used multiple times)')

        parser.add_option('--public', dest='is_public', action='store_true',

                help='mark image as public')

        parser.add_option('--size', dest='size', metavar='SIZE',

                help='set image size')

    def main(self, name, location):

        params = {}

        for key in ('checksum', 'container_format', 'disk_format', 'id',

                    'owner', 'is_public', 'size'):

            val = getattr(self.options, key)

            if val is not None:

                params[key] = val

        properties = {}

        for property in self.options.properties or []:

            key, sep, val = property.partition('=')

            if not sep:

                log.error("Invalid property '%s'", property)

                return 1

            properties[key.strip()] = val.strip()

        self.client.register(name, location, params, properties)

@command(api='image')

class glance_members(object):

    """get image members"""

    def main(self, image_id):

        members = self.client.list_members(image_id)

        for member in members:

            print member['member_id']

@command(api='image')

class glance_shared(object):

    """list shared images"""

    def main(self, member):

        images = self.client.list_shared(member)

        for image in images:

            print image['image_id']

@command(api='image')

class glance_addmember(object):

    """add a member to an image"""

    def main(self, image_id, member):

        self.client.add_member(image_id, member)

@command(api='image')

class glance_delmember(object):

    """remove a member from an image"""

    def main(self, image_id, member):

        self.client.remove_member(image_id, member)

@command(api='image')

class glance_setmembers(object):

    """set the members of an image"""

    def main(self, image_id, *member):

        self.client.set_members(image_id, member)

class store_command(object):

    """base class for all store_* commands"""

    @classmethod

    def update_parser(cls, parser):

        parser.add_option('--account', dest='account', metavar='NAME',

                help='use account NAME')

        parser.add_option('--container', dest='container', metavar='NAME',

                help='use container NAME')

    def main(self):

        self.config.override('storage_account', self.options.account)

        self.config.override('storage_container', self.options.container)

        # Use the more efficient Pithos client if available

        if 'pithos' in self.config.get('apis').split():

            self.client = clients.PithosClient(self.config)

@command(api='storage')

class store_create(object):

    """create a container"""

    @classmethod

    def update_parser(cls, parser):

        parser.add_option('--account', dest='account', metavar='ACCOUNT',

                help='use account ACCOUNT')

    def main(self, container):

        self.config.override('storage_account', self.options.account)

        self.client.create_container(container)

@command(api='storage')

class store_container(store_command):

    """get container info"""

    def main(self):

        store_command.main(self)

        reply = self.client.get_container_meta()

        print_dict(reply)

@command(api='storage')

class store_upload(store_command):

    """upload a file"""

    def main(self, path, remote_path=None):

        store_command.main(self)

        if remote_path is None:

            remote_path = basename(path)

        with open(path) as f:

            self.client.create_object(remote_path, f)

@command(api='storage')

class store_download(store_command):

    """download a file"""

    def main(self, remote_path, local_path):

        store_command.main(self)

        f = self.client.get_object(remote_path)

        out = open(local_path, 'w') if local_path != '-' else stdout

        block = 4096

        data = f.read(block)

        while data:

            out.write(data)

            data = f.read(block)

@command(api='storage')

class store_delete(store_command):

    """delete a file"""

    def main(self, path):

        store_command.main(self)

        self.client.delete_object(path)

def print_groups(groups):

    print

    print 'Groups:'

    for group in groups:

        print '  %s' % group

def print_commands(group, commands):

    print

    print 'Commands:'

    for name, cls in _commands[group].items():

        if name in commands:

            print '  %s %s' % (name.ljust(10), cls.description)

def main():

    ch = logging.StreamHandler()

    ch.setFormatter(logging.Formatter('%(message)s'))

    log.addHandler(ch)

    parser = OptionParser(add_help_option=False)

    parser.usage = '%prog <group> <command> [options]'

    parser.add_option('--help', dest='help', action='store_true',

            default=False, help='show this help message and exit')

    parser.add_option('-v', dest='verbose', action='store_true', default=False,

            help='use verbose output')

    parser.add_option('-d', dest='debug', action='store_true', default=False,

            help='use debug output')

    # Do a preliminary parsing, ignore any errors since we will print help

    # anyway if we don't reach the main parsing.

    _error = parser.error

    parser.error = lambda msg: None

    options, args = parser.parse_args(argv)

    parser.error = _error

    if options.debug:

        log.setLevel(logging.DEBUG)

    elif options.verbose:

        log.setLevel(logging.INFO)

    else:

        log.setLevel(logging.WARNING)

    try:

        config = Config()

    except ConfigError, e:

        log.error('%s', e.args[0])

        exit(1)

    apis = config.get('apis').split()

    # Find available groups based on the given APIs

    available_groups = []

    for group, group_commands in _commands.items():

        for name, cls in group_commands.items():

            if cls.api is None or cls.api in apis:

                available_groups.append(group)

                break

    if len(args) < 2:

        parser.print_help()

        print_groups(available_groups)

        exit(0)

    group = args[1]

    if group not in available_groups:

        parser.print_help()

        print_groups(available_groups)

        exit(1)

    # Find available commands based on the given APIs

    available_commands = []

    for name, cls in _commands[group].items():

        if cls.api is None or cls.api in apis:

            available_commands.append(name)

            continue

    parser.usage = '%%prog %s <command> [options]' % group

    if len(args) < 3:

        parser.print_help()

        print_commands(group, available_commands)

        exit(0)

    name = args[2]

    if name not in available_commands:

        parser.print_help()

        print_commands(group, available_commands)

        exit(1)

    cls = _commands[group][name]

    syntax = '%s [options]' % cls.syntax if cls.syntax else '[options]'

    parser.usage = '%%prog %s %s %s' % (group, name, syntax)

    parser.epilog = ''

    if hasattr(cls, 'update_parser'):

        cls.update_parser(parser)

    options, args = parser.parse_args(argv)

    if options.help:

        parser.print_help()

        exit(0)

    cmd = cls()

    cmd.config = config

    cmd.options = options

    if cmd.api:

        client_name = cmd.api.capitalize() + 'Client'

        client = getattr(clients, client_name, None)

        if client:

            cmd.client = client(config)

    try:

        ret = cmd.main(*args[3:])

        exit(ret)

    except TypeError as e:

        if e.args and e.args[0].startswith('main()'):

            parser.print_help()

            exit(1)

        else:

            raise

    except clients.ClientError, err:

        log.error('%s', err.message)

        log.info('%s', err.details)

        exit(2)

if __name__ == '__main__':

    main()