from plstackapi.core.models import PlCoreBase
from plstackapi.core.models import DeploymentNetwork
-from plstackapi.openstack.driver import OpenStackDriver
-
-# Create your models here.
-
class Site(PlCoreBase):
def __unicode__(self): return u'%s' % (self.name)
+ def save(self, *args, **kwds):
+ self.os_manager.save_site(self)
+ super(Site, self).save(*args, **kwds)
+
+
+ def delete(self, *args, **kwds):
+ self.os_manager.delete_site(self)
+ super(Site, self).delete(*args, **kwds)
+
+
class SitePrivilege(PlCoreBase):
- user = models.ForeignKey('User', related_name='site_privileges')
+ user = models.ForeignKey('PLUser', related_name='site_privileges')
site = models.ForeignKey('Site', related_name='site_privileges')
role = models.ForeignKey('Role')
def __unicode__(self): return u'%s %s %s' % (self.site, self.user, self.role)
def save(self, *args, **kwds):
- driver = OpenStackDriver()
- driver.add_user_role(user_id=user.user_id,
- tenant_id=site.tenant_id,
- role_name=role.name)
+ self.os_manager.driver.add_user_role(self.user.user_id, self.site.tenant_id, self.role.role_type)
super(SitePrivilege, self).save(*args, **kwds)
def delete(self, *args, **kwds):
- driver = OpenStackDriver()
- driver.delete_user_role(user_id=user.user_id,
- tenant_id=site.tenant_id,
- role_name=role.name)
+ self.os_manager.driver.delete_user_role(self.user.user_id, self.site.tenant_id, self.role.role_type)
super(SitePrivilege, self).delete(*args, **kwds)