Synnefo

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

import itertools

import warnings

import functools

from optparse import make_option

from django.core.management.base import NoArgsCommand, CommandError, BaseCommand

from django.db import transaction

from django.conf import settings

from synnefo.api.util import get_existing_users

from synnefo.lib.utils import case_unique

from synnefo.db.models import Network, VirtualMachine

from synnefo.userdata.models import PublicKeyPair

from snf_django.lib import astakos

def warn(*msgs):

    print "WARNING: %s" % ' '.join(msgs)

get_displayname = functools.partial(astakos.get_displayname,

                                 settings.CYCLADES_SERVICE_TOKEN,

                                 url=settings.ASTAKOS_URL.replace('im/authenticate',

                                                                 'service/api/user_catalogs'))

get_user_uuid = functools.partial(astakos.get_user_uuid,

                                 settings.CYCLADES_SERVICE_TOKEN,

                                 url=settings.ASTAKOS_URL.replace('im/authenticate',

                                                                 'service/api/user_catalogs'))

def _validate_db_state(usernames):

    usernames = filter(bool, usernames)

    invalid_case_users = case_unique(usernames)

    if invalid_case_users:

        invalid_case_users.append(invalid_case_users[0].lower())

        raise CommandError("Duplicate case insensitive user identifiers exist %r" % invalid_case_users)

    uuidusers = filter(lambda uid:'@' in uid or uid == None, usernames)

    if len(uuidusers) != len(usernames):

        warn('It seems that mixed uuid/email user identifiers exist in database.')

        return False

    return True

@transaction.commit_manually

def delete_user(username, only_stats=True, dry=True):

    vms = VirtualMachine.objects.filter(userid__exact=username)

    networks = Network.objects.filter(userid__exact=username)

    keys = PublicKeyPair.objects.filter(user__exact=username)

    if not len(list(itertools.ifilter(bool, map(lambda q: q.count(), [vms,

                                                                      networks,

                                                                      keys])))):

        print "No entries exist for '%s'" % username

        return -1

    if only_stats:

        print "The following entries will be deleted if you decide to remove this user"

        print "%d Virtual Machines" % vms.exclude(operstate='DESTROYED').count()

        print "%d Destroyed Virtual Machines" % vms.filter(operstate='DESTROYED').count()

        print "%d Networks" % networks.count()

        print "%d PublicKeyPairs" % keys.count()

        return

    for o in itertools.chain(vms, networks):

        o.delete()

    for key in keys:

        key.delete()

    if dry:

        print "Skipping database commit."

        transaction.rollback()

    else:

        transaction.commit()

        print "User entries removed."

@transaction.commit_on_success

def merge_user(username):

    vms = VirtualMachine.objects.filter(userid__iexact=username)

    networks = Network.objects.filter(userid__iexact=username)

    keys = PublicKeyPair.objects.filter(user__iexact=username)

    for o in itertools.chain(vms, networks):

        o.userid = username.lower()

        o.save()

    for key in keys:

        key.user = username.lower()

        key.save()

def migrate_user(username, uuid):

    """

    Warn: no transaction handling. Consider wrapping within another function.

    """

    vms = VirtualMachine.objects.filter(userid__exact=username)

    networks = Network.objects.filter(userid__exact=username)

    keys = PublicKeyPair.objects.filter(user__exact=username)

    for o in itertools.chain(vms, networks):

        o.userid = uuid or o.userid

        o.save()

    for key in keys:

        key.user = uuid

        key.save()

@transaction.commit_manually

def migrate_users(usernames, dry=True):

    usernames = filter(bool, usernames)

    count = 0

    for u in usernames:

        if not '@' in u:

            warn('Skipping %s. It doesn\'t seem to be an email' % u)

            continue

        try:

            uuid = get_user_uuid(u)

            print "%s -> %s" % (u, uuid)

            if not uuid:

                raise Exception("No uuid for %s" % u)

            migrate_user(u, uuid)

            count += 1

        except Exception, e:

            print "ERROR: User id migration failed (%s)" % e

    if dry:

        print "Skipping database commit."

        transaction.rollback()

    else:

        transaction.commit()

        print "Migrated %d users" % count

