planetstack/core/models/role.py

   1 import os
   2 import datetime
   3 from django.db import models
   4 from core.models import PlCoreBase
   5
   6 class Role(PlCoreBase):
   7
   8     ROLE_CHOICES = (('admin', 'Admin'), ('pi', 'Principle Investigator'), ('tech', 'Technician'), ('user','User'))
   9     role = models.CharField(null=True, blank=True,max_length=256, unique=True, choices=ROLE_CHOICES)
  10     role_type = models.CharField(max_length=80, unique=True)
  11
  12     def __unicode__(self):  return u'%s' % (self.role_type)
  13
  14
  15     def save(self, *args, **kwds):
  16         if not hasattr(self, 'os_manager'):
  17             from openstack.manager import OpenStackManager
  18             setattr(self, 'os_manager', OpenStackManager())
  19         self.os_manager.save_role(self)
  20         super(Role, self).save(*args, **kwds)
  21
  22     def delete(self, *args, **kwds):
  23         if not hasattr(self, 'os_manager'):
  24             from openstack.manager import OpenStackManager
  25             setattr(self, 'os_manager', OpenStackManager())
  26         self.os_manager.delete_role(self)
  27         super(Role, self).delete(*args, **kwds)
  28