X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=planetstack%2Fcore%2Fmodels%2Fsliver.py;h=d9d2028b34619d3197221cfc3b99dc71e28acc04;hb=1e2a0962d5a15b1a1eabacb2ce837afada1dd848;hp=8ebcbd4dd555d25e97fe75849c8dec5c978b155d;hpb=732913407cc255ab864e39522585e1b98e654af1;p=plstackapi.git diff --git a/planetstack/core/models/sliver.py b/planetstack/core/models/sliver.py index 8ebcbd4..d9d2028 100644 --- a/planetstack/core/models/sliver.py +++ b/planetstack/core/models/sliver.py @@ -1,41 +1,143 @@ import os from django.db import models +from django.db.models import Q from django.core import exceptions -from core.models import PlCoreBase +from core.models import PlCoreBase,PlCoreBaseManager,PlCoreBaseDeletionManager from core.models import Image -from core.models import Key from core.models import Slice from core.models import Node from core.models import Site -from core.models import DeploymentNetwork -from openstack.manager import OpenStackManager +from core.models import Deployment +from core.models import User +from core.models import Tag +from core.models import Flavor +from django.contrib.contenttypes import generic +from planetstack.config import Config + +config = Config() + +def get_default_flavor(deployment = None): + # Find a default flavor that can be used for a sliver. This is particularly + # useful in evolution. It's also intended this helper function can be used + # for admin.py when users + + if deployment: + flavors = deployment.flavors.all() + else: + flavors = Flavor.objects.all() + + if not flavors: + return None + + for flavor in flavors: + if flavor.default: + return flavor + + return flavors[0] + +class SliverDeletionManager(PlCoreBaseDeletionManager): + def get_queryset(self): + parent=super(SliverDeletionManager, self) + try: + backend_type = config.observer_backend_type + except AttributeError: + backend_type = None + + parent_queryset = parent.get_queryset() if hasattr(parent, "get_queryset") else parent.get_query_set() + if (backend_type): + return parent_queryset.filter(Q(node__deployment__backend_type=backend_type)) + else: + return parent_queryset + + # deprecated in django 1.7 in favor of get_queryset(). + def get_query_set(self): + return self.get_queryset() + + +class SliverManager(PlCoreBaseManager): + def get_queryset(self): + parent=super(SliverManager, self) + + try: + backend_type = config.observer_backend_type + except AttributeError: + backend_type = None + + parent_queryset = parent.get_queryset() if hasattr(parent, "get_queryset") else parent.get_query_set() + + if backend_type: + return parent_queryset.filter(Q(node__deployment__backend_type=backend_type)) + else: + return parent_queryset + + # deprecated in django 1.7 in favor of get_queryset(). + def get_query_set(self): + return self.get_queryset() # Create your models here. class Sliver(PlCoreBase): - instance_id = models.CharField(null=True, blank=True, max_length=200, help_text="Nova instance id") + objects = SliverManager() + deleted_objects = SliverDeletionManager() + instance_id = models.CharField(null=True, blank=True, max_length=200, help_text="Nova instance id") name = models.CharField(max_length=200, help_text="Sliver name") instance_name = models.CharField(blank=True, null=True, max_length=200, help_text="OpenStack generated name") ip = models.GenericIPAddressField(help_text="Sliver ip address", blank=True, null=True) image = models.ForeignKey(Image, related_name='slivers') - key = models.ForeignKey(Key, related_name='slivers') + #key = models.ForeignKey(Key, related_name='slivers') + creator = models.ForeignKey(User, related_name='slivers', blank=True, null=True) slice = models.ForeignKey(Slice, related_name='slivers') node = models.ForeignKey(Node, related_name='slivers') - deploymentNetwork = models.ForeignKey(DeploymentNetwork, verbose_name='deployment', related_name='sliver_deploymentNetwork') + deploymentNetwork = models.ForeignKey(Deployment, verbose_name='deployment', related_name='sliver_deploymentNetwork') numberCores = models.IntegerField(verbose_name="Number of Cores", help_text="Number of cores for sliver", default=0) + flavor = models.ForeignKey(Flavor, help_text="Flavor of this instance", default=get_default_flavor) + tags = generic.GenericRelation(Tag) + userData = models.TextField(blank=True, null=True, help_text="user_data passed to instance during creation") - - def __unicode__(self): return u'%s' % (self.instance_name) + def __unicode__(self): + if self.instance_name: + return u'%s' % (self.instance_name) + elif self.id: + return u'uninstantiated-%s' % str(self.id) + elif self.slice: + return u'unsaved-sliver on %s' % self.slice.name + else: + return u'unsaved-sliver' def save(self, *args, **kwds): - if not self.name: - self.name = self.slice.name - if not hasattr(self, 'os_manager'): - setattr(self, 'os_manager', OpenStackManager()) - self.os_manager.save_sliver(self) + self.name = self.slice.slicename + if not self.creator and hasattr(self, 'caller'): + self.creator = self.caller + self.deploymentNetwork = self.node.deployment + +# XXX smbaker - disabled for now, was causing fault in tenant view create slice +# if not self.deploymentNetwork.test_acl(slice=self.slice): +# raise exceptions.ValidationError("Deployment %s's ACL does not allow any of this slice %s's users" % (self.deploymentNetwork.name, self.slice.name)) + super(Sliver, self).save(*args, **kwds) - def delete(self, *args, **kwds): - if not hasattr(self, 'os_manager'): - setattr(self, 'os_manager', OpenStackManager()) - self.os_manager.delete_sliver(self) - super(Sliver, self).delete(*args, **kwds) + def can_update(self, user): + return self.slice.can_update(user) + + def all_ips(self): + ips={} + for ns in self.networksliver_set.all(): + ips[ns.network.name] = ns.ip + return ips + + def all_ips_string(self): + result = [] + ips = self.all_ips() + for key in sorted(ips.keys()): + #result.append("%s = %s" % (key, ips[key])) + result.append(ips[key]) + return ", ".join(result) + all_ips_string.short_description = "addresses" + + @staticmethod + def select_by_user(user): + if user.is_admin: + qs = Sliver.objects.all() + else: + slices = Slice.select_by_user(user) + qs = Sliver.objects.filter(slice__in=slices) + return qs