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 DeploymentNetwork
   5 from openstack.manager import OpenStackManager
   6
   7 class Site(PlCoreBase):
   8
   9     tenant_id = models.CharField(null=True, blank=True, max_length=200, help_text="Keystone tenant id")
  10     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
  23     def save(self, *args, **kwds):
  24         if not hasattr(self, 'os_manager'):
  25             setattr(self, 'os_manager', OpenStackManager())
  26             self.os_manager.save_site(self)
  27         super(Site, self).save(*args, **kwds)
  28
  29
  30     def delete(self, *args, **kwds):
  31         if not hasattr(self, 'os_manager'):
  32             setattr(self, 'os_manager', OpenStackManager())
  33             self.os_manager.delete_site(self)
  34         super(Site, self).delete(*args, **kwds)
  35
  36
  37 class SitePrivilege(PlCoreBase):
  38
  39     user = models.ForeignKey('User', related_name='site_privileges')
  40     site = models.ForeignKey('Site', related_name='site_privileges')
  41     role = models.ForeignKey('Role')
  42
  43     def __unicode__(self):  return u'%s %s %s' % (self.site, self.user, self.role)
  44
  45     def save(self, *args, **kwds):
  46         if not hasattr(self, 'os_manager'):
  47             setattr(self, 'os_manager', OpenStackManager())
  48             self.os_manager.driver.add_user_role(self.user.kuser_id, self.site.tenant_id, self.role.role_type)
  49         super(SitePrivilege, self).save(*args, **kwds)
  50
  51     def delete(self, *args, **kwds):
  52         if not hasattr(self, 'os_manager'):
  53             setattr(self, 'os_manager', OpenStackManager())
  54             self.os_manager.driver.delete_user_role(self.user.kuser_id, self.site.tenant_id, self.role.role_type)
  55         super(SitePrivilege, self).delete(*args, **kwds)
  56
  57