From a24a9c516adfecf42fdad24be13958cd64a3dadf Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Sun, 8 Jan 2023 18:11:24 -0500
Subject: [PATCH] fix song review migration

---
 legacy/management/commands/migrate_journal.py | 19 ++++++++++++-------
 legacy/models.py                              |  6 +++++-
 2 files changed, 17 insertions(+), 8 deletions(-)

diff --git a/legacy/management/commands/migrate_journal.py b/legacy/management/commands/migrate_journal.py
index a9d43dc3..181dde0c 100644
--- a/legacy/management/commands/migrate_journal.py
+++ b/legacy/management/commands/migrate_journal.py
@@ -117,7 +117,7 @@ class Command(BaseCommand):
         print("Deleting migrated user pieces")
         # Piece.objects.all().delete()
         for cls in classes:  # Collection
-            print(cls)
+            print(f"Deleting {cls}")
             cls.objects.all().delete()
 
     def collection(self, options):
@@ -143,8 +143,12 @@ class Command(BaseCommand):
                 c.catalog_item.save()
                 for citem in entity.collectionitem_list:
                     if citem.song:
-                        LinkModel = AlbumLink
-                        old_id = citem.song.album_id
+                        if citem.song.album_id:
+                            LinkModel = AlbumLink
+                            old_id = citem.song.album_id
+                        else:
+                            LinkModel = SongLink
+                            old_id = citem.song.id
                     else:
                         LinkModel = model_link[citem.item.__class__]
                         old_id = citem.item.id
@@ -153,7 +157,6 @@ class Command(BaseCommand):
                         item = Item.objects.get(uid=item_link.new_uid)
                         c.append_item(item, metadata={"note": citem.comment})
                     else:
-                        # TODO convert song to album
                         print(f"{c.owner} {c.id} {c.title} {citem.item} were skipped")
                 CollectionLink.objects.create(old_id=entity.id, new_uid=c.uid)
             qs = (
@@ -199,7 +202,9 @@ class Command(BaseCommand):
                                 edited_time=entity.edited_time,
                             )
                             ReviewLink.objects.create(
-                                old_id=entity.id, new_uid=review.uid
+                                module=entity.__class__.__module__.split(".")[0],
+                                old_id=entity.id,
+                                new_uid=review.uid,
                             )
                         except Exception as e:
                             print(f"Convert failed for {typ} {entity.id}: {e}")
@@ -306,12 +311,12 @@ class Command(BaseCommand):
             self.initshelf()
         elif options["collection"]:
             if options["clear"]:
-                self.clear([Collection, Like])
+                self.clear([CollectionLink, Like, Collection])
             else:
                 self.collection(options)
         elif options["review"]:
             if options["clear"]:
-                self.clear([Review])
+                self.clear([ReviewLink, Review])
             else:
                 self.review(options)
         elif options["mark"]:
diff --git a/legacy/models.py b/legacy/models.py
index 94354978..ac3175a6 100644
--- a/legacy/models.py
+++ b/legacy/models.py
@@ -33,5 +33,9 @@ class CollectionLink(models.Model):
 
 
 class ReviewLink(models.Model):
-    old_id = models.IntegerField(unique=True)
+    module = models.CharField(max_length=20)
+    old_id = models.IntegerField()
     new_uid = models.UUIDField()
+
+    class Meta:
+        unique_together = [["module", "old_id"]]