tools: Fix bug

1 files changed, 16 insertions, 13 deletions

diff --git a/tools/migrate_db.py b/tools/migrate_db.py
index b5c12f8..70918c2 100755
--- a/tools/migrate_db.py
+++ b/tools/migrate_db.py
@@ -39,7 +39,10 @@ class MigrateDB(DB):
             for i, r in enumerate(row[:]):
                 if isinstance(r, str):
                     detect = chardet.detect(r)
-                    row[i] = unicode(r, detect['encoding'] or 'utf8', 'replace')
+                    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):
@@ -254,19 +257,19 @@ if __name__ == '__main__':
         pk = map_video.get(row['videoid'], 0)
         if pk:
             video = SJContent.objects.get(pk=pk)
-        try:
-            like = SJLike.objects.get(
-                user=user,
-                datetime=datetime.fromtimestamp(row['date']),
-                content=video)
-        except SJContent.DoesNotExist:
-            like = SJLike(
-                user=user,
-                datetime=datetime.fromtimestamp(row['date']),
-                content=video)
-        like.save()
+            try:
+                like = SJLike.objects.get(
+                    user=user,
+                    datetime=datetime.fromtimestamp(row['date']),
+                    content=video)
+            except SJLike.DoesNotExist:
+                like = SJLike(
+                    user=user,
+                    datetime=datetime.fromtimestamp(row['date']),
+                    content=video)
+            like.save()
 
-        row['__sjlike_pk'] = like.pk
+            row['__sjlike_pk'] = like.pk
 
 
     # Migrate sj_chat table