class Command(NoArgsCommand):

    help = "Quotas migration helper"

    option_list = BaseCommand.option_list + (

        make_option('--strict',

                    dest='strict',

                    action="store_false",

                    default=True,

                    help="Exit on warnings."),

        make_option('--validate-db',

                    dest='validate',

                    action="store_true",

                    default=True,

                    help=("Check if cyclades database contents are valid for "

                          "migration.")),

        make_option('--migrate-users',

                    dest='migrate_users',

                    action="store_true",

                    default=False,

                    help=("Convert emails to uuids for all users stored in "

                          "database.")),

        make_option('--merge-user',

                    dest='merge_user',

                    default=False,

                    help="Merge case insensitive duplicates of a user."),

        make_option('--delete-user',

                    dest='delete_user',

                    action='store',

                    default=False,

                    help="Delete user entries."),

        make_option('--user-entries',

                    dest='user_entries',

                    action='store',

                    default=False,

                    help="Display user summary."),

        make_option('--dry',

                    dest='dry',

                    action="store_true",

                    default=False,

                    help="Do not commit database changes. Do not communicate "

                         "with quotaholder"),

        make_option('--user',

                    dest='user',

                    action="store",

                    default=False,)

    )

    def resolve_conflicts(self, options):

        conflicting = map(options.get, ['migrate_users',

                                        'merge_user'])

        if len(filter(bool, conflicting)) > 1:

            raise CommandError('You can use only one of --validate,'

                               '--migrate-users')

    def handle(self, *args, **options):

        self.resolve_conflicts(options)

        self.strict = options.get('strict')

        self.dry = options.get('dry')

        if options.get('validate') and not options.get('merge_user') and not \

                options.get('delete_user') and not options.get('user_entries'):

            usernames = get_existing_users()

            _validate_db_state(usernames)

        if options.get('migrate_users'):

            migrate_users(usernames, dry=self.dry)

        if options.get('merge_user'):

            merge_user(options.get('merge_user'))

            print "Merge finished."

        if options.get('delete_user'):

            entries = delete_user(options.get('delete_user'), only_stats=True)

            if entries == -1:

                return

            confirm = raw_input("Type 'yes of course' if you are sure you want"

                                " to remove those entries: ")

            if not confirm == 'yes of course':

                return

            else:

                delete_user(options.get('delete_user'), only_stats=False,

                            dry=self.dry)

        if options.get('user_entries'):

            delete_user(options.get('user_entries'))

# 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 itertools import product

from optparse import make_option

from django.core.management.base import BaseCommand, CommandError

from synnefo.db.models import Flavor

class Command(BaseCommand):

    output_transaction = True

    option_list = BaseCommand.option_list + (

        make_option("-n", "--dry-run", dest="dry_run", action="store_true"),

    )

    args = "<cpu>[,<cpu>,...] " \

           "<ram>[,<ram>,...] " \

           "<disk>[,<disk>,...] " \

           "<disk template>[,<disk template>,...]"

    help = "Create one or more flavors.\n\nThe flavors that will be created"\

           " are those belonging to the cartesian product of the arguments"

    def handle(self, *args, **options):

        if len(args) != 4:

            raise CommandError("Invalid number of arguments")

        cpus = args[0].split(',')

        rams = args[1].split(',')

        disks = args[2].split(',')

        templates = args[3].split(',')

        flavors = []

        for cpu, ram, disk, template in product(cpus, rams, disks, templates):

            try:

                flavors.append((int(cpu), int(ram), int(disk), template))

            except ValueError:

                raise CommandError("Invalid values")

        for cpu, ram, disk, template in flavors:

            if options["dry_run"]:

                flavor = Flavor(cpu=cpu, ram=ram, disk=disk,

                                disk_template=template)

                self.stdout.write("Creating flavor '%s'\n" % (flavor.name,))

            else:

                flavor, created = \

                    Flavor.objects.get_or_create(cpu=cpu, ram=ram, disk=disk,

                                                 disk_template=template)

                if created:

                    self.stdout.write("Created flavor '%s'\n" % (flavor.name,))

                else:

                    self.stdout.write("Flavor '%s' already exists\n"

                                      % flavor.name)

                    if flavor.deleted:

                        msg = "Flavor '%s' is marked as deleted. Use"\

                        " 'snf-manage flavor-modify' to restore this flavor\n"\

                        % flavor.name

                        self.stdout.write(msg)

# Copyright 2012-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 snf_django.management.commands import ListCommand

from synnefo.db.models import Flavor, VirtualMachine

