Update record serializing to handle None values

nationalarchives · Nov 14, 2024 · 95998ca · 95998ca
1 parent 059d226
commit 95998ca
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/etna/records/blocks.py b/etna/records/blocks.py
@@ -107,11 +107,13 @@ def extract_references(self, value):
         return []
 
     def get_api_representation(self, value, context=None):
-        return {
-            "title": value.summary_title,
-            "iaid": value.iaid,
-            "reference_number": value.reference_number,
-        }
+        if value:
+            return {
+                "title": value.summary_title,
+                "iaid": value.iaid,
+                "reference_number": value.reference_number,
+            }
+        return None
 
     class Meta:
         icon = "archive"

diff --git a/etna/records/serializers.py b/etna/records/serializers.py
@@ -3,8 +3,10 @@
 
 class RecordSerializer(serializers.Serializer):
     def to_representation(self, instance):
-        return {
-            "title": instance.summary_title,
-            "iaid": instance.iaid,
-            "reference_number": instance.reference_number,
-        }
+        if instance:
+            return {
+                "title": instance.summary_title,
+                "iaid": instance.iaid,
+                "reference_number": instance.reference_number,
+            }
+        return None