tools: Add sj_user DB table migration

1 files changed, 50 insertions, 3 deletions

diff --git a/tools/migrate_db.py b/tools/migrate_db.py
index 16bf30c..c47ddee 100755
--- a/tools/migrate_db.py
+++ b/tools/migrate_db.py
@@ -1,8 +1,26 @@
 #!/usr/bin/env python
 
+import os
+import sys
+
+sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
+os.environ['DJANGO_SETTINGS_MODULE'] = 'scannerjammer.settings'
+
 from db import db as DB
 from pprint import pprint
 
+import django
+from datetime import datetime
+
+from django.conf import settings
+from django.contrib.auth.models import User
+
+from backend.models import SJLike
+from backend.models import SJRoom
+from backend.models import SJContent
+from backend.models import SJSearch
+from backend.models import SJUserProfile
+
 class MigrateDB(DB):
 
     def __init__(self, *args, **kwargs):
@@ -54,7 +72,36 @@ class MigrateDB(DB):
 if __name__ == '__main__':
     db = MigrateDB()
     db.connect()
+
+    # migrate sj_user table
     for row in db.get_sj_user_table():
-        for key, value in row.iteritems():
-             print key, ', ', value, ', ', type(value)
-        print
+        try:
+            user = User.objects.get(username=row['username'])
+        except User.DoesNotExist:
+            user = User.objects.create_user(
+                row['username'], 
+                '%s@example.com' % row['username'],
+                row['password']
+            )
+        try:
+            user_profile = user.get_profile()
+        except django.core.exceptions.ObjectDoesNotExist:
+            user_profile = SJUserProfile(user=user)
+            user_profile.lastseen_chat = datetime.fromtimestamp(0)
+            user_profile.lastseen_webcam = datetime.fromtimestamp(0)
+            user_profile.save()
+
+        user.date_joined = datetime.fromtimestamp(row['joindate'])
+        user.last_login = datetime.fromtimestamp(row['seendate'])
+        user.save()
+
+        user_profile.nickname = row['nickname'] or ''
+        user_profile.score = row['score']
+        user_profile.access = row['access']
+        user_profile.bio = row['bio'] or ''
+        user_profile.lastseen_chat = datetime.fromtimestamp(row['seendate'])
+        user_profile.save()
+
+        #for key, value in row.iteritems():
+        #     print key, ', ', value, ', ', type(value)
+        #print