save() and delete() uses caller's credentials to init openstackdriver client

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

diff --git a/plstackapi/core/models/key.py b/plstackapi/core/models/key.py
index d0a0a07..3e4f56f 100644 (file)
--- a/plstackapi/core/models/key.py
+++ b/plstackapi/core/models/key.py
@@ -1,15 +1,30 @@
 import os
 from django.db import models
 from plstackapi.core.models import PlCoreBase
-from plstackapi.core.models import User
+from plstackapi.core.models import PLUser
 
 # Create your models here.
 
 class Key(PlCoreBase):
     name = models.CharField(max_length=256, unique=True)
+    key_id = models.CharField(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(User, related_name='keys')
+    user = models.ForeignKey(PLUser, related_name='keys')
 
     def __unicode__(self):  return u'%s' % (self.name)
+
+    def save(self, *args, **kwds):
+        if not self.key_id:
+            key_fields = {'name': self.name,
+                          'key': self.key}
+            nova_key = self.driver.create_keypair(**key_fields)
+            self.key_id = nova_key.id
+        super(Key, self).save(*args, **kwds)
+
+    def delete(self, *args, **kwds):
+        if self.key_id:
+            self.driver.delete_keypair(self.key_id)
+        super(Key, self).delete(*args, **kwds) 
+