plstackapi/core/models/key.py

import os
from django.db import models
from plstackapi.core.models import PlCoreBase
from plstackapi.core.models import PLUser

# Create your models here.

class Key(PlCoreBase):
    key_id = models.CharField(null=True, blank=True, max_length=256, unique=True)
    key = models.CharField(max_length=512)
    type = models.CharField(max_length=256)
    blacklisted = models.BooleanField(default=False)
    user = models.ForeignKey(PLUser, related_name='keys')

    def __unicode__(self):  return u'%s' % (self.user.email[:self.user.email.find('@')])

    def save(self, *args, **kwds):
        self.os_manager.save_key(self)
        super(Key, self).save(*args, **kwds)

    def delete(self, *args, **kwds):
        self.os_manager.delete_key(self)
        super(Key, self).delete(*args, **kwds)