migration tools fix

1 files changed, 23 insertions, 16 deletions

diff --git a/tools/migrate_video.py b/tools/migrate_video.py
index f7553b2..dacd33d 100755
--- a/tools/migrate_video.py
+++ b/tools/migrate_video.py
@@ -87,54 +87,61 @@ class MigrateDB(DB):
 
 if __name__ == '__main__':
 
-    admin = User.objects.get(username='admin')
-
     db = MigrateDB()
     db.connect()
 
     # Migrate sj_video table
+    
+    print "loading users..."
     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:
-		print "ERROR in users"
+		print "ERROR in users '%s' does not exist" % row['username']
 
-        map_user[row['id']] = user.pk
+        map_user[row['id']] = user
     
+    print "loading rooms..."
     sj_room = list(db.get_sj_room_table())
     map_room = {}
     for row in sj_room:
         try:
             room = SJRoom.objects.get(name=row['name'])
         except SJRoom.DoesNotExist:
-		print "ERROR"
-        map_room[row['id']] = room.pk
+		print "ERROR in room '%s' does not exist" % row['name']
+        map_room[row['id']] = room
+
 
     sj_video = list(db.get_sj_video_table())
     map_video = {}
     for row in sj_video:
-        user = admin
-        pk = map_user.get(row['userid'], 0)
-        if pk:
-            user = User.objects.get(pk=pk)
-        room = SJRoom.objects.get(name='main')
-        pk = map_room.get(row['roomid'], 0)
-        if pk:
-            room = SJRoom.objects.get(pk=pk)
+        if row['userid'] in map_user:
+		user = map_user[row['userid']]
+	else:
+		print "unknown user in migrate: '%d'" % row['userid']
+        
+        if row['roomid'] in map_room:
+		room = map_room[row['roomid']]
+	else:
+		print "unknown room in migrate: '%d'" % row['roomid']
         try:
             video = SJContent.objects.get(
                 user=user,
                 datetime=datetime.fromtimestamp(row['date']),
                 room=room,
+		old_id=row['id'],
                 content_type='video')
+	    print "video '%d' exist" % row['date']
         except SJContent.DoesNotExist:
             video = SJContent(
                 user=user,
                 datetime=datetime.fromtimestamp(row['date']),
                 room=room,
+		old_id=row['id'],
                 content_type='video')
+	    print "video '%d' created" % row['date']
         video.settings = dict(
             url=row['url'],
             title=row['title'],
@@ -145,6 +152,6 @@ if __name__ == '__main__':
         )
         video.save()
 
-        row['__sjcontent_video_pk'] = video.pk
-        map_video[row['id']] = video.pk
+        #row['__sjcontent_video_pk'] = video.pk
+        #map_video[row['id']] = video.pk