class Command(ListCommand):

    help = "List available server flavors"

    object_class = Flavor

    deleted_field = "deleted"

    def get_vms(flavor):

        return VirtualMachine.objects.filter(flavor=flavor, deleted=False)\

                                     .count()

    FIELDS = {

        "id": ("id", "Flavor's unique ID"),

        "name": ("name", "Flavor's unique name"),

        "cpu": ("cpu", "Number of CPUs"),

        "ram": ("ram", "Size(MB) of RAM"),

        "disk": ("disk", "Size(GB) of disk"),

        "template": ("disk_template", "Disk template"),

        "vms": (get_vms, "Number of active servers using this flavor")

    }

    fields = ["id", "name", "cpu", "ram", "disk", "template", "vms"]

# 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 optparse import make_option

from django.core.management.base import BaseCommand, CommandError

from synnefo.management.common import get_flavor

from snf_django.management.utils import parse_bool

from logging import getLogger

log = getLogger(__name__)

class Command(BaseCommand):

    args = "<flavor id>"

    help = "Modify a flavor"

    option_list = BaseCommand.option_list + (

        make_option(

            "--deleted",

            dest="deleted",

            metavar="True|False",

            choices=["True", "False"],

            default=None,

            help="Mark/unmark a flavor as deleted"),

    )

    def handle(self, *args, **options):

        if len(args) != 1:

            raise CommandError("Please provide a flavor ID")

        flavor = get_flavor(args[0])

        deleted = options['deleted']

        if deleted:

            deleted = parse_bool(deleted)

            log.info("Marking flavor %s as deleted=%s", flavor, deleted)

            flavor.deleted = deleted

            flavor.save()

        else:

            log.info("Nothing changed!")

# 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 optparse import make_option

from django.core.management.base import BaseCommand, CommandError

from synnefo.management import common

from synnefo.logic import servers

class Command(BaseCommand):

    help = "Attach a floating IP to a VM or router"

    option_list = BaseCommand.option_list + (

        make_option(

            '--machine',

            dest='machine',

            default=None,

            help='The server id the floating-ip will be attached to'),

    )

    @common.convert_api_faults

    def handle(self, *args, **options):

        if not args or len(args) > 1:

            raise CommandError("Command accepts exactly one argument")

        floating_ip_id = args[0]  # this is the floating-ip address

        device = options['machine']

        if not device:

            raise CommandError('Please give either a server or a router id')

        #get the vm

        vm = common.get_vm(device)

        floating_ip = common.get_floating_ip_by_id(floating_ip_id,

                                                   for_update=True)

        servers.create_port(vm.userid, floating_ip.network,

                            use_ipaddress=floating_ip, machine=vm)

        self.stdout.write("Attached %s to %s.\n" % (floating_ip, vm))

# 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 optparse import make_option

from django.core.management.base import BaseCommand, CommandError

from synnefo.management.common import convert_api_faults

from synnefo.logic import ips

from synnefo.api import util

class Command(BaseCommand):

    help = "Allocate a new floating IP"

    option_list = BaseCommand.option_list + (

        make_option(

            '--pool',

            dest='pool',

            help="The ID of the floating IP pool(network) to allocate the"

                 " address from"),

        make_option(

            '--address',

            dest='address',

            help="The address to be allocated"),

        make_option(

            '--owner',

            dest='owner',

            default=None,

            help='The owner of the floating IP'),

    )

    @convert_api_faults

    def handle(self, *args, **options):

        if args:

            raise CommandError("Command doesn't accept any arguments")

        network_id = options['pool']

        address = options['address']

        owner = options['owner']

        if not owner:

            raise CommandError("'owner' is required for floating IP creation")

        if network_id is not None:

            network = util.get_network(network_id, owner, for_update=True,

                                       non_deleted=True)

        else:

            network = None

        floating_ip = ips.create_floating_ip(userid=owner,

                                             network=network,

                                             address=address)

        self.stdout.write("Created floating IP '%s'.\n" % floating_ip)

# 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 optparse import make_option

from django.core.management.base import BaseCommand, CommandError

from synnefo.management import common

from synnefo.logic import servers

