diff options
| -rwxr-xr-x | tools/migrate_db.py | 31 |
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 |