Synnefo » ./kamaki

#!/usr/bin/env python

# Copyright 2011-2012 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.

from __future__ import print_function

import gevent.monkey

#Monkey-patch everything for gevent early on

gevent.monkey.patch_all()

import inspect

import logging

import sys

from argparse import ArgumentParser

from base64 import b64encode

from os.path import abspath, basename, exists

from sys import exit, stdout, stderr

try:

    from collections import OrderedDict

except ImportError:

    from ordereddict import OrderedDict

from colors import magenta, red, yellow, bold

#from progress.bar import IncrementalBar

#from requests.exceptions import ConnectionError

from . import clients

from .config import Config

#from .utils import print_list, print_dict, print_items, format_size

_commands = OrderedDict()

GROUPS = {}

CLI_LOCATIONS = ['', 'kamaki', 'kamaki.commands']

class CLIError(Exception):

    def __init__(self, message, status=0, details='', importance=0):

        """importance is set by the raiser

        0 is the lowest possible importance

        Suggested values: 0, 1, 2, 3

        """

        super(CLIError, self).__init__(message, status, details)

        self.message = message

        self.status = status

        self.details = details

        self.importance = importance

    def __unicode__(self):

        return unicode(self.message)

def command(group=None, name=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

        short_description, sep, long_description = cls.__doc__.partition('\n')

        cls.description = short_description

        cls.long_description = long_description or short_description

        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('____', '[:').replace('___', ':').replace('__',']').replace('_', ' ') for x in args[:n])

            optional = ' '.join('[%s]' % x.replace('____', '[:').replace('___', ':').replace('__', ']').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

def set_api_description(api, description):

    """Method to be called by api CLIs

    Each CLI can set more than one api descriptions"""

    GROUPS[api] = description

def main():

    def print_groups():

        print('\nGroups:')

        for group in _commands:

            description = GROUPS.get(group, '')

            print(' ', group.ljust(12), description)

    def print_commands(group):

        description = GROUPS.get(group, '')

        if description:

            print('\n' + description)

        print('\nCommands:')

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

            print(' ', name.ljust(14), cls.description)

    def manage_logging_handlers(args):

        """This is mostly to handle logging for clients package"""

        def add_handler(name, level, prefix=''):

            h = logging.StreamHandler()

            fmt = logging.Formatter(prefix + '%(message)s')

            h.setFormatter(fmt)

            logger = logging.getLogger(name)

            logger.addHandler(h)

            logger.setLevel(level)

        if args.silent:

            add_handler('', logging.CRITICAL)

        elif args.debug:

            add_handler('requests', logging.INFO, prefix='* ')

            add_handler('clients.send', logging.DEBUG, prefix='> ')

            add_handler('clients.recv', logging.DEBUG, prefix='< ')

        elif args.verbose:

            add_handler('requests', logging.INFO, prefix='* ')

            add_handler('clients.send', logging.INFO, prefix='> ')

            add_handler('clients.recv', logging.INFO, prefix='< ')

        elif args.include:

            add_handler('clients.recv', logging.INFO)

        else:

            add_handler('', logging.WARNING)

    def load_groups(config):

        """load groups and import CLIs and Modules"""

        loaded_modules = {}

        for api in config.apis():

            api_cli = config.get(api, 'cli')

            if None == api_cli or len(api_cli)==0:

                print('Warnig: No Command Line Interface "%s" given for API "%s"'%(api_cli, api))

                print('\t(cli option in config file)')

                continue

            if not loaded_modules.has_key(api_cli):

                loaded_modules[api_cli] = False

                for location in CLI_LOCATIONS:

                    location += api_cli if location == '' else '.%s'%api_cli

                    try:

                        __import__(location)

                        loaded_modules[api_cli] = True

                        break

                    except ImportError:

                        pass

                if not loaded_modules[api_cli]:

                    print('Warning: failed to load Command Line Interface "%s" for API "%s"'%(api_cli, api))

                    print('\t(No suitable cli in known paths)')

                    continue

            if not GROUPS.has_key(api):

                GROUPS[api] = 'No description (interface: %s)'%api_cli

    def init_parser(exe):

        parser = ArgumentParser(add_help=False)

        parser.prog = '%s <group> <command>' % exe

        parser.add_argument('-h', '--help', dest='help', action='store_true',

                          default=False,

                          help="Show this help message and exit")

        parser.add_argument('--config', dest='config', metavar='PATH',

                          help="Specify the path to the configuration file")

        parser.add_argument('-d', '--debug', dest='debug', action='store_true',

                          default=False,

                          help="Include debug output")

        parser.add_argument('-i', '--include', dest='include', action='store_true',

                          default=False,

                          help="Include protocol headers in the output")

        parser.add_argument('-s', '--silent', dest='silent', action='store_true',

                          default=False,

                          help="Silent mode, don't output anything")

        parser.add_argument('-v', '--verbose', dest='verbose', action='store_true',

                          default=False,

                          help="Make the operation more talkative")

        parser.add_argument('-V', '--version', dest='version', action='store_true',

                          default=False,

                          help="Show version number and quit")

        parser.add_argument('-o', dest='options', action='append',

                          default=[], metavar="KEY=VAL",

                          help="Override a config value")

        return parser

    def find_term_in_args(arg_list, term_list):

        """find an arg_list term in term_list. All other terms up to found

        term are rearanged at the end of arg_list, preserving relative order

        """

        arg_tail = []

        while len(arg_list) > 0:

            group = arg_list.pop(0)

            if group not in term_list:

                arg_tail.append(group)

            else:

                arg_list += arg_tail

                return group

        return None

    """Main Code"""

    exe = basename(sys.argv[0])

    parser = init_parser(exe)

    args, argv = parser.parse_known_args()

    #print version

    if args.version:

        import kamaki

        print("kamaki %s" % kamaki.__version__)

        exit(0)

    config = Config(args.config) if args.config else Config()

    #load config options from command line

    for option in args.options:

        keypath, sep, val = option.partition('=')

        if not sep:

            print("Invalid option '%s'" % option)

            exit(1)

        section, sep, key = keypath.partition('.')

        if not sep:

            print("Invalid option '%s'" % option)

            exit(1)

        config.override(section.strip(), key.strip(), val.strip())

    load_groups(config)

    group = find_term_in_args(argv, _commands)

    if not group:

        parser.print_help()

        print_groups()

        exit(0)

    parser.prog = '%s %s <command>' % (exe, group)

    command = find_term_in_args(argv, _commands[group])

    if not command:

        parser.print_help()

        print_commands(group)

        exit(0)

    cmd = _commands[group][command]()

    parser.prog = '%s %s %s' % (exe, group, command)

    if cmd.syntax:

        parser.prog += '  %s' % cmd.syntax

    parser.description = cmd.description

    parser.epilog = ''

    if hasattr(cmd, 'update_parser'):

        cmd.update_parser(parser)

    #check other args

    args, argv = parser.parse_known_args()

    if group != argv[0]:

        errmsg = red('Invalid command group '+argv[0])

        print(errmsg, file=stderr)

        exit(1)

    if command != argv[1]:

        errmsg = red('Invalid command "%s" in group "%s"'%(argv[1], argv[0]))

        print(errmsg, file=stderr)

        exit(1)

    if args.help:

        parser.print_help()

        exit(0)

    manage_logging_handlers(args)

    cmd.args = args

    cmd.config = config

    try:

        ret = cmd.main(*argv[2:])

        exit(ret)

    except TypeError as e:

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

            parser.print_help()

            exit(1)

        else:

            raise

    except CLIError as err:

        errmsg = 'CLI Error '

        errmsg += '(%s): '%err.status if err.status else ': '

        errmsg += err.message if err.message else ''

        if err.importance == 1:

            errmsg = yellow(errmsg)

        elif err.importance == 2:

            errmsg = magenta(errmsg)

        elif err.importance > 2:

            errmsg = red(errmsg)

        print(errmsg, file=stderr)

        exit(1)

if __name__ == '__main__':

    main()

        return self.get(path, success=success**kwargs)

        return self.get(path, success=success**kwargs)

        return self.delete(path, success=success**kwargs)

        detail = 'detail' if details else ''

# Copyright 2011-2012 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.command

from kamaki.cli import command, set_api_description

set_api_description('astakos', 'Astakos API commands')

from kamaki.clients.astakos import AstakosClient, ClientError

from .cli_utils import raiseCLIError

from kamaki.utils import print_dict

class _astakos_init(object):

	def main(self):

		token = self.config.get('astakos', 'token') or self.config.get('global', 'token')

		base_url = self.config.get('astakos', 'url') or self.config.get('global', 'url')

		if base_url is None:

			raise ClientError('no URL for astakos')

		self.client = AstakosClient(base_url=base_url, token=token)

@command()

class astakos_authenticate(_astakos_init):

    """Authenticate a user"""

    def main(self):

    	super(astakos_authenticate, self).main()

    	try:

        	reply = self.client.authenticate()

        except ClientError as err:

        	raiseCLIError(err)

        print_dict(reply)

# Copyright 2011-2012 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.command

from kamaki.clients import ClientError

from kamaki.cli import CLIError

#from inspect import getargspec

def raiseCLIError(err, importance = -1):

	if importance < 0:

		if err.status <= 0:

			importance = 0

		elif err.status <= 400:

			importance = 1

		elif err.status <= 500:

			importance = 2

		else:

			importance = 3

	raise CLIError(err.message, err.status, err.details, importance)

def dict2file(d, f, depth = 0):

    for k, v in d.items():

        f.write('%s%s: '%('\t'*depth, k))

        if type(v) is dict:

            f.write('\n')

            dict2file(v, f, depth+1)

        elif type(v) is list:

            f.write('\n')

            list2file(v, f, depth+1)

        else:

            f.write(' %s\n'%unicode(v))

def list2file(l, f, depth = 1):

    for item in l:

        if type(item) is dict:

            dict2file(item, f, depth+1)

        elif type(item) is list:

            list2file(item, f, depth+1)

        else:

            f.write('%s%s\n'%('\t'*depth, unicode(item)))

# Copyright 2012 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.

from kamaki.cli import command, set_api_description, CLIError

from kamaki.utils import print_dict, print_items, print_list, format_size

from colors import bold

set_api_description('server', "Compute/Cyclades API server commands")

set_api_description('flavor', "Compute/Cyclades API flavor commands")

set_api_description('image', "Compute/Cyclades or Glance API image commands")

set_api_description('network', "Compute/Cyclades API network commands")

from kamaki.clients.cyclades import CycladesClient, ClientError

from .cli_utils import raiseCLIError

from colors import yellow

class _init_cyclades(object):

    def main(self):

        token = self.config.get('compute', 'token') or self.config.get('global', 'token')

        base_url = self.config.get('compute', 'url') or self.config.get('global', 'url')

        self.client = CycladesClient(base_url=base_url, token=token)

@command()

class server_list(_init_cyclades):

    """List servers"""

    def print_servers(self, servers):

        for server in servers:

            sname = server.pop('name')

            sid = server.pop('id')

            print('%s (%s)'%(bold(sname), bold(unicode(sid))))

            if getattr(self.args, 'detail'):

                server_info.print_server(server)

            print('- - -')

    def update_parser(self, parser):

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

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

    def main(self):

        super(self.__class__, self).main()

        try:

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

            self.print_servers(servers)

            #print_items(servers)

        except ClientError as err:

            raiseCLIError(err)

@command()

class server_info(_init_cyclades):

    """Get server details"""

    @classmethod

    def print_server(self,server):

        addr_dict = {}

        if server.has_key('addresses'):

            for addr in server['addresses']['values']:

                ips = addr.pop('values', [])

                for ip in ips:

                    addr['IPv%s'%ip['version']] = ip['addr']

                if addr.has_key('firewallProfile'):

                    addr['firewall'] = addr.pop('firewallProfile')

                addr_dict[addr['name']] = addr

            server['addresses'] = addr_dict

        print_dict(server, ident=14)

    def main(self, server_id):

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        except ValueError as err:

            raise CLIError(message='Server id must be positive integer',

                importance=1)

        self.print_server(server)

@command()

class server_create(_init_cyclades):

    """Create a server"""

    def update_parser(self, parser):

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

                          action='append', default=[],

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

                          help='add a personality file')

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

        super(self.__class__, self).main()

        personalities = []

        for personality in self.args.personalities:

            p = personality.split(',')

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

            path = p[0]

            if not path:

                raise CLIError(message='Invalid personality argument %s'%p, importance=1)

            if not exists(path):

                raise CLIError(message="File %s does not exist" % path, importance=1)

            with open(path) as f:

                contents = b64encode(f.read())

            d = {'path': p[1] or abspath(path), 'contents': contents}

            if p[2]:

                d['owner'] = p[2]

            if p[3]:

                d['group'] = p[3]

            if p[4]:

                d['mode'] = int(p[4])

            personalities.append(d)

        try:

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

                personalities)

        except ClientError as err:

            raiseCLIError(err)

        print_dict(reply)

