create OSModelAdmin. Update some Admin definitions to subclass OSModelAdmin

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

diff --git a/plstackapi/core/models/role.py b/plstackapi/core/models/role.py
index 4b4b2e4..b9f3e2f 100644 (file)
--- a/plstackapi/core/models/role.py
+++ b/plstackapi/core/models/role.py
@@ -2,9 +2,6 @@ import os
 import datetime
 from django.db import models
 from plstackapi.core.models import PlCoreBase
-from plstackapi.openstack.driver import OpenStackDriver
-
-# Create your models here.
 
 class Role(PlCoreBase):
 
@@ -16,17 +13,13 @@ class Role(PlCoreBase):
 
 
     def save(self, *args, **kwds):
-        driver = OpenStackDriver() 
         if not self.role_id:
-            keystone_role = driver.create_role(name=self.role_type)
+            keystone_role = self.driver.create_role(name=self.role_type)
             self.role_id = keystone_role.id
-
         super(Role, self).save(*args, **kwds)
     
     def delete(self, *args, **kwds):
-        driver = OpenStackDriver()
         if self.role_id:
-            driver.delete_role({'id': self.role_id})   
-
+            self.driver.delete_role({'id': self.role_id})   
         super(Role, self).delete(*args, **kwds)