From e07cc3385fa276481a532c7008e8e1a23264df04 Mon Sep 17 00:00:00 2001
From: "Maksim A. Boyko" <maksim.a.boyko@gmail.com>
Date: Sun, 28 Oct 2012 11:55:16 -0400
Subject: tools: Add update SJSearch

---
 tools/migrate_db.py | 31 ++++++++++++++++++++++++++++++-
 1 file changed, 30 insertions(+), 1 deletion(-)

(limited to 'tools/migrate_db.py')

diff --git a/tools/migrate_db.py b/tools/migrate_db.py
index 652b638..18a70f8 100755
--- a/tools/migrate_db.py
+++ b/tools/migrate_db.py
@@ -126,6 +126,34 @@ if __name__ == '__main__':
         map_user[row['id']] = user.pk
 
 
+    # Migrate sj_search_log table
+
+    sj_search_log = list(db.get_sj_search_log_table())
+    for row in sj_search_log:
+        user = admin
+        pk = map_user.get(row['userid'], 0)
+        if pk:
+            user = User.objects.get(pk=pk)
+        try:
+            sjsearch = SJSearch.objects.get(
+                user=user,
+                datetime=datetime.fromtimestamp(row['date'])
+            )
+        except SJSearch.DoesNotExist:
+            sjsearch = SJSearch(
+                user=user,
+                datetime=datetime.fromtimestamp(row['date'])
+            )
+        sjsearch.settings = dict(
+            url=row['url'],
+            count=row['count'],
+            terms=row['terms'],
+            title=row['title']
+        )
+        sjsearch.save()
+
+    sys.exit(0)
+
     # Migrate sj_room table
 
     sj_room = list(db.get_sj_room_table())
@@ -205,7 +233,8 @@ if __name__ == '__main__':
             likes=row['likes'],
             removed=row['removed'],
             thumbnail=row['thumbnail'],
-            username=row['username'])
+            username=row['username']
+        )
         video.save()
 
         row['__sjcontent_video_pk'] = video.pk
-- 
cgit v1.2.3-70-g09d2