@command()

class server_rename(_init_cyclades):

    """Update a server's name"""

    def main(self, server_id, new_name):

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        except ValueError:

            raise CLIError(message='Server id must be positive integer', importance=1)

@command()

class server_delete(_init_cyclades):

    """Delete a server"""

    def main(self, server_id):

        super(self.__class__, self).main()

        try:

            self.client.delete_server(int(server_id))

        except ClientError as err:

            raiseCLIError(err)

        except ValueError:

            raise CLIError(message='Server id must be positive integer', importance=1)

@command()

class server_reboot(_init_cyclades):

    """Reboot a server"""

    def update_parser(self, parser):

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

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

    def main(self, server_id):

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        except ValueError:

            raise CLIError(message='Server id must be positive integer', importance=1)

@command()

class server_start(_init_cyclades):

    """Start a server"""

    def main(self, server_id):

        super(self.__class__, self).main()

        try:

            self.client.start_server(int(server_id))

        except ClientError as err:

            raiseCLIError(err)

        except ValueError:

            raise CLIError(message='Server id must be positive integer', importance=1)

@command()

class server_shutdown(_init_cyclades):

    """Shutdown a server"""

    def main(self, server_id):

        super(self.__class__, self).main()

        try:

            self.client.shutdown_server(int(server_id))

        except ClientError as err:

            raiseCLIError(err)

        except ValueError:

            raise CLIError(message='Server id must be positive integer', importance=1)

