blob: 76cc0a4354a5c09387d0f1cd2acd171128efb08a [file] [log] [blame]
Siobhan Tully4bc09f22013-04-10 21:15:21 -04001import os
2from django.db import models
3from plstackapi.core.models import PlCoreBase
4from plstackapi.core.models import DeploymentNetwork
Siobhan Tully4bc09f22013-04-10 21:15:21 -04005
6
7class Site(PlCoreBase):
8
Tony Mackcdec0902013-04-15 00:38:49 -04009 tenant_id = models.CharField(max_length=200, help_text="Keystone tenant id")
Siobhan Tully4bc09f22013-04-10 21:15:21 -040010 name = models.CharField(max_length=200, help_text="Name for this Site")
11 site_url = models.URLField(null=True, blank=True, max_length=512, help_text="Site's Home URL Page")
12 enabled = models.BooleanField(default=True, help_text="Status for this Site")
13 longitude = models.FloatField(null=True, blank=True)
14 latitude = models.FloatField(null=True, blank=True)
15 login_base = models.CharField(max_length=50, unique=True, help_text="Prefix for Slices associated with this Site")
16 is_public = models.BooleanField(default=True, help_text="Indicates the visibility of this site to other members")
17 abbreviated_name = models.CharField(max_length=80)
18
19 deployments = models.ManyToManyField(DeploymentNetwork, blank=True, related_name='sites')
20
21 def __unicode__(self): return u'%s' % (self.name)
22
Tony Mackd9b46072013-04-14 13:51:19 -040023 def save(self, *args, **kwds):
Tony Mackd9b46072013-04-14 13:51:19 -040024 if not self.tenant_id:
Tony Mackfdd4d802013-04-27 13:02:33 -040025 tenant = self.driver.create_tenant(tenant_name=self.login_base,
26 description=self.name,
27 enabled=self.enabled)
Tony Mackd9b46072013-04-14 13:51:19 -040028 self.tenant_id = tenant.id
Tony Mack9bcbe4f2013-04-29 08:13:27 -040029 # give caller an admin role at the tenant they've created
30 self.driver.add_user_role(self.caller.user_id, tenant.id, 'admin')
31
Tony Mackd9b46072013-04-14 13:51:19 -040032 # update the record
33 if self.id:
Tony Mackfdd4d802013-04-27 13:02:33 -040034 self.driver.update_tenant(self.tenant_id,
Tony Mackfdd4d802013-04-27 13:02:33 -040035 description=self.name,
36 enabled=self.enabled)
Tony Mackd9b46072013-04-14 13:51:19 -040037
38 super(Site, self).save(*args, **kwds)
39
40
41 def delete(self, *args, **kwds):
Tony Mackd9b46072013-04-14 13:51:19 -040042 if self.tenant_id:
Tony Mackfdd4d802013-04-27 13:02:33 -040043 self.driver.delete_tenant(self.tenant_id)
Tony Mackd9b46072013-04-14 13:51:19 -040044 super(Site, self).delete(*args, **kwds)
45
46
Siobhan Tully4bc09f22013-04-10 21:15:21 -040047class SitePrivilege(PlCoreBase):
48
Siobhan Tully53437282013-04-26 19:30:27 -040049 user = models.ForeignKey('PLUser', related_name='site_privileges')
Siobhan Tully4bc09f22013-04-10 21:15:21 -040050 site = models.ForeignKey('Site', related_name='site_privileges')
51 role = models.ForeignKey('Role')
52
53 def __unicode__(self): return u'%s %s %s' % (self.site, self.user, self.role)
54
Tony Mack00d361f2013-04-28 10:28:42 -040055 def save(self, *args, **kwds):
56 self.driver.add_user_role(self.user.user_id, self.site.tenant_id, self.role.role_type)
57 super(SitePrivilege, self).save(*args, **kwds)
58
59 def delete(self, *args, **kwds):
60 self.driver.delete_user_role(self.user.user_id, self.site.tenant_id, self.role.role_type)
61 super(SitePrivilege, self).delete(*args, **kwds)
62
Siobhan Tully4bc09f22013-04-10 21:15:21 -040063