fix NameError

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

diff --git a/plstackapi/core/models/user.py b/plstackapi/core/models/user.py
index 765fb62..30bd2a9 100644 (file)
--- a/plstackapi/core/models/user.py
+++ b/plstackapi/core/models/user.py
@@ -3,6 +3,8 @@ import datetime
 from django.db import models
 from plstackapi.core.models import PlCoreBase
 from plstackapi.core.models import Site
+from plstackapi.openstack.driver import OpenStackDriver
+
 
 # Create your models here.
 
@@ -10,7 +12,9 @@ class User(PlCoreBase):
     user_id = models.CharField(max_length=256, unique=True)
     firstname = models.CharField(help_text="person's given name", max_length=200)
     lastname = models.CharField(help_text="person's surname", max_length=200)
-    email = models.EmailField(help_text="e-mail address")
+    email = models.EmailField(help_text="e-mail address", null=True)
+    password = models.CharField(max_length=256, null=True, blank=True)
+    
     phone = models.CharField(null=True, blank=True, help_text="phone number contact", max_length=100)
     user_url = models.URLField(null=True, blank=True)
     is_admin = models.BooleanField(default=False)
@@ -18,3 +22,23 @@ class User(PlCoreBase):
     site = models.ForeignKey(Site, related_name='users', verbose_name="Site this user will be homed too")
 
     def __unicode__(self):  return u'%s' % (self.email)
+
+    def save(self, *args, **kwds):
+        driver  = OpenStackDriver()
+        if not self.user_id:
+            name = self.email[:self.email.find('@')]
+            user_fields = {'name': name,
+                           'email': self.email,
+                           'password': self.password,
+                           'enabled': self.enabled}
+            user = driver.create_user(**user_fields)
+            self.user_id = user.id
+
+        self.password = None
+        super(User, self).save(*args, **kwds)   
+
+    def delete(self, *args, **kwds):
+        driver = OpenStackDriver()
+        if self.user_id:
+            driver.delete_user(self.user_id)
+        super(User, self).delete(*args, **kwds)