@command()

class server_console(_init_cyclades):

    """Get a VNC console"""

    def main(self, server_id):

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        except ValueError:

            raise CLIError(message='Server id must be positive integer', importance=1)

        print_dict(reply)

@command()

class server_firewall(_init_cyclades):

    """Set the server's firewall profile"""

    def main(self, server_id, profile):

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        except ValueError:

            raise CLIError(message='Server id must be positive integer', importance=1)

@command()

class server_addr(_init_cyclades):

    """List a server's nic address"""

    def main(self, server_id):

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        except ValueError:

            raise CLIError(message='Server id must be positive integer', importance=1)

        print_list(reply)

@command()

class server_meta(_init_cyclades):

    """Get a server's metadata"""

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

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        except ValueError:

            raise CLIError(message='Server id must be positive integer', importance=1)

        print_dict(reply)

@command()

class server_addmeta(_init_cyclades):

    """Add server metadata"""

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

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        except ValueError:

            raise CLIError(message='Server id must be positive integer', importance=1)

        print_dict(reply)

@command()

class server_setmeta(_init_cyclades):

    """Update server's metadata"""

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

        super(self.__class__, self).main()

        metadata = {key: val}

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        except ValueError:

            raise CLIError(message='Server id must be positive integer', importance=1)

        print_dict(reply)

