Fix the citation relationship data (#10431)

CenterForOpenScience · Aug 16, 2023 · 59f4037 · 59f4037
1 parent d06b201
commit 59f4037
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/api/base/serializers.py b/api/base/serializers.py
@@ -911,6 +911,10 @@ def to_representation(self, value):
                     elif related_class.view_name == 'node-storage':
                         related_id = resolved_url.kwargs['node_id']
                         related_type = 'node-storage'
+                    elif related_class.view_name == 'node-citation' \
+                            or related_class.view_name == 'registration-citation':
+                        related_id = resolved_url.kwargs['node_id']
+                        related_type = 'citation'
                     elif related_type in ('preprint_providers', 'preprint-providers', 'registration-providers'):
                         related_id = resolved_url.kwargs['provider_id']
                     elif related_type in ('registrations', 'draft_nodes'):