class Command(BaseCommand):

    help = "Dettach a floating IP from a VM or router"

    @common.convert_api_faults

    def handle(self, *args, **options):

        if not args or len(args) > 1:

            raise CommandError("Command accepts exactly one argument")

        floating_ip_id = args[0]

        #get the floating-ip

        floating_ip = common.get_floating_ip_by_id(floating_ip_id,

                                                   for_update=True)

        if not floating_ip.nic:

            raise CommandError('This floating IP is not attached to a device')

        nic = floating_ip.nic

        vm = nic.machine

        servers.delete_port(nic)

        self.stdout.write("Dettached floating IP %s from  %s.\n"

                          % (floating_ip_id, vm))

# Copyright 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 optparse import make_option

from django.db import transaction

from django.core.management.base import CommandError

from snf_django.management.commands import RemoveCommand

from synnefo.management import common

from synnefo.logic import ips

class Command(RemoveCommand):

    args = "<Floating-IP ID> [<Floating-IP ID> ...]"

    help = "Release a floating IP"

    @common.convert_api_faults

    @transaction.commit_on_success

    def handle(self, *args, **options):

        if not args:

            raise CommandError("Please provide a floating-ip ID")

        force = options['force']

        message = "floating IPs" if len(args) > 1 else "floating IP"

        self.confirm_deletion(force, message, args)

        for floating_ip_id in args:

            self.stdout.write("\n")

            try:

                floating_ip = common.get_floating_ip_by_id(floating_ip_id,

                                                           for_update=True)

                ips.delete_floating_ip(floating_ip)

                self.stdout.write("Deleted floating IP '%s'.\n" %

                                  floating_ip_id)

            except CommandError as e:

                self.stdout.write("Error -- %s\n" % e.message)

# Copyright 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 snf_django.management.commands import ListCommand

from synnefo.db.models import IPAddressLog

from optparse import make_option

from logging import getLogger

log = getLogger(__name__)

class Command(ListCommand):

    help = "Information about a floating IP"

    option_list = ListCommand.option_list + (

        make_option(

            '--address',

            dest='address',

            help="Display IP history only for this address"),

        make_option(

            '--server',

            dest='server',

            help="Display IP history only for this server"),

        make_option(

            '--active',

            dest="active",

            action="store_true",

            default=False,

            help="Display only IPs that are currently in use")

    )

    object_class = IPAddressLog

    order_by = "allocated_at"

    FIELDS = {

        "address": ("address", "The IP address"),

        "server": ("server_id", "The the server connected to"),

        "network": ("network_id", "The id of the network"),

        "allocated_at": ("allocated_at", "Datetime IP allocated to server"),

        "released_at": ("released_at", "Datetime IP released from server"),

        "active": ("active", "Whether IP still allocated to server"),

    }

    fields = ["address", "server", "network", "allocated_at", "released_at"]

    def handle_args(self, *args, **options):

        if options["address"]:

            self.filters["address"] = options["address"]

        if options["server"]:

            self.filters["server_id"] = options["server"]

        if options["active"]:

            self.filters["active"] = True

# 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 optparse import make_option

from django.core.management.base import BaseCommand, CommandError

from synnefo.management.common import convert_api_faults

from snf_django.management.utils import parse_bool

from synnefo.db.models import Network

from synnefo.logic import networks, subnets

from synnefo.management import pprint

import ipaddr

NETWORK_FLAVORS = Network.FLAVORS.keys()

