added file extension field to the serilaizer

NASA-IMPACT · Feb 22, 2024 · 4e41e9d · 4e41e9d
1 parent cbaebbc
commit 4e41e9d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/sde_collections/serializers.py b/sde_collections/serializers.py
@@ -91,6 +91,7 @@ class Meta:
 class CandidateURLAPISerializer(serializers.ModelSerializer):
     document_type = serializers.SerializerMethodField()
     title = serializers.SerializerMethodField()
+    file_extension = serializers.SerializerMethodField()
 
     class Meta:
         model = CandidateURL
@@ -99,6 +100,7 @@ class Meta:
             "title",
             "document_type",
             "hash",
+            "file_extension",
         )
 
     def get_document_type(self, obj):
@@ -112,6 +114,9 @@ def get_document_type(self, obj):
     def get_title(self, obj):
         return obj.generated_title if obj.generated_title else obj.scraped_title
 
+    def get_file_extension(self, obj):
+        return obj.fileext
+
 
 class BasePatternSerializer(serializers.ModelSerializer):
     match_pattern_type_display = serializers.CharField(source="get_match_pattern_type_display", read_only=True)