update Sliver admin
[plstackapi.git] / plstackapi / core / models / sliver.py
1 import os
2 from django.db import models
3 from plstackapi.core.models import PlCoreBase
4 from plstackapi.core.models import Flavor
5 from plstackapi.core.models import Image
6 from plstackapi.core.models import Key
7 from plstackapi.core.models import Slice
8 from plstackapi.core.models import Node
9 from plstackapi.core.models import Site
10 from plstackapi.core.models import DeploymentNetwork
11 from plstackapi.openstack.driver import OpenStackDriver
12
13 # Create your models here.
14 class Sliver(PlCoreBase):
15     instance_id = models.CharField(max_length=200, help_text="Nova instance id")    
16     name = models.CharField(max_length=200, help_text="Sliver name")
17     flavor = models.ForeignKey(Flavor, related_name='slivers')
18     image = models.ForeignKey(Image, related_name='slivers')
19     key = models.ForeignKey(Key, related_name='slivers')
20     slice = models.ForeignKey(Slice, related_name='slivers')
21     node = models.ForeignKey(Node, related_name='slivers')
22     #site = models.ForeignKey(Site, 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         driver = OpenStackDriver()
29         if not self.instance_id:
30             instance = driver.spawn_instance(name=self.name,
31                                    key_name = self.key.name,
32                                    flavor_id = self.flavor.flavor_id,
33                                    image_id = self.image.image_id,
34                                    hostname = self.node.name )
35             self.instance_id = instance.id
36
37         super(Sliver, self).save(*args, **kwds)
38
39     def delete(self, *args, **kwds):
40         driver = OpenStackDriver()
41         if self.instance_id:
42             driver.destroy_instance(self.instance_id)
43
44         super(Sliver, self).delete(*args, **kwds)