@command()

class server_delmeta(_init_cyclades):

    """Delete server metadata"""

    def main(self, server_id, key):

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        except ValueError:

            raise CLIError(message='Server id must be positive integer', importance=1)

@command()

class server_stats(_init_cyclades):

    """Get server statistics"""

    def main(self, server_id):

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        except ValueError:

            raise CLIError(message='Server id must be positive integer', importance=1)

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

@command()

class flavor_list(_init_cyclades):

    """List flavors"""

    def update_parser(self, parser):

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

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

    def main(self):

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        print_items(flavors)

@command()

class flavor_info(_init_cyclades):

    """Get flavor details"""

    def main(self, flavor_id):

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        except ValueError:

            raise CLIError(message='Server id must be positive integer', importance=1)

        print_dict(flavor)

@command()

class image_list(_init_cyclades):

    """List images"""

    def update_parser(self, parser):

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

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

    def main(self):

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        print_items(images)

@command()

class image_info(_init_cyclades):

    """Get image details"""

    def main(self, image_id):

        super(self.__class__, self).main()

        try:

            image = self.client.get_image_details(image_id)

        except ClientError as err:

            raiseCLIError(err)

        print_dict(image)

@command()

class image_delete(_init_cyclades):

    """Delete image"""

    def main(self, image_id):

        super(self.__class__, self).main()

        try:

            self.client.delete_image(image_id)

        except ClientError as err:

            raiseCLIError(err)

