2 from django.db import models
3 from plstackapi.openstack.driver import OpenStackDriver
5 os.environ.setdefault("DJANGO_SETTINGS_MODULE", "plstackapi.planetstack.settings")
6 # Create your models here.
8 class PlCoreBase(models.Model):
10 created = models.DateTimeField(auto_now_add=True)
11 updated = models.DateTimeField(auto_now=True)
16 class Site(PlCoreBase):
17 tenant_id = models.CharField(max_length=200, help_text="Keystone tenant id")
18 name = models.CharField(max_length=200, unique=True, help_text="Name for this Site")
19 site_url = models.URLField(help_text="Site's Home URL Page")
20 enabled = models.BooleanField(default=True, help_text="Status for this Site")
21 longitude = models.FloatField(null=True, blank=True)
22 latitude = models.FloatField(null=True, blank=True)
23 login_base = models.CharField(max_length=50, help_text="Prefix for Slices associated with this Site")
24 is_public = models.BooleanField(default=True, help_text="Indicates the visibility of this site to other members")
25 abbreviated_name = models.CharField(max_length=80)
27 def __unicode__(self): return u'%s' % (self.name)
29 def save(self, *args, **kwargs):
30 driver = OpenStackDriver()
32 tenant = driver.create_tenant(name=self.login_base,
33 description=self.name,
35 self.tenant_id = tenants.id
38 self.driver.update_tenant(self.tenant_id, name=self.login_base,
39 description=self.name, enabled=self.enabled)
40 super(Site, self).save(*args, **kwargs)
42 def delete(self, *args, **kwds):
43 # delete keystone tenant
44 driver = OpenStackDriver()
45 driver.delete_tenant(self.tenant_id)
46 super(Site, self).delete(*args, **kwargs)
48 class Slice(PlCoreBase):
49 tenant_id = models.CharField(max_length=200, help_text="Keystone tenant id")
50 name = models.CharField(help_text="The Name of the Slice", max_length=80)
51 SLICE_CHOICES = (('plc', 'PLC'), ('delegated', 'Delegated'), ('controller','Controller'), ('none','None'))
52 instantiation = models.CharField(help_text="The instantiation type of the slice", max_length=80, choices=SLICE_CHOICES)
53 omf_friendly = models.BooleanField()
54 description=models.TextField(blank=True,help_text="High level description of the slice and expected activities", max_length=1024)
55 slice_url = models.URLField(blank=True)
56 site = models.ForeignKey(Site, related_name='slices', help_text="The Site this Node belongs too")
58 def __unicode__(self): return u'%s' % (self.name)
60 def save(self, *args, **kwds):
61 # sync keystone tenant
62 driver = OpenStackDriver()
64 tenant_fields = {'name': self.name,
65 'enabled': self.enabled,
66 'description': self.description}
68 tenant = driver.create_tenant(name=self.name,
69 description=self.description,
71 self.tenant_id = tenants.id
74 self.driver.update_tenant(self.tenant_id, name=self.name,
75 description=self.description, enabled=self.enabled)
76 super(Slice, self).save(*args, **kwargs)
78 def delete(self, *args, **kwds):
79 # delete keystone tenant
80 driver = OpenStackDriver()
81 driver.delete_tenant(self.tenant_id)
82 super(Slice, self).delete(*args, **kwargs)
84 class DeploymentNetwork(PlCoreBase):
85 name = models.CharField(max_length=200, unique=True, help_text="Name of the Deployment Network")
87 def __unicode__(self): return u'%s' % (self.name)
89 class SiteDeploymentNetwork(PlCoreBase):
91 unique_together = ['site', 'deploymentNetwork']
93 site = models.ForeignKey(Site, related_name='deploymentNetworks')
94 deploymentNetwork = models.ForeignKey(DeploymentNetwork, related_name='sites')
95 name = models.CharField(default="Blah", max_length=100)
98 def __unicode__(self): return u'%s::%s' % (self.site, self.deploymentNetwork)
101 class Sliver(PlCoreBase):
102 tenant_id = models.CharField(help_text="Keystone tenant id")
103 instance_id = models.CharField()
104 name = models.CharField()
105 flavor = models.CharField()
106 image = models.CharField()
107 slice = models.ForeignKey(Slice, related_name='slice')
108 siteDeploymentNetwork = models.ForeignKey(SiteDeploymentNetwork)
109 node = models.ForeignKey(Node, related_name='node')
111 def __unicode__(self): return u'%s::%s' % (self.slice, self.siteDeploymentNetwork)
113 def save(self, *args, **kwds):
114 driver = OpenStackDriver()
115 instance = driver.spawn_instances(name=self.name,
117 hostnames=self.node.name,
120 self.instance_id = instance.id
121 super(Sliver, self).save(*args, **kwds)
123 def delete(self, *args, **kwds):
124 driver = OpenStackDriver()
125 driver.destroy_instance(name=self.name, id=self.instance_id)
126 super(Sliver, self).delete(*args, **kwds)
129 class Node(PlCoreBase):
130 name = models.CharField(max_length=200, unique=True, help_text="Name of the Node")
131 siteDeploymentNetwork = models.ForeignKey(SiteDeploymentNetwork, help_text="The Site and Deployment Network this Node belongs too.")
133 def __unicode__(self): return u'%s' % (self.name)
135 class Network(PlCoreBase):
136 slice = models.ForeignKey(Slice, related_name='slice')
137 name = models.CharField(max_length=200, unique=True)
138 quantum_id = models.CharField(max_length=200, unique=True)
140 def __unicode__(self): return u'%s' % (self.name)
142 def save(self, *args, **kwargs):
143 os_shell = OpenStackShell()
144 network_fields = {'name': self.name}
147 # check if quantum network record exists
148 networks = os_shell.quantum.list_networks(name=self.name)
150 network = os_shell.quantum.create_network(name=self.name,
151 admin_state_up=False)
153 network = networks[0]
154 self.quantum_id = network.id
155 super(Network, self).save(*args, **kwargs)
157 def delete(self, *args, **kwargs):
158 # delete quantum network
159 os_shell = OpenStackShell()
160 os_shell.quantum.delete_network(self.quantum_id)
162 super(Network, self).delete(*args, **kwargs)
164 class SubNet(PlCoreBase):
165 network = models.ForeignKey(Network, related_name='network')
166 name = models.CharField(max_length=200, unique=True)
167 quantum_id = models.CharField(max_length=200, unique=True)
168 cidr = models.CharField(max_length=20)
169 ip_version = models.IntegerField()
170 start = models.IPAddressField()
171 end = models.IPAddressField()
173 def __unicode__(self): return u'%s' % (self.name)
175 def save(self, *args, **kwargs):
176 os_shell = OpenStackShell()
177 subnet_fields = {'network_id': self.network.quantum_id,
179 'ip_version': self.ip_version,
181 'allocation_pools': {'start': self.start,
185 subnets = os_shell.quantum.list_subnets(name=self.name)
187 subnet = os_shell.quantum.create_subnet(**subnet_fields)
190 self.quantum_id = subnet.id
191 super(SubNet, self).save(*args, **kwargs)
194 def delete(self, *args, **kwargs):
195 # delete quantum network
196 os_shell = OpenStackShell()
197 os_shell.quantum.delete_subnet(self.quantum_id)
199 super(SubNet, self).delete(*args, **kwargs)