RoleAdmin and Role model work when openstack is disable

[plstackapi.git] / plstackapi / core / models / role.py

diff --git a/plstackapi/core/models/role.py b/plstackapi/core/models/role.py
index 12743ef..41bfd62 100644 (file)
--- a/plstackapi/core/models/role.py
+++ b/plstackapi/core/models/role.py
@@ -3,13 +3,20 @@ import datetime
 from django.db import models
 from plstackapi.core.models import PlCoreBase
 
-# Create your models here.
-
 class Role(PlCoreBase):
 
-    ROLE_CHOICES = (('admin', 'Admin'), ('pi', 'Principle Investigator'), ('user','User'))
+    #ROLE_CHOICES = (('admin', 'Admin'), ('pi', 'Principle Investigator'), ('user','User'))
     role_id = models.CharField(max_length=256, unique=True)
-    role_type = models.CharField(max_length=80, unique=True, choices=ROLE_CHOICES)
+    role_type = models.CharField(max_length=80, unique=True)
 
     def __unicode__(self):  return u'%s' % (self.role_type)
 
+
+    def save(self, *args, **kwds):
+        self.os_manager.save_role(self)
+        super(Role, self).save(*args, **kwds)
+    
+    def delete(self, *args, **kwds):
+        self.os_manager.delete_role(self)   
+        super(Role, self).delete(*args, **kwds)
+