@command()

class image_properties(_init_cyclades):

    """Get image properties"""

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

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        print_dict(reply)

@command()

class image_addproperty(_init_cyclades):

    """Add an image property"""

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

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        print_dict(reply)

@command()

class image_setproperty(_init_cyclades):

    """Update an image property"""

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

        super(self.__class__, self).main()

        metadata = {key: val}

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        print_dict(reply)

@command()

class image_delproperty(_init_cyclades):

    """Delete an image property"""

    def main(self, image_id, key):

        super(self.__class__, self).main()

        try:

            self.client.delete_image_metadata(image_id, key)

        except ClientError as err:

            raiseCLIError(err)

@command()

class network_list(_init_cyclades):

    """List networks"""

    def update_parser(self, parser):

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

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

    def print_networks(self, nets):

        for net in nets:

            netname = bold(net.pop('name'))

            netid = bold(unicode(net.pop('id')))

            print('%s (%s)'%(netname, netid))

            if getattr(self.args, 'detail'):

                network_info.print_network(net)

                print('- - -')

    def main(self):

        super(self.__class__, self).main()

        try:

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

        except ClientError as err:

            raiseCLIError(err)

        self.print_networks(networks)

@command()

class network_create(_init_cyclades):

    """Create a network"""

    def update_parser(self, parser):

        try:

            super(self.__class__, self).update_parser(parser)

        except AttributeError:

            pass

        parser.add_argument('--with-cidr', action='store', dest='cidr', default=False,

            help='specific cidr for new network')

        parser.add_argument('--with-gateway', action='store', dest='gateway', default=False,

            help='specific getaway for new network')

        parser.add_argument('--with-dhcp', action='store', dest='dhcp', default=False,

            help='specific dhcp for new network')

        parser.add_argument('--with-type', action='store', dest='type', default=False,

            help='specific type for new network')

    def main(self, name):

        super(self.__class__, self).main()

        try:

            reply = self.client.create_network(name, cidr=getattr(self.args, 'cidr'),

                gateway=getattr(self.args, 'gateway'), dhcp=getattr(self.args, 'gateway'),

                type=getattr(self.args, 'type'))

        except ClientError as err:

            raiseCLIError(err)

        print_dict(reply)

@command()

class network_info(_init_cyclades):

    """Get network details"""

    @classmethod

    def print_network(self, net):

        if net.has_key('attachments'):

            att = net['attachments']['values']

            net['attachments'] = att if len(att) > 0 else None

        print_dict(net, ident=14)

    def main(self, network_id):

        super(self.__class__, self).main()

        try:

            network = self.client.get_network_details(network_id)

        except ClientError as err:

            raiseCLIError(err)

        network_info.print_network(network)

@command()

class network_rename(_init_cyclades):

    """Update network name"""

    def main(self, network_id, new_name):

        super(self.__class__, self).main()

        try:

            self.client.update_network_name(network_id, new_name)

        except ClientError as err:

            raiseCLIError(err)

@command()

class network_delete(_init_cyclades):

    """Delete a network"""

    def main(self, network_id):

        super(self.__class__, self).main()

        try:

            self.client.delete_network(network_id)

        except ClientError as err:

            raiseCLIError(err)

@command()

class network_connect(_init_cyclades):

    """Connect a server to a network"""

    def main(self, server_id, network_id):

        super(self.__class__, self).main()

        try:

            self.client.connect_server(server_id, network_id)

        except ClientError as err:

            raiseCLIError(err)

@command()

class network_disconnect(_init_cyclades):

    """Disconnect a nic that connects a server to a network"""

    def main(self, nic_id):

        super(self.__class__, self).main()

        try:

            server_id = nic_id.split('-')[1]

            self.client.disconnect_server(server_id, nic_id)

        except IndexError:

            raise CLIError(message='Incorrect nic format', importance=1,

                details='nid_id format: nic-<server_id>-<nic_index>')

        except ClientError as err:

            raiseCLIError(err)

# Copyright 2011-2012 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.command

from kamaki.cli import command, set_api_description

from kamaki.utils import print_dict, print_items

set_api_description('image', "Compute/Cyclades or Glance API image commands")

from kamaki.clients.image import ImageClient, ClientError

from .cli_utils import raiseCLIError

class _init_image(object):

    def main(self):

        try:

            token = self.config.get('image', 'token') or self.config.get('global', 'token')

            base_url = self.config.get('image', 'url') or self.config.get('global', 'url')

            self.client = ImageClient(base_url=base_url, token=token)

        except ClientError as err:

            raiseCLIError(err)

@command()

class image_public(_init_image):

    """List public images"""

    def update_parser(self, parser):

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

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

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

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

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

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

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

                help='filter by name')

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

                help='filter by minimum size')

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

                help='filter by maximum size')

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

                help='filter by status')

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

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

    def main(self):

    	super(self.__class__, self).main()

        filters = {}

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

                       'size_max', 'status'):

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

            if val is not None:

                filters[filter] = val

        order = self.args.order or ''

        try:

            images = self.client.list_public(self.args.detail,

                filters=filters, order=order)

        except ClientError as err:

            raiseCLIError(err)

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

@command()

class image_meta(_init_image):

    """Get image metadata"""

    def main(self, image_id):

    	super(self.__class__, self).main()

        try:

            image = self.client.get_meta(image_id)

        except ClientError as err:

            raiseCLIError(err)

        print_dict(image)

@command()

class image_register(_init_image):

    """Register an image"""

    def update_parser(self, parser):

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

                help='set image checksum')

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

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

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

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

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

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

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

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

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

                metavar='KEY=VAL',

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

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

                help='mark image as public')

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

                help='set image size')

    def main(self, name, location):

    	super(self.__class__, self).main()

        if not location.startswith('pithos://'):

            account = self.config.get('storage', 'account').split()[0]

            if account[-1] == '/':

                account = account[:-1]

            container = self.config.get('storage', 'container')

            location = 'pithos://%s/%s'%(account, location) \

                if container is None or len(container) == 0 \

                else 'pithos://%s/%s/%s' % (account, container, location)

        params = {}

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

                    'owner', 'size'):

            val = getattr(self.args, key)

            if val is not None:

                params[key] = val

        if self.args.is_public:

            params['is_public'] = 'true'

        properties = {}

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

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

            if not sep:

                raise CLIError(message="Invalid property '%s'" % property, importance=1)

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

        try:

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

        except ClientError as err:

            raiseCLIError(err)

@command()

class image_members(_init_image):

    """Get image members"""

    def main(self, image_id):

    	super(self.__class__, self).main()

        try:

            members = self.client.list_members(image_id)

        except ClientError as err:

            raiseCLIError(err)

        for member in members:

            print(member['member_id'])

@command()

class image_shared(_init_image):

    """List shared images"""

    def main(self, member):

    	super(self.__class__, self).main()

        try:

            images = self.client.list_shared(member)

        except ClientError as err:

            raiseCLIError(err)

        for image in images:

            print(image['image_id'])

@command()

class image_addmember(_init_image):

    """Add a member to an image"""

    def main(self, image_id, member):

    	super(self.__class__, self).main()

        try:

            self.client.add_member(image_id, member)

        except ClientError as err:

            raiseCLIError(err)

@command()

class image_delmember(_init_image):

    """Remove a member from an image"""

    def main(self, image_id, member):

        super(self.__class__, self).main()

        try:

            self.client.remove_member(image_id, member)

        except ClientError as err:

            raiseCLIError(err)

@command()

class image_setmembers(_init_image):

    """Set the members of an image"""

    def main(self, image_id, *member):

        super(self.__class__, self).main()

        try:

            self.client.set_members(image_id, member)

        except ClientError as err:

            raiseCLIError(err)

# Copyright 2011-2012 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.command

from kamaki.cli import command, set_api_description, CLIError

from kamaki.clients.utils import filter_in

from kamaki.utils import format_size

set_api_description('store', 'Pithos+ storage commands')

from kamaki.clients.pithos import PithosClient, ClientError

from .cli_utils import raiseCLIError

from kamaki.utils import print_dict, pretty_keys, print_list

from colors import bold

from sys import stdout, exit

import signal

from progress.bar import IncrementalBar

class ProgressBar(IncrementalBar):

    #suffix = '%(percent)d%% - %(eta)ds'

    suffix = '%(percent)d%%'

class _pithos_init(object):

    def main(self):

        self.token = self.config.get('store', 'token') or self.config.get('global', 'token')

        self.base_url = self.config.get('store', 'url') or self.config.get('global', 'url')

        self.account = self.config.get('store', 'account') or self.config.get('global', 'account')

        self.container = self.config.get('store', 'container') or self.config.get('global', 'container')

        self.client = PithosClient(base_url=self.base_url, token=self.token, account=self.account,

            container=self.container)

class _store_account_command(_pithos_init):

    """Base class for account level storage commands"""

    def update_parser(self, parser):

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

                          help="Specify an account to use")

    def progress(self, message):

        """Return a generator function to be used for progress tracking"""

        MESSAGE_LENGTH = 25

        def progress_gen(n):

            msg = message.ljust(MESSAGE_LENGTH)

            for i in ProgressBar(msg).iter(range(n)):

                yield

            yield

        return progress_gen

    def main(self):

        super(_store_account_command, self).main()

        if hasattr(self.args, 'account') and self.args.account is not None:

            self.client.account = self.args.account

class _store_container_command(_store_account_command):

    """Base class for container level storage commands"""

    def __init__(self):

        self.container = None

        self.path = None

    def update_parser(self, parser):

        super(_store_container_command, self).update_parser(parser)

        parser.add_argument('--container', dest='container', metavar='NAME', default=None,

            help="Specify a container to use")

    def extract_container_and_path(self, container_with_path, path_is_optional=True):

        assert isinstance(container_with_path, str)

        if ':' not in container_with_path:

            if hasattr(self.args, 'container'):

                self.container = getattr(self.args, 'container')

            else:

                self.container = self.client.container

            if self.container is None:

                self.container = container_with_path

            else:

                self.path = container_with_path

            if not path_is_optional and self.path is None:

                raise CLIError(message="Object path is missing", status=11)

            return

        cnp = container_with_path.split(':')

        self.container = cnp[0]

        try:

            self.path = cnp[1]

        except IndexError:

            if path_is_optional:

                self.path = None

            else:

                raise CLIError(message="Object path is missing", status=11)

    def main(self, container_with_path=None, path_is_optional=True):

        super(_store_container_command, self).main()

        if container_with_path is not None:

            self.extract_container_and_path(container_with_path, path_is_optional)

            self.client.container = self.container

        elif hasattr(self.args, 'container'):

            self.client.container = getattr(self.args,'container')

        self.container = self.client.container

"""

@command()

class store_test(_store_container_command):

    ""Test various stuff""

    def main(self):

        super(self.__class__, self).main('pithos')

        r = self.client.list_containers()

        for item in r:

            if item['name'].startswith('c1_') or item['name'].startswith('c2_') \

            or item['name'].startswith('c3_'):

                self.client.container = item['name']

                self.client.del_container(delimiter='/')

                self.client.del_container()

"""

@command()

class store_list(_store_container_command):

    """List containers, object trees or objects in a directory

    """

    def update_parser(self, parser):

        super(self.__class__, self).update_parser(parser)

        parser.add_argument('-l', action='store_true', dest='detail', default=False,

            help='show detailed output')

        parser.add_argument('-N', action='store', dest='show_size', default=1000,

            help='print output in chunks of size N')

        parser.add_argument('-n', action='store', dest='limit', default=None,

            help='show limited output')

        parser.add_argument('--marker', action='store', dest='marker', default=None,

            help='show output greater then marker')

        parser.add_argument('--prefix', action='store', dest='prefix', default=None,

            help='show output starting with prefix')

        parser.add_argument('--delimiter', action='store', dest='delimiter', default=None, 

            help='show output up to the delimiter')

        parser.add_argument('--path', action='store', dest='path', default=None, 

            help='show output starting with prefix up to /')

        parser.add_argument('--meta', action='store', dest='meta', default=None, 

            help='show output having the specified meta keys (e.g. --meta "meta1 meta2 ..."')

        parser.add_argument('--if-modified-since', action='store', dest='if_modified_since', 

            default=None, help='show output if modified since then')

        parser.add_argument('--if-unmodified-since', action='store', dest='if_unmodified_since',

            default=None, help='show output if not modified since then')

        parser.add_argument('--until', action='store', dest='until', default=None,

            help='show metadata until that date')

        dateformat = '%d/%m/%Y %H:%M:%S'

        parser.add_argument('--format', action='store', dest='format', default=dateformat,

            help='format to parse until date (default: d/m/Y H:M:S)')