plstackapi/core/models/sliver.py

   1 import os
   2 from django.db import models
   3 from django.core import exceptions
   4 from plstackapi.core.models import PlCoreBase
   5 from plstackapi.core.models import Flavor
   6 from plstackapi.core.models import Image
   7 from plstackapi.core.models import Key
   8 from plstackapi.core.models import Slice
   9 from plstackapi.core.models import Node
  10 from plstackapi.core.models import Site
  11 from plstackapi.core.models import DeploymentNetwork
  12 from plstackapi.openstack.driver import OpenStackDriver
  13
  14 # Create your models here.
  15 class Sliver(PlCoreBase):
  16     instance_id = models.CharField(max_length=200, help_text="Nova instance id")
  17     name = models.CharField(max_length=200, help_text="Sliver name")
  18     ip = models.GenericIPAddressField(help_text="Sliver ip address", blank=True, null=True)
  19     flavor = models.ForeignKey(Flavor, related_name='slivers')
  20     image = models.ForeignKey(Image, related_name='slivers')
  21     key = models.ForeignKey(Key, related_name='slivers')
  22     slice = models.ForeignKey(Slice, related_name='slivers')
  23     node = models.ForeignKey(Node, related_name='slivers')
  24     deploymentNetwork = models.ForeignKey(DeploymentNetwork, related_name='sliver_deploymentNetwork')
  25
  26     def __unicode__(self):  return u'%s::%s' % (self.slice, self.deploymentNetwork)
  27
  28     def save(self, *args, **kwds):
  29         if not self.slice.subnet.exists():
  30             raise exceptions.ValidationError, "Slice %s has no subnet" % self.slice.name
  31
  32         if not self.instance_id:
  33             driver = OpenStackDriver()
  34             instance = driver.spawn_instance(name=self.name,
  35                                    key_name = self.key.name,
  36                                    flavor_id = self.flavor.flavor_id,
  37                                    image_id = self.image.image_id,
  38                                    hostname = self.node.name )
  39             self.instance_id = instance.id
  40
  41         super(Sliver, self).save(*args, **kwds)
  42
  43     def delete(self, *args, **kwds):
  44         driver = OpenStackDriver()
  45         if self.instance_id:
  46             driver.destroy_instance(self.instance_id)
  47
  48         super(Sliver, self).delete(*args, **kwds)