plstackapi/core/models/role.py

   1 import os
   2 import datetime
   3 from django.db import models
   4 from plstackapi.core.models import PlCoreBase
   5 from plstackapi.openstack.driver import OpenStackDriver
   6
   7 # Create your models here.
   8
   9 class Role(PlCoreBase):
  10
  11     #ROLE_CHOICES = (('admin', 'Admin'), ('pi', 'Principle Investigator'), ('user','User'))
  12     role_id = models.CharField(max_length=256, unique=True, blank=True)
  13     role_type = models.CharField(max_length=80, unique=True)
  14
  15     def __unicode__(self):  return u'%s' % (self.role_type)
  16
  17
  18     def save(self, *args, **kwds):
  19
  20         driver = OpenStackDriver()
  21         if not self.role_id:
  22             keystone_role = driver.create_role(name=self.role_type)
  23             self.role_id = keystone_role.id
  24
  25         super(Role, self).save(*args, **kwds)
  26
  27     def delete(self, *args, **kwds):
  28         driver = OpenStackDriver()
  29         if self.role_id:
  30             driver.delete_role({'id': self.role_id})
  31
  32         super(Role, self).delete(*args, **kwds)
  33