class Command(BaseCommand):

    can_import_settings = True

    output_transaction = True

    help = "Create a new network"

    option_list = BaseCommand.option_list + (

        make_option(

            '--name',

            dest='name',

            help="Name of the network"),

        make_option(

            '--owner',

            dest='owner',

            help="The owner of the network"),

        make_option(

            '--subnet',

            dest='subnet',

            default=None,

            # required=True,

            help='IPv4 subnet of the network'),

        make_option(

            '--gateway',

            dest='gateway',

            default=None,

            help='IPv4 gateway of the network'),

        make_option(

            '--subnet6',

            dest='subnet6',

            default=None,

            help='IPv6 subnet of the network'),

        make_option(

            '--gateway6',

            dest='gateway6',

            default=None,

            help='IPv6 gateway of the network'),

        make_option(

            '--dhcp',

            dest='dhcp',

            default="False",

            choices=["True", "False"],

            metavar="True|False",

            help='Automatically assign IPs'),

        make_option(

            '--public',

            dest='public',

            action='store_true',

            default=False,

            help='Network is public'),

        make_option(

            '--flavor',

            dest='flavor',

            default=None,

            choices=NETWORK_FLAVORS,

            help='Network flavor. Choices: ' + ', '.join(NETWORK_FLAVORS)),

        make_option(

            '--mode',

            dest='mode',

            default=None,

            help="Overwrite flavor connectivity mode."),

        make_option(

            '--link',

            dest='link',

            default=None,

            help="Overwrite flavor connectivity link."),

        make_option(

            '--mac-prefix',

            dest='mac_prefix',

            default=None,

            help="Overwrite flavor connectivity MAC prefix"),

        make_option(

            '--tags',

            dest='tags',

            default=None,

            help='The tags of the Network (comma separated strings)'),

        make_option(

            '--floating-ip-pool',

            dest='floating_ip_pool',

            default="False",

            choices=["True", "False"],

            metavar="True|False",

            help="Use the network as a Floating IP pool."),

        make_option(

            '--allocation-pool',

            dest='allocation_pools',

            action='append',

            help="IP allocation pools to be used for assigning IPs to"

                 " VMs. Can be used multiple times. Syntax: \n"

                 "192.168.42.220,192.168.42.240. Starting IP must proceed "

                 "ending IP. If no allocation pools are given, the whole "

                 "subnet range is used, excluding the gateway IP, the "

                 "broadcast address and the network address"),

    )

    @convert_api_faults

    def handle(self, *args, **options):

        if args:

            raise CommandError("Command doesn't accept any arguments")

        name = options['name']

        subnet = options['subnet']

        gateway = options['gateway']

        subnet6 = options['subnet6']

        gateway6 = options['gateway6']

        public = options['public']

        flavor = options['flavor']

        mode = options['mode']

        link = options['link']

        mac_prefix = options['mac_prefix']

        tags = options['tags']

        userid = options["owner"]

        allocation_pools = options["allocation_pools"]

        floating_ip_pool = parse_bool(options["floating_ip_pool"])

        dhcp = parse_bool(options["dhcp"])

        if name is None:

            name = ""

        if flavor is None:

            raise CommandError("flavor is required")

        if ((subnet is None) and (subnet6 is None)) and dhcp is not False:

            raise CommandError("Cannot set DHCP without subnet or subnet6")

        if subnet is None and gateway is not None:

            raise CommandError("Cannot use gateway without subnet")

        if subnet is None and allocation_pools is not None:

            raise CommandError("Cannot use allocation-pools without subnet")

        if subnet6 is None and gateway6 is not None:

            raise CommandError("Cannot use gateway6 without subnet6")

        if not (userid or public):

            raise CommandError("'owner' is required for private networks")

        network = networks.create(userid=userid, name=name, flavor=flavor,

                                  public=public, mode=mode,

                                  link=link, mac_prefix=mac_prefix, tags=tags,

                                  floating_ip_pool=floating_ip_pool)

        if subnet is not None:

            alloc = None

            if allocation_pools is not None:

                alloc = subnets.parse_allocation_pools(allocation_pools)

                alloc.sort()

            name = "IPv4 Subnet of Network %s" % network.id

            subnets.create_subnet(network.id, cidr=subnet, name=name,

                                  ipversion=4, gateway=gateway, dhcp=dhcp,

                                  user_id=userid,

                                  allocation_pools=alloc)

        if subnet6 is not None:

            name = "IPv6 Subnet of Network %s" % network.id

            subnets.create_subnet(network.id, cidr=subnet6, name=name,

                                  ipversion=6, gateway=gateway6,

                                  dhcp=dhcp, user_id=userid)

        self.stdout.write("Created network '%s' in DB:\n" % network)

        pprint.pprint_network(network, stdout=self.stdout)

        pprint.pprint_network_subnets(network, stdout=self.stdout)

        networks.create_network_in_backends(network)

        # TODO: Add --wait option to track job progress and report successful

        # creation in each backend.

        self.stdout.write("\nSuccessfully issued job to create network in"

                          " backends\n")

# Copyright 2012-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 optparse import make_option

from snf_django.management.commands import ListCommand

from synnefo.db.models import Network

from synnefo.settings import (CYCLADES_SERVICE_TOKEN as ASTAKOS_TOKEN,

                              ASTAKOS_AUTH_URL)

from logging import getLogger

log = getLogger(__name__)

