planetstack/core/models/sliver.py

   1 import os
   2 from django.db import models
   3 from django.core import exceptions
   4 from core.models import PlCoreBase
   5 from core.models import Image
   6 from core.models import Slice
   7 from core.models import Node
   8 from core.models import Site
   9 from core.models import Deployment
  10 from core.models import User
  11 from core.models import Tag
  12 from core.models import Flavor
  13 from django.contrib.contenttypes import generic
  14
  15 def get_default_flavor(deployment = None):
  16     # Find a default flavor that can be used for a sliver. This is particularly
  17     # useful in evolution. It's also intended this helper function can be used
  18     # for admin.py when users
  19
  20     if deployment:
  21         flavors = deployment.flavors.all()
  22     else:
  23         flavors = Flavor.objects.all()
  24
  25     if not flavors:
  26         print "XXX none"
  27         return None
  28
  29     for flavor in flavors:
  30         if flavor.default:
  31             print "XXX", flavor
  32             return flavor
  33
  34     print "XXX", flavors[0]
  35     return flavors[0]
  36
  37 # Create your models here.
  38 class Sliver(PlCoreBase):
  39     instance_id = models.CharField(null=True, blank=True, max_length=200, help_text="Nova instance id")
  40     name = models.CharField(max_length=200, help_text="Sliver name")
  41     instance_name = models.CharField(blank=True, null=True, max_length=200, help_text="OpenStack generated name")
  42     ip = models.GenericIPAddressField(help_text="Sliver ip address", blank=True, null=True)
  43     image = models.ForeignKey(Image, related_name='slivers')
  44     #key = models.ForeignKey(Key, related_name='slivers')
  45     creator = models.ForeignKey(User, related_name='slivers', blank=True, null=True)
  46     slice = models.ForeignKey(Slice, related_name='slivers')
  47     node = models.ForeignKey(Node, related_name='slivers')
  48     deploymentNetwork = models.ForeignKey(Deployment, verbose_name='deployment', related_name='sliver_deploymentNetwork')
  49     numberCores = models.IntegerField(verbose_name="Number of Cores", help_text="Number of cores for sliver", default=0)
  50     flavor = models.ForeignKey(Flavor, help_text="Flavor of this instance", default=get_default_flavor)
  51     tags = generic.GenericRelation(Tag)
  52     userData = models.TextField(blank=True, null=True, help_text="user_data passed to instance during creation")
  53
  54     def __unicode__(self):
  55         if self.instance_name:
  56             return u'%s' % (self.instance_name)
  57         elif self.id:
  58             return u'uninstantiated-%s' % str(self.id)
  59         elif self.slice:
  60             return u'unsaved-sliver on %s' % self.slice.name
  61         else:
  62             return u'unsaved-sliver'
  63
  64     def save(self, *args, **kwds):
  65         if not self.name:
  66             self.name = self.slice.name
  67         if not self.creator and hasattr(self, 'caller'):
  68             self.creator = self.caller
  69         self.deploymentNetwork = self.node.deployment
  70
  71 # XXX smbaker - disabled for now, was causing fault in tenant view create slice
  72 #        if not self.deploymentNetwork.test_acl(slice=self.slice):
  73 #            raise exceptions.ValidationError("Deployment %s's ACL does not allow any of this slice %s's users" % (self.deploymentNetwork.name, self.slice.name))
  74
  75         super(Sliver, self).save(*args, **kwds)
  76
  77     def can_update(self, user):
  78         return self.slice.can_update(user)
  79
  80     def all_ips(self):
  81         ips={}
  82         for ns in self.networksliver_set.all():
  83            ips[ns.network.name] = ns.ip
  84         return ips
  85
  86     def all_ips_string(self):
  87         result = []
  88         ips = self.all_ips()
  89         for key in sorted(ips.keys()):
  90             #result.append("%s = %s" % (key, ips[key]))
  91             result.append(ips[key])
  92         return ", ".join(result)
  93     all_ips_string.short_description = "addresses"
  94
  95     @staticmethod
  96     def select_by_user(user):
  97         if user.is_admin:
  98             qs = Sliver.objects.all()
  99         else:
 100             slices = Slice.select_by_user(user)
 101             qs = Sliver.objects.filter(slice__in=slices)
 102         return qs