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     flavor = models.ForeignKey(Flavor, related_name='slivers')
  19     image = models.ForeignKey(Image, related_name='slivers')
  20     key = models.ForeignKey(Key, related_name='slivers')
  21     slice = models.ForeignKey(Slice, related_name='slivers')
  22     node = models.ForeignKey(Node, related_name='slivers')
  23     deploymentNetwork = models.ForeignKey(DeploymentNetwork, related_name='sliver_deploymentNetwork')
  24
  25     def __unicode__(self):  return u'%s::%s' % (self.slice, self.deploymentNetwork)
  26
  27     def save(self, *args, **kwds):
  28         if not self.slice.subnet:
  29             raise exceptions.ValidationError, "Slice %s has no subnet" % self.slice.name
  30
  31         if not self.instance_id:
  32             driver = OpenStackDriver()
  33             instance = driver.spawn_instance(name=self.name,
  34                                    key_name = self.key.name,
  35                                    flavor_id = self.flavor.flavor_id,
  36                                    image_id = self.image.image_id,
  37                                    hostname = self.node.name )
  38             self.instance_id = instance.id
  39
  40         super(Sliver, self).save(*args, **kwds)
  41
  42     def delete(self, *args, **kwds):
  43         driver = OpenStackDriver()
  44         if self.instance_id:
  45             driver.destroy_instance(self.instance_id)
  46
  47         super(Sliver, self).delete(*args, **kwds)