Tony Mack | 5cbec5d | 2013-04-02 07:55:11 -0400 | [diff] [blame] | 1 | import os |
Tony Mack | c7e6700 | 2013-04-08 16:17:02 -0400 | [diff] [blame] | 2 | from datetime import datetime |
Tony Mack | 7130ac3 | 2013-03-22 21:58:00 -0400 | [diff] [blame] | 3 | from django.db import models |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 4 | from plstackapi.openstack.driver import OpenStackDriver |
Tony Mack | 7130ac3 | 2013-03-22 21:58:00 -0400 | [diff] [blame] | 5 | |
| 6 | # Create your models here. |
| 7 | |
| 8 | class PlCoreBase(models.Model): |
| 9 | |
| 10 | created = models.DateTimeField(auto_now_add=True) |
| 11 | updated = models.DateTimeField(auto_now=True) |
| 12 | |
| 13 | class Meta: |
| 14 | abstract = True |
| 15 | |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 16 | def save(self): |
| 17 | if not self.id: |
| 18 | self.created = datetime.date.today() |
| 19 | self.updated = datetime.datetime.today() |
| 20 | super(PlCoreBase, self).save() |
| 21 | |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 22 | class Role(PlCoreBase): |
| 23 | |
| 24 | ROLE_CHOICES = (('admin', 'Admin'), ('pi', 'Principle Investigator'), ('user','User')) |
Tony Mack | f2162d9 | 2013-04-07 19:32:20 -0400 | [diff] [blame] | 25 | role_id = models.CharField(max_length=256, unique=True) |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 26 | role_type = models.CharField(max_length=80, unique=True, choices=ROLE_CHOICES) |
| 27 | |
| 28 | def __unicode__(self): return u'%s' % (self.role_type) |
| 29 | |
| 30 | def save(self): |
| 31 | if not self.id: |
| 32 | self.created = datetime.date.today() |
| 33 | self.updated = datetime.datetime.today() |
| 34 | super(Role, self).save() |
| 35 | |
Tony Mack | 7130ac3 | 2013-03-22 21:58:00 -0400 | [diff] [blame] | 36 | class Site(PlCoreBase): |
Tony Mack | d7cb771 | 2013-03-26 13:21:36 -0400 | [diff] [blame] | 37 | tenant_id = models.CharField(max_length=200, help_text="Keystone tenant id") |
Tony Mack | 67f60ce | 2013-04-02 10:46:07 -0400 | [diff] [blame] | 38 | name = models.CharField(max_length=200, help_text="Name for this Site") |
Tony Mack | c689d50 | 2013-04-02 11:32:04 -0400 | [diff] [blame] | 39 | site_url = models.URLField(null=True, blank=True, max_length=512, help_text="Site's Home URL Page") |
Tony Mack | 7130ac3 | 2013-03-22 21:58:00 -0400 | [diff] [blame] | 40 | enabled = models.BooleanField(default=True, help_text="Status for this Site") |
| 41 | longitude = models.FloatField(null=True, blank=True) |
| 42 | latitude = models.FloatField(null=True, blank=True) |
| 43 | login_base = models.CharField(max_length=50, help_text="Prefix for Slices associated with this Site") |
| 44 | is_public = models.BooleanField(default=True, help_text="Indicates the visibility of this site to other members") |
| 45 | abbreviated_name = models.CharField(max_length=80) |
| 46 | |
| 47 | def __unicode__(self): return u'%s' % (self.name) |
| 48 | |
Tony Mack | d7cb771 | 2013-03-26 13:21:36 -0400 | [diff] [blame] | 49 | def save(self, *args, **kwargs): |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 50 | driver = OpenStackDriver() |
Tony Mack | d7cb771 | 2013-03-26 13:21:36 -0400 | [diff] [blame] | 51 | if not self.id: |
Tony Mack | 39444e4 | 2013-04-02 10:22:24 -0400 | [diff] [blame] | 52 | tenant = driver.create_tenant(tenant_name=self.login_base, |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 53 | description=self.name, |
| 54 | enabled=self.enabled) |
Tony Mack | d8c390f | 2013-04-02 10:36:19 -0400 | [diff] [blame] | 55 | self.tenant_id = tenant.id |
Tony Mack | d7cb771 | 2013-03-26 13:21:36 -0400 | [diff] [blame] | 56 | else: |
| 57 | # update record |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 58 | self.driver.update_tenant(self.tenant_id, name=self.login_base, |
| 59 | description=self.name, enabled=self.enabled) |
Tony Mack | d7cb771 | 2013-03-26 13:21:36 -0400 | [diff] [blame] | 60 | super(Site, self).save(*args, **kwargs) |
| 61 | |
Tony Mack | e1c8c38 | 2013-03-26 13:37:59 -0400 | [diff] [blame] | 62 | def delete(self, *args, **kwds): |
| 63 | # delete keystone tenant |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 64 | driver = OpenStackDriver() |
| 65 | driver.delete_tenant(self.tenant_id) |
Tony Mack | e1c8c38 | 2013-03-26 13:37:59 -0400 | [diff] [blame] | 66 | super(Site, self).delete(*args, **kwargs) |
| 67 | |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 68 | class User(PlCoreBase): |
| 69 | user_id = models.CharField(max_length=256, unique=True) |
| 70 | firstname = models.CharField(help_text="person's given name", max_length=200) |
| 71 | lastname = models.CharField(help_text="person's surname", max_length=200) |
| 72 | email = models.EmailField(help_text="e-mail address") |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 73 | phone = models.CharField(null=True, blank=True, help_text="phone number contact", max_length=100) |
| 74 | user_url = models.URLField(null=True, blank=True) |
| 75 | is_admin = models.BooleanField(default=False) |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 76 | site = models.ForeignKey(Site, related_name='site_user', verbose_name="Site this user will be homed too") |
| 77 | |
| 78 | def __unicode__(self): return u'%s' % (self.email) |
| 79 | |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 80 | def save(self, *args, **kwds): |
| 81 | driver = OpenStackDriver() |
| 82 | name = self.email[:self.email.find('@')] |
| 83 | fields = {'name': name, 'email': self.email, 'password': self.password, |
| 84 | 'enabled': self.enabled} |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 85 | if not self.id: |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 86 | user = driver.create_user(**fields) |
| 87 | else: |
| 88 | driver.update_user(self.user_id, **fields) |
| 89 | super(User, self).save(*args, **kwds) |
| 90 | |
| 91 | def delete(self, *args, **kwds): |
| 92 | driver = OpenStackDriver() |
| 93 | driver.delete_user(self.user_id) |
| 94 | super(User, self).delete(*args, **kwds) |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 95 | |
| 96 | class SitePrivilege(PlCoreBase): |
| 97 | |
| 98 | user = models.ForeignKey('User') |
| 99 | site = models.ForeignKey('Site') |
| 100 | role = models.ForeignKey('Role') |
| 101 | |
| 102 | def __unicode__(self): return u'%s %s %s' % (self.site, self.user, self.role) |
| 103 | |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 104 | def save(self, *args, **kwds): |
| 105 | driver = OpenStackDriver() |
| 106 | driver.add_user_role(user_id=user.user_id, |
| 107 | tenant_id=site.tenant_id, |
| 108 | role_name=role.name) |
| 109 | super(SitePrivilege, self).save(*args, **kwds) |
| 110 | |
| 111 | def delete(self, *args, **kwds): |
| 112 | driver = OpenStackDriver() |
| 113 | driver.delete_user_role(user_id=user.user_id, |
| 114 | tenant_id=site.tenant_id, |
| 115 | role_name=role.name) |
| 116 | super(SitePrivilege, self).delete(*args, **kwds) |
| 117 | |
| 118 | |
| 119 | class DeploymentNetwork(PlCoreBase): |
| 120 | name = models.CharField(max_length=200, unique=True, help_text="Name of the Deployment Network") |
| 121 | |
| 122 | def __unicode__(self): return u'%s' % (self.name) |
| 123 | |
| 124 | class SiteDeploymentNetwork(PlCoreBase): |
| 125 | class Meta: |
| 126 | unique_together = ['site', 'deploymentNetwork'] |
| 127 | |
| 128 | site = models.ForeignKey(Site, related_name='deploymentNetworks') |
| 129 | deploymentNetwork = models.ForeignKey(DeploymentNetwork, related_name='sites') |
| 130 | name = models.CharField(default="Blah", max_length=100) |
| 131 | |
| 132 | |
| 133 | def __unicode__(self): return u'%s::%s' % (self.site, self.deploymentNetwork) |
| 134 | |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 135 | |
Tony Mack | 7130ac3 | 2013-03-22 21:58:00 -0400 | [diff] [blame] | 136 | class Slice(PlCoreBase): |
Tony Mack | d7cb771 | 2013-03-26 13:21:36 -0400 | [diff] [blame] | 137 | tenant_id = models.CharField(max_length=200, help_text="Keystone tenant id") |
Tony Mack | 7130ac3 | 2013-03-22 21:58:00 -0400 | [diff] [blame] | 138 | name = models.CharField(help_text="The Name of the Slice", max_length=80) |
Tony Mack | a3a40cf | 2013-04-02 11:21:21 -0400 | [diff] [blame] | 139 | enabled = models.BooleanField(default=True, help_text="Status for this Slice") |
Tony Mack | 7130ac3 | 2013-03-22 21:58:00 -0400 | [diff] [blame] | 140 | SLICE_CHOICES = (('plc', 'PLC'), ('delegated', 'Delegated'), ('controller','Controller'), ('none','None')) |
| 141 | instantiation = models.CharField(help_text="The instantiation type of the slice", max_length=80, choices=SLICE_CHOICES) |
| 142 | omf_friendly = models.BooleanField() |
| 143 | description=models.TextField(blank=True,help_text="High level description of the slice and expected activities", max_length=1024) |
Tony Mack | c689d50 | 2013-04-02 11:32:04 -0400 | [diff] [blame] | 144 | slice_url = models.URLField(blank=True, max_length=512) |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 145 | site = models.ForeignKey(Site, related_name='site_slice', help_text="The Site this Node belongs too") |
Tony Mack | af905fc | 2013-04-06 01:30:40 -0400 | [diff] [blame] | 146 | network_id = models.CharField(max_length=256, help_text="Quantum network") |
| 147 | router_id = models.CharField(max_length=256, help_text="Quantum router id") |
Tony Mack | 7130ac3 | 2013-03-22 21:58:00 -0400 | [diff] [blame] | 148 | |
| 149 | def __unicode__(self): return u'%s' % (self.name) |
| 150 | |
Tony Mack | d7cb771 | 2013-03-26 13:21:36 -0400 | [diff] [blame] | 151 | def save(self, *args, **kwds): |
Tony Mack | e1c8c38 | 2013-03-26 13:37:59 -0400 | [diff] [blame] | 152 | # sync keystone tenant |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 153 | driver = OpenStackDriver() |
| 154 | |
Tony Mack | d7cb771 | 2013-03-26 13:21:36 -0400 | [diff] [blame] | 155 | if not self.id: |
Tony Mack | 39444e4 | 2013-04-02 10:22:24 -0400 | [diff] [blame] | 156 | tenant = driver.create_tenant(tenant_name=self.name, |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 157 | description=self.description, |
| 158 | enabled=self.enabled) |
Tony Mack | d8c390f | 2013-04-02 10:36:19 -0400 | [diff] [blame] | 159 | self.tenant_id = tenant.id |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 160 | |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 161 | # create a network |
Tony Mack | af905fc | 2013-04-06 01:30:40 -0400 | [diff] [blame] | 162 | network = driver.create_network(name=self.name) |
| 163 | self.network_id = network['id'] |
| 164 | # create router |
| 165 | router = driver.create_router(name=self.name) |
| 166 | self.router_id = router['id'] |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 167 | |
Tony Mack | d7cb771 | 2013-03-26 13:21:36 -0400 | [diff] [blame] | 168 | else: |
| 169 | # update record |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 170 | self.driver.update_tenant(self.tenant_id, name=self.name, |
| 171 | description=self.description, enabled=self.enabled) |
Tony Mack | 6299cb7 | 2013-04-02 11:23:26 -0400 | [diff] [blame] | 172 | super(Slice, self).save(*args, **kwds) |
Tony Mack | d7cb771 | 2013-03-26 13:21:36 -0400 | [diff] [blame] | 173 | |
Tony Mack | e1c8c38 | 2013-03-26 13:37:59 -0400 | [diff] [blame] | 174 | def delete(self, *args, **kwds): |
| 175 | # delete keystone tenant |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 176 | driver = OpenStackDriver() |
| 177 | driver.delete_tenant(self.tenant_id) |
Tony Mack | 6299cb7 | 2013-04-02 11:23:26 -0400 | [diff] [blame] | 178 | super(Slice, self).delete(*args, **kwds) |
Tony Mack | e1c8c38 | 2013-03-26 13:37:59 -0400 | [diff] [blame] | 179 | |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 180 | class SliceMembership(PlCoreBase): |
| 181 | user = models.ForeignKey('User') |
| 182 | slice = models.ForeignKey('Slice') |
| 183 | role = models.ForeignKey('Role') |
| 184 | |
| 185 | def __unicode__(self): return u'%s %s %s' % (self.slice, self.user, self.role) |
| 186 | |
Tony Mack | 1fbdeca | 2013-04-06 00:10:36 -0400 | [diff] [blame] | 187 | def save(self, *args, **kwds): |
| 188 | driver = OpenStackDriver() |
| 189 | driver.add_user_role(user_id=user.user_id, |
| 190 | tenant_id=slice.tenant_id, |
| 191 | role_name=role.name) |
| 192 | super(SliceMembership, self).save(*args, **kwds) |
| 193 | |
| 194 | def delete(self, *args, **kwds): |
| 195 | driver = OpenStackDriver() |
| 196 | driver.delete_user_role(user_id=user.user_id, |
| 197 | tenant_id=slice.tenant_id, |
| 198 | role_name=role.name) |
| 199 | super(SliceMembership, self).delete(*args, **kwds) |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 200 | |
| 201 | class SubNet(PlCoreBase): |
| 202 | subnet_id = models.CharField(max_length=256, unique=True) |
| 203 | cidr = models.CharField(max_length=20) |
| 204 | ip_version = models.IntegerField() |
| 205 | start = models.IPAddressField() |
| 206 | end = models.IPAddressField() |
| 207 | slice = models.ForeignKey(Slice, related_name='slice_subnet') |
| 208 | |
| 209 | def __unicode__(self): return u'%s' % (self.name) |
| 210 | |
| 211 | def save(self, *args, **kwargs): |
| 212 | driver = OpenStackDriver() |
| 213 | if not self.id: |
| 214 | subnet = driver.create_subnet(network_name=self.slice.name, |
| 215 | cidr_ip = self.cidr, |
| 216 | ip_version=self.ip_version, |
| 217 | start = self.start, |
| 218 | end = self.end) |
| 219 | |
| 220 | self.subnet_id = subnet.id |
| 221 | |
Tony Mack | af905fc | 2013-04-06 01:30:40 -0400 | [diff] [blame] | 222 | # add subnet as interface to slice router |
| 223 | driver.add_router_interface(self.slice.router_id, subnet.id) |
| 224 | |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 225 | super(SubNet, self).save(*args, **kwargs) |
| 226 | |
| 227 | |
| 228 | def delete(self, *args, **kwargs): |
| 229 | # delete quantum network |
| 230 | driver = OpenStackDriver() |
| 231 | driver.delete_subnet(self.subnet_id) |
Tony Mack | af905fc | 2013-04-06 01:30:40 -0400 | [diff] [blame] | 232 | driver.delete_router_interface(self.slice.router_id, self.subnet.id) |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 233 | super(SubNet, self).delete(*args, **kwargs) |
| 234 | |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 235 | class Node(PlCoreBase): |
| 236 | name = models.CharField(max_length=200, unique=True, help_text="Name of the Node") |
| 237 | siteDeploymentNetwork = models.ForeignKey(SiteDeploymentNetwork, help_text="The Site and Deployment Network this Node belongs too.") |
| 238 | |
| 239 | def __unicode__(self): return u'%s' % (self.name) |
| 240 | |
| 241 | class Image(PlCoreBase): |
| 242 | image_id = models.CharField(max_length=256, unique=True) |
| 243 | name = models.CharField(max_length=256, unique=True) |
| 244 | disk_format = models.CharField(max_length=256) |
| 245 | container_format = models.CharField(max_length=256) |
| 246 | |
| 247 | def __unicode__(self): return u'%s' % (self.name) |
| 248 | |
| 249 | |
| 250 | class Flavor(PlCoreBase): |
| 251 | flavor_id = models.IntegerField(unique=True) |
| 252 | name = models.CharField(max_length=256, unique=True) |
| 253 | memory_mb = models.IntegerField() |
| 254 | disk_gb = models.IntegerField() |
| 255 | vcpus = models.IntegerField() |
| 256 | |
| 257 | def __unicode__(self): return u'%s' % (self.name) |
| 258 | |
| 259 | class Key(PlCoreBase): |
| 260 | name = models.CharField(max_length=256, unique=True) |
| 261 | key = models.CharField(max_length=512) |
| 262 | type = models.CharField(max_length=256) |
| 263 | blacklisted = models.BooleanField() |
| 264 | user = models.ForeignKey(User) |
| 265 | |
| 266 | def __unicode__(self): return u'%s' % (self.name) |
| 267 | |
| 268 | def save(self, *args, **kwds): |
| 269 | driver = OpenStackDriver() |
| 270 | if not self.id: |
| 271 | keypair = driver.create_keypair(name=self.name, key=self.key) |
| 272 | super(Key, self).save(*args, **kwds) |
| 273 | |
| 274 | def delete(self, *args, **kwds): |
| 275 | driver = OpenStackDriver() |
| 276 | driver.delete_keypair(self.name) |
| 277 | super(Key, self).delete(*args, **kwds) |
| 278 | |
| 279 | |
| 280 | |
Tony Mack | 7130ac3 | 2013-03-22 21:58:00 -0400 | [diff] [blame] | 281 | class Sliver(PlCoreBase): |
Tony Mack | ebb2487 | 2013-04-02 08:20:59 -0400 | [diff] [blame] | 282 | instance_id = models.CharField(max_length=200, help_text="Nova instance id") |
Tony Mack | d9ae9da | 2013-04-02 08:58:22 -0400 | [diff] [blame] | 283 | name = models.CharField(max_length=200, help_text="Sliver name") |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 284 | flavor = models.ForeignKey(Flavor, related_name='sliver_flavor') |
| 285 | image = models.ForeignKey(Image, related_name='sliver_image') |
| 286 | key = models.ForeignKey(Key, related_name='sliver_key') |
Tony Mack | 27527b9 | 2013-04-02 08:59:35 -0400 | [diff] [blame] | 287 | slice = models.ForeignKey(Slice, related_name='sliver_slice') |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 288 | siteDeploymentNetwork = models.ForeignKey(SiteDeploymentNetwork, related_name='sliver_deployment') |
| 289 | node = models.ForeignKey(Node, related_name='sliver_node') |
Tony Mack | 7130ac3 | 2013-03-22 21:58:00 -0400 | [diff] [blame] | 290 | |
| 291 | def __unicode__(self): return u'%s::%s' % (self.slice, self.siteDeploymentNetwork) |
| 292 | |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 293 | def save(self, *args, **kwds): |
| 294 | driver = OpenStackDriver() |
| 295 | instance = driver.spawn_instances(name=self.name, |
| 296 | keyname=self.name, |
| 297 | hostnames=self.node.name, |
Tony Mack | 3d1f069 | 2013-04-04 23:55:54 -0400 | [diff] [blame] | 298 | flavor=self.flavor.name, |
| 299 | image=self.image.name) |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 300 | self.instance_id = instance.id |
| 301 | super(Sliver, self).save(*args, **kwds) |
| 302 | |
| 303 | def delete(self, *args, **kwds): |
| 304 | driver = OpenStackDriver() |
| 305 | driver.destroy_instance(name=self.name, id=self.instance_id) |
| 306 | super(Sliver, self).delete(*args, **kwds) |
Tony Mack | 7823b89 | 2013-03-29 20:35:26 -0400 | [diff] [blame] | 307 | |