d509cc0707dcfb32bace28b0bd27c8503203eeae
[plstackapi.git] / plstackapi / core / models / slice.py
1 import os
2 from django.db import models
3 from plstackapi.core.models import PlCoreBase
4 from plstackapi.core.models import Site
5 from plstackapi.core.models import PLUser
6 from plstackapi.core.models import Role
7 from plstackapi.core.models import DeploymentNetwork
8
9 # Create your models here.
10
11 class Slice(PlCoreBase):
12     tenant_id = models.CharField(max_length=200, help_text="Keystone tenant id")
13     name = models.CharField(unique=True, help_text="The Name of the Slice", max_length=80)
14     enabled = models.BooleanField(default=True, help_text="Status for this Slice")
15     SLICE_CHOICES = (('plc', 'PLC'), ('delegated', 'Delegated'), ('controller','Controller'), ('none','None'))
16     instantiation = models.CharField(help_text="The instantiation type of the slice", max_length=80, choices=SLICE_CHOICES)
17     omf_friendly = models.BooleanField()
18     description=models.TextField(blank=True,help_text="High level description of the slice and expected activities", max_length=1024)
19     slice_url = models.URLField(blank=True, max_length=512)
20     site = models.ForeignKey(Site, related_name='slices', help_text="The Site this Node belongs too")
21     network_id = models.CharField(max_length=256, help_text="Quantum network")
22     router_id = models.CharField(max_length=256, help_text="Quantum router id")
23
24     SVC_CLASS_CHOICES = (('besteffort', 'Best Effort'), ('silver', 'Silver'), ('gold','Gold'))
25     serviceClass = models.CharField(verbose_name="Service Class",default="besteffort",help_text="The Service Class of this slice", max_length=30, choices=SVC_CLASS_CHOICES)
26
27
28     def __unicode__(self):  return u'%s' % (self.name)
29
30     def save(self, *args, **kwds):
31         self.os_manager.save_slice(self)
32         super(Slice, self).save(*args, **kwds)
33
34     def delete(self, *args, **kwds):
35         self.os_manager.delete_slice(self)
36         super(Slice, self).delete(*args, **kwds)    
37
38 class SliceMembership(PlCoreBase):
39     user = models.ForeignKey('PLUser', related_name='slice_memberships')
40     slice = models.ForeignKey('Slice', related_name='slice_memberships')
41     role = models.ForeignKey('Role')
42
43     def __unicode__(self):  return u'%s %s %s' % (self.slice, self.user, self.role)
44
45     def save(self, *args, **kwds):
46         self.os_manager.driver.add_user_role(self.user.user_id, self.slice.tenant_id, self.role.role_type)
47         super(SliceMembership, self).save(*args, **kwds)
48
49     def delete(self, *args, **kwds):
50         self.os_manager.driver.delete_user_role(self.user.user_id, self.slice.tenant_id, self.role.role_type)
51         super(SliceMembership, self).delete(*args, **kwds)