class Command(ListCommand):

    help = "List networks"

    option_list = ListCommand.option_list + (

        make_option(

            '--public',

            action='store_true',

            dest='public',

            default=False,

            help="List only public networks"),

        make_option(

            '--ipv6',

            action='store_true',

            dest='ipv6',

            default=False,

            help="Include IPv6 information"),

    )

    object_class = Network

    select_related = []

    prefetch_related = ["subnets"]

    deleted_field = "deleted"

    user_uuid_field = "userid"

    astakos_auth_url = ASTAKOS_AUTH_URL

    astakos_token = ASTAKOS_TOKEN

    def get_machines(network):

        return network.machines.filter(deleted=False).count()

    def get_backends(network):

        return network.backend_networks.values_list("backend_id", flat=True)

    def get_subnet_ipv4(network):

        return _get_subnet_field(network, "cidr", 4)

    def get_subnet_ipv6(network):

        return _get_subnet_field(network, "cidr", 6)

    def get_gateway_ipv4(network):

        return _get_subnet_field(network, "gateway", 4)

    def get_gateway_ipv6(network):

        return _get_subnet_field(network, "gateway", 6)

    def get_subnets(network):

        return network.subnets.values_list('id', flat=True)

    FIELDS = {

        "id": ("id", "The ID of the network"),

        "name": ("name", "The name of the network"),

        "user.uuid": ("userid", "The UUID of the network's owner"),

        "public": ("public", "Whether network is public or private"),

        "flavor": ("flavor", "The network's flavor"),

        "state": ("state", "The network's state"),

        "subnets": (get_subnets, "The IDs of the associated subnets"),

        "subnet.ipv4":  (get_subnet_ipv4, "The IPv4 subnet of the network"),

        "gateway.ipv4": (get_gateway_ipv4, "The IPv4 gateway of the network"),

        "subnet.ipv6":  (get_subnet_ipv6, "The IPv6 subnet of the network"),

        "gateway.ipv6":  (get_gateway_ipv6, "The IPv6 gateway of the network"),

        "created": ("created", "The date the network was created"),

        "updated": ("updated", "The date the network was updated"),

        "deleted": ("deleted", "Whether the network is deleted or not"),

        "mode": ("mode", "The mode of the network"),

        "link": ("link", "The link of the network"),

        "mac_prefix": ("mac_prefix", "The network's MAC prefix"),

        "drained": ("drained", "Whether network is drained or not"),

        "vms": (get_machines, "Number of connected servers"),

        "backends": (get_backends, "IDs of Ganeti backends that the network is"

                                   " connected to"),

        "floating_ip_pool": ("floating_ip_pool",

                             "Whether the network is a floating IP pool"),

    }

    fields = ["id", "name", "user.uuid", "state", "public", "subnet.ipv4",

              "gateway.ipv4", "link", "mac_prefix",  "drained",

              "floating_ip_pool"]

    def handle_args(self, *args, **options):

        if options["public"]:

            self.filters["public"] = True

        if options["ipv6"]:

            self.fields.extend(["subnet.ipv6", "gateway.ipv6"])

def _get_subnet_field(network, field, version=4):

    for subnet in network.subnets.all():

        if subnet.ipversion == version:

            return getattr(subnet, field)

    return None

# Copyright 2012-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 optparse import make_option

from django.core.management.base import BaseCommand, CommandError

from synnefo.db.models import Backend

from synnefo.management.common import (get_network, get_backend)

from snf_django.management.utils import parse_bool

from synnefo.logic import networks, backend as backend_mod

from django.db import transaction

class Command(BaseCommand):

    args = "<network id>"

    help = "Modify a network."

    option_list = BaseCommand.option_list + (

        make_option(

            '--name',

            dest='name',

            metavar='NAME',

            help="Rename a network"),

        make_option(

            '--userid',

            dest='userid',

            help="Change the owner of the network."),

        make_option(

            "--drained",

            dest="drained",

            metavar="True|False",

            choices=["True", "False"],

            help="Set as drained to exclude for IP allocation."

                 " Only used for public networks."),

        make_option(

            "--floating-ip-pool",

            dest="floating_ip_pool",

            metavar="True|False",

            choices=["True", "False"],

            help="Convert network to a floating IP pool. During this"

                 " conversation the network will be created to all"

                 " available Ganeti backends."),

        make_option(

            '--add-reserved-ips',

            dest="add_reserved_ips",

            help="Comma seperated list of IPs to externally reserve."),

        make_option(

            '--remove-reserved-ips',

            dest="remove_reserved_ips",

            help="Comma seperated list of IPs to externally release."),

        make_option(

            "--add-to-backend",

            dest="add_to_backend",

            metavar="BACKEND_ID",

            help="Create a network to a Ganeti backend."),

        make_option(

            "--remove-from-backend",

            dest="remove_from_backend",

            metavar="BACKEND_ID",

            help="Remove a network from a Ganeti backend."),

    )

    @transaction.commit_on_success

    def handle(self, *args, **options):

        if len(args) != 1:

            raise CommandError("Please provide a network ID")

        network = get_network(args[0])

        new_name = options.get("name")

        if new_name is not None:

            old_name = network.name

            network = networks.rename(network, new_name)

            self.stdout.write("Renamed network '%s' from '%s' to '%s'.\n" %

                              (network, old_name, new_name))

        drained = options.get("drained")

        if drained is not None:

            drained = parse_bool(drained)

            network.drained = drained

            network.save()

            self.stdout.write("Set network '%s' as drained=%s.\n" %

                              (network, drained))

        new_owner = options.get("userid")

        if new_owner is not None:

            if "@" in new_owner:

                raise CommandError("Invalid owner UUID.")

            old_owner = network.userid

            network.userid = new_owner

            network.save()

            msg = "Changed the owner of network '%s' from '%s' to '%s'.\n"

            self.stdout.write(msg % (network, old_owner, new_owner))

        floating_ip_pool = options["floating_ip_pool"]

        if floating_ip_pool is not None:

            floating_ip_pool = parse_bool(floating_ip_pool)

            if floating_ip_pool is False and network.floating_ip_pool is True:

                if network.ips.filter(deleted=False, floating_ip=True)\

                              .exists():

                    msg = ("Cannot make network a non floating IP pool."

                           " There are still reserved floating IPs.")

                    raise CommandError(msg)

            network.floating_ip_pool = floating_ip_pool

            network.save()

            self.stdout.write("Set network '%s' as floating-ip-pool=%s.\n" %

                              (network, floating_ip_pool))

            if floating_ip_pool is True:

                for backend in Backend.objects.filter(offline=False):

                    bnet, jobs =\

                        backend_mod.ensure_network_is_active(backend,

                                                             network.id)

                    if jobs:

                        msg = ("Sent job to create network '%s' in backend"

                               " '%s'\n" % (network, backend))

                        self.stdout.write(msg)

        add_reserved_ips = options.get('add_reserved_ips')

        remove_reserved_ips = options.get('remove_reserved_ips')

        if add_reserved_ips or remove_reserved_ips:

            if add_reserved_ips:

                add_reserved_ips = add_reserved_ips.split(",")

                for ip in add_reserved_ips:

                    network.reserve_address(ip, external=True)

            if remove_reserved_ips:

                remove_reserved_ips = remove_reserved_ips.split(",")

                for ip in remove_reserved_ips:

                    network.release_address(ip, external=True)

        add_to_backend = options["add_to_backend"]

        if add_to_backend is not None:

            backend = get_backend(add_to_backend)

            bnet, jobs = backend_mod.ensure_network_is_active(backend,

                                                              network.id)

            if jobs:

                msg = "Sent job to create network '%s' in backend '%s'\n"

                self.stdout.write(msg % (network, backend))

        remove_from_backend = options["remove_from_backend"]

        if remove_from_backend is not None:

            backend = get_backend(remove_from_backend)

            if network.nics.filter(machine__backend=backend,

                                   machine__deleted=False).exists():

                msg = "Cannot remove. There are still connected VMs to this"\

                      " network"

                raise CommandError(msg)

            backend_mod.delete_network(network, backend, disconnect=True)

            msg = "Sent job to delete network '%s' from backend '%s'\n"

            self.stdout.write(msg % (network, backend))

# 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 THE REGENTS AND CONTRIBUTORS ``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 THE REGENTS 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 django.core.management.base import CommandError

from snf_django.management.commands import RemoveCommand

from snf_django.lib.api import faults

from synnefo.logic import networks

from synnefo.management.common import get_network, convert_api_faults

class Command(RemoveCommand):

    can_import_settings = True

    args = "<Network ID> [<Network ID> ...]"

    help = "Remove a network from the Database, and Ganeti"

    @convert_api_faults

    def handle(self, *args, **options):

        if not args:

            raise CommandError("Please provide a network ID")

        force = options['force']