migrate tools

1 files changed, 144 insertions, 0 deletions

diff --git a/tools/migrate_user.py b/tools/migrate_user.py
new file mode 100755
index 0000000..bf941f6
--- /dev/null
+++ b/tools/migrate_user.py
@@ -0,0 +1,144 @@
+#!/usr/bin/env python
+
+import os
+import sys
+import chardet
+import json
+
+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
+
+from backend.views import thraw
+from backend.views import freeze
+
+class MigrateDB(DB):
+
+    def __init__(self, *args, **kwargs):
+        super(MigrateDB, self).__init__(*args, **kwargs)
+
+    def get_table(self, name):
+        self.execute('SELECT * FROM %s' % name)
+        fields = [d[0] for d in self.cursor.description]
+        rows = self.cursor.fetchall()
+        for row in rows:
+            row = list(row)
+            for i, r in enumerate(row[:]):
+                if isinstance(r, str):
+                    detect = chardet.detect(r)
+                    try:
+                        row[i] = unicode(r, detect['encoding'] or 'utf8', 'replace')
+                    except LookupError:
+                        row[i] = unicode(r, 'utf8', 'replace')
+            yield dict(zip(fields, row))
+
+    def get_radio_chat_table(self):
+        return self.get_table('radio_chat')
+
+    def get_search_log_table(self):
+        return self.get_table('search_log')
+
+    def get_sj_bg_table(self):
+        return self.get_table('sj_bg')
+
+    def get_sj_chat_table(self):
+        return self.get_table('sj_chat')
+
+    def get_sj_likes_table(self):
+        return self.get_table('sj_likes')
+
+    def get_sj_likes_tmp_table(self):
+        return self.get_table('sj_likes_tmp')
+
+    def get_sj_room_table(self):
+        return self.get_table('sj_room')
+
+    def get_sj_search_log_table(self):
+        return self.get_table('sj_search_log')
+
+    def get_sj_session_table(self):
+        return self.get_table('sj_sesson')
+
+    def get_sj_url_table(self):
+        return self.get_table('sj_url')
+
+    def get_sj_url_tmp_table(self):
+        return self.get_table('sj_url_tmp')
+
+    def get_sj_user_table(self):
+        return self.get_table('sj_user')
+
+    def get_sj_video_table(self):
+        return self.get_table('sj_video')
+
+if __name__ == '__main__':
+
+    admin = User.objects.get(username='admin')
+
+    db = MigrateDB()
+    db.connect()
+
+    # Migrate sj_user table
+
+    sj_user = list(db.get_sj_user_table())
+    map_user = {}
+    for row in sj_user:
+        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()
+	
+	settings_text_nice = {}
+	settings_old = row['settings']
+	
+	if settings_old is not None:
+		slinez = settings_old.split("\n")
+		for l in slinez:
+			pair = l.split("\t")
+			if len(pair) >= 2 and pair[0] is not None and pair[1] is not None:
+				settings_text_nice[pair[0]] = pair[1]
+	
+        user_profile.nickname = row['username'] or '' #fixed
+        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.settings_text = json.dumps(settings_text_nice)
+	user_profile.save()
+
+        #for key, value in row.iteritems():
+        #     print key, ', ', value, ', ', type(value)
+        #print
+
+        row['__user_pk'] = user.pk
+        map_user[row['id']] = user.pk
+