From ff9290ec6fcb12fffa4de2f4dcca341faf95a90a Mon Sep 17 00:00:00 2001
From: Sofia Papagiannaki <papagian@gmail.com>
Date: Mon, 21 May 2012 16:25:03 +0300
Subject: [PATCH] associate a superuser created with an AstakosUser

---
 .../astakos/im/fixtures/admin_user.json            |   11 --------
 snf-astakos-app/astakos/im/models.py               |   29 +++++++++++++++++++-
 2 files changed, 28 insertions(+), 12 deletions(-)
 delete mode 100644 snf-astakos-app/astakos/im/fixtures/admin_user.json

diff --git a/snf-astakos-app/astakos/im/fixtures/admin_user.json b/snf-astakos-app/astakos/im/fixtures/admin_user.json
deleted file mode 100644
index 1757ce3..0000000
--- a/snf-astakos-app/astakos/im/fixtures/admin_user.json
+++ /dev/null
@@ -1,11 +0,0 @@
-[
-    {
-        "model": "im.AstakosUser",
-        "pk": 1,
-        "fields": {
-            "level": 0,
-            "invitations": 10000,
-            "updated": "2012-01-24"
-   	}
-    }
-]
diff --git a/snf-astakos-app/astakos/im/models.py b/snf-astakos-app/astakos/im/models.py
index 657cc73..739849d 100644
--- a/snf-astakos-app/astakos/im/models.py
+++ b/snf-astakos-app/astakos/im/models.py
@@ -49,6 +49,7 @@ from django.core.exceptions import ValidationError
 from django.template.loader import render_to_string
 from django.core.mail import send_mail
 from django.db import transaction
+from django.db.models.signals import post_save, post_syncdb
 
 from astakos.im.settings import DEFAULT_USER_LEVEL, INVITATIONS_PER_LEVEL, \
 AUTH_TOKEN_DURATION, BILLING_FIELDS, QUEUE_CONNECTION, SITENAME, \
@@ -328,4 +329,30 @@ class EmailChange(models.Model):
 
     def activation_key_expired(self):
         expiration_date = timedelta(days=EMAILCHANGE_ACTIVATION_DAYS)
-        return self.requested_at + expiration_date < datetime.now()
\ No newline at end of file
+        return self.requested_at + expiration_date < datetime.now()
+
+def create_astakos_user(u):
+    try:
+        AstakosUser.objects.get(user_ptr=u.pk)
+    except AstakosUser.DoesNotExist:
+        extended_user = AstakosUser(user_ptr_id=u.pk)
+        extended_user.__dict__.update(u.__dict__)
+        extended_user.renew_token()
+        extended_user.save()
+    except:
+        pass
+
+def superuser_post_syncdb(sender, **kwargs):
+    # if there was created a superuser
+    # associate it with an AstakosUser
+    admins = User.objects.filter(is_superuser=True)
+    for u in admins:
+        create_astakos_user(u)
+
+post_syncdb.connect(superuser_post_syncdb)
+
+def superuser_post_save(sender, instance, **kwargs):
+    if instance.is_superuser:
+        create_astakos_user(instance)
+
+post_save.connect(superuser_post_save, sender=User)
\ No newline at end of file
-- 
1.7.10.4