planetstack/core/models/site.py

   1 import os
   2 from django.db import models
   3 from core.models import PlCoreBase
   4 from core.models import Deployment
   5
   6 class Site(PlCoreBase):
   7
   8     tenant_id = models.CharField(null=True, blank=True, max_length=200, help_text="Keystone tenant id")
   9     name = models.CharField(max_length=200, help_text="Name for this Site")
  10     site_url = models.URLField(null=True, blank=True, max_length=512, help_text="Site's Home URL Page")
  11     enabled = models.BooleanField(default=True, help_text="Status for this Site")
  12     longitude = models.FloatField(null=True, blank=True)
  13     latitude = models.FloatField(null=True, blank=True)
  14     login_base = models.CharField(max_length=50, unique=True, help_text="Prefix for Slices associated with this Site")
  15     is_public = models.BooleanField(default=True, help_text="Indicates the visibility of this site to other members")
  16     abbreviated_name = models.CharField(max_length=80)
  17
  18     deployments = models.ManyToManyField(Deployment, blank=True, related_name='sites')
  19
  20     def __unicode__(self):  return u'%s' % (self.name)
  21
  22 class SitePrivilege(PlCoreBase):
  23
  24     user = models.ForeignKey('User', related_name='site_privileges')
  25     site = models.ForeignKey('Site', related_name='site_privileges')
  26     role = models.ForeignKey('Role')
  27
  28     def __unicode__(self):  return u'%s %s %s' % (self.site, self.user, self.role)
  29
  30     def save(self, *args, **kwds):
  31         if not hasattr(self, 'os_manager'):
  32             from openstack.manager import OpenStackManager
  33             setattr(self, 'os_manager', OpenStackManager())
  34             self.os_manager.driver.add_user_role(self.user.kuser_id, self.site.tenant_id, self.role.role_type)
  35         super(SitePrivilege, self).save(*args, **kwds)
  36
  37     def delete(self, *args, **kwds):
  38         if not hasattr(self, 'os_manager'):
  39             from openstack.manager import OpenStackManager
  40             setattr(self, 'os_manager', OpenStackManager())
  41             self.os_manager.driver.delete_user_role(self.user.kuser_id, self.site.tenant_id, self.role.role_type)
  42         super(SitePrivilege, self).delete(*args, **kwds)
  43
  44