Update serialization: separate seeAlso and related

(#958)
ecds · Dec 5, 2023 · 374f419 · 374f419
1 parent 87f80b4
commit 374f419
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 6 deletions.
diff --git a/apps/iiif/manifests/models.py b/apps/iiif/manifests/models.py
@@ -259,8 +259,37 @@ def related_links(self):
         :return: List of links related to Manifest
         :rtype: list
         """
-        links = [link.link for link in self.relatedlink_set.all()]
-        links.append(self.get_volume_url())
+        links = [
+            {
+                "@id": link.link,
+                "format": link.format,
+            } if link.format else link.link
+            for link in self.relatedlink_set.all()
+        ]
+        links.append({
+            "@id": self.get_volume_url(),
+            "format": "text/html"
+        })
+        return links
+
+    @property
+    def see_also_links(self):
+        """List of links for IIIF v2 'seeAlso' field (structured data).
+
+        :return: List of links to structured data describing Manifest
+        :rtype: list
+        """
+        links = []
+        for link in self.relatedlink_set.all():
+            if link.data_type.lower() == "dataset":
+                links.append(
+                    {
+                        "@id": link.link,
+                        "format": link.format,
+                    }
+                    if link.format
+                    else link.link
+                )
         return links
 
     # TODO: Is this needed? It doesn't seem to be called anywhere.

diff --git a/apps/iiif/manifests/tests/tests.py b/apps/iiif/manifests/tests/tests.py
@@ -175,7 +175,7 @@ def test_serialized_related_links(self):
                 [manifest]
             )
         )
-        assert 'seeAlso' not in no_links.keys()
+        assert not no_links['seeAlso']
 
         link = RelatedLink(link='images.org', manifest=manifest)
         link.save()

diff --git a/apps/iiif/serializers/manifest.py b/apps/iiif/serializers/manifest.py
@@ -125,10 +125,9 @@ def get_dump_object(self, obj):
                             )
                         )
                     }
-                ]
+                ],
+                "seeAlso": obj.see_also_links,
             }
-            if obj.relatedlink_set.exists():
-                data["seeAlso"] = [related.link for related in obj.relatedlink_set.all()]
             return data
         return None