X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=plstackapi%2Fcore%2Fmodels%2Fsite.py;h=ef7db95d093b6e71faadcbd11cbf12aa5c716444;hb=00d361f893df2241849d6e6ef3fe31e50087b5e0;hp=67b3e52d65a50023636f1b64e0898aa57dd188ae;hpb=53106f3b838a34c37bfbb971826903802b62c0ff;p=plstackapi.git

diff --git a/plstackapi/core/models/site.py b/plstackapi/core/models/site.py
index 67b3e52..ef7db95 100644
--- a/plstackapi/core/models/site.py
+++ b/plstackapi/core/models/site.py
@@ -50,4 +50,12 @@ class SitePrivilege(PlCoreBase):
 
     def __unicode__(self):  return u'%s %s %s' % (self.site, self.user, self.role)
 
+    def save(self, *args, **kwds):
+        self.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):
+        self.driver.delete_user_role(self.user.user_id, self.site.tenant_id, self.role.role_type)
+        super(SitePrivilege, self).delete(*args, **kwds)
+