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
   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(DeploymentNetwork, blank=True, related_name='sites')
  19
  20     def __unicode__(self):  return u'%s' % (self.name)
  21
  22     def save(self, *args, **kwds):
  23         if not hasattr(self, 'os_manager'):
  24             from openstack.manager import OpenStackManager
  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             from openstack.manager import OpenStackManager
  33             setattr(self, 'os_manager', OpenStackManager())
  34         self.os_manager.delete_site(self)
  35         super(Site, self).delete(*args, **kwds)
  36
  37
  38 class SitePrivilege(PlCoreBase):
  39
  40     user = models.ForeignKey('User', related_name='site_privileges')
  41     site = models.ForeignKey('Site', related_name='site_privileges')
  42     role = models.ForeignKey('Role')
  43
  44     def __unicode__(self):  return u'%s %s %s' % (self.site, self.user, self.role)
  45
  46     def save(self, *args, **kwds):
  47         if not hasattr(self, 'os_manager'):
  48             from openstack.manager import OpenStackManager
  49             setattr(self, 'os_manager', OpenStackManager())
  50             self.os_manager.driver.add_user_role(self.user.kuser_id, self.site.tenant_id, self.role.role_type)
  51         super(SitePrivilege, self).save(*args, **kwds)
  52
  53     def delete(self, *args, **kwds):
  54         if not hasattr(self, 'os_manager'):
  55             from openstack.manager import OpenStackManager
  56             setattr(self, 'os_manager', OpenStackManager())
  57             self.os_manager.driver.delete_user_role(self.user.kuser_id, self.site.tenant_id, self.role.role_type)
  58         super(SitePrivilege, self).delete(*args, **kwds)
  59
  60