Synnefo » ./kamaki

# Copyright 2011-2013 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

from kamaki.cli.command_tree import CommandTree

from kamaki.cli.commands.cyclades import _init_cyclades

from kamaki.cli.errors import CLIBaseUrlError

from kamaki.cli.argument import (

    FlagArgument, ValueArgument, KeyValueArgument, IntArgument)

from kamaki.cli.commands import _command_init, errors, addLogSettings

from kamaki.cli.commands import _optional_json

from kamaki.clients.volume import VolumeClient

snapshot_cmds = CommandTree('snapshot', 'Server image snapshot operations')

volume_cmds = CommandTree('volume', 'Server volume operations')

_commands = [snapshot_cmds, volume_cmds]

class _init_volume(_command_init):

    @errors.generic.all

    @addLogSettings

    def _run(self, service='volume'):

        if getattr(self, 'cloud', None):

            base_url = self._custom_url(service) or self._custom_url(

                'volume')

            if base_url:

                token = self._custom_token(service) or self._custom_token(

                    'volume') or self.config.get_cloud('token')

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

                return

        else:

            self.cloud = 'default'

        if getattr(self, 'auth_base', False):

            cyclades_endpoints = self.auth_base.get_service_endpoints(

                self._custom_type('volume') or 'volume',

                self._custom_version('volume') or '')

            base_url = cyclades_endpoints['publicURL']

            token = self.auth_base.token

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

        else:

            raise CLIBaseUrlError(service='volume')

    def main(self):

        self._run()

@command(volume_cmds)

class volume_list(_init_cyclades, _optional_json):

    """List a server volume or the volumes for all servers"""

    arguments = dict(

        server_id=IntArgument('list volumes of this server', '--server-id')

    )

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self):

        servers = [self.client.get_server_details(self['server_id'])] if (

            self['server_id']) else self.client.list_servers(detail=True)

        volumes = [dict(

            server_id=s['id'],

            volumes=' '.join('%s' % v for v in s['volumes'])) for s in servers]

        self._print(volumes)

    def main(self):

        super(self.__class__, self)._run()

        self._run()

@command(snapshot_cmds)

class snapshot_create(_init_volume, _optional_json):

    """Create a snapshot of a volume"""

    arguments = dict(

        name=ValueArgument(

            'The snapshot name (otherwise it is created automatically)',

            '--name'),

        description=ValueArgument('Snapshot description', '--description'),

        metadata=KeyValueArgument(

            'Snapshot custom metadata (can be repeated)', '--metadata')

    )

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self, volume_id):

        self._print(self.client.create_snapshot(

            volume_id, self['name'], self['description'], self['metadata']))

    def main(self, volume_id):

        super(self.__class__, self)._run()

        self._run(volume_id)

@command(snapshot_cmds)

class snapshot_list(_init_cyclades, _optional_json):

    """List all snapshots"""

    arguments = dict(

        detail=FlagArgument('show detailed output', ('-l', '--details'))

    )

    @errors.generic.all

    @errors.cyclades.connection

    def _run(self):

        r = [img for img in self.client.list_images(detail=True) if (

            img.get('is_snapshot', False))]

        if not self['detail']:

            r = [dict(name=img['name'], id=img['id']) for img in r]

        self._print(r)

    def main(self):

        super(self.__class__, self)._run()

        self._run()