tools: Add update SJSearch

1 files changed, 30 insertions, 1 deletions

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