added excluded field to the serializer

NASA-IMPACT · Feb 14, 2024 · cbfbb0d · cbfbb0d
1 parent aca9a0a
commit cbfbb0d
Showing 1 changed file with 11 additions and 18 deletions.
diff --git a/sde_collections/serializers.py b/sde_collections/serializers.py
@@ -12,12 +12,8 @@
 
 
 class CollectionSerializer(serializers.ModelSerializer):
-    curation_status_display = serializers.CharField(
-        source="get_curation_status_display", read_only=True
-    )
-    workflow_status_display = serializers.CharField(
-        source="get_workflow_status_display", read_only=True
-    )
+    curation_status_display = serializers.CharField(source="get_curation_status_display", read_only=True)
+    workflow_status_display = serializers.CharField(source="get_workflow_status_display", read_only=True)
 
     class Meta:
         model = Collection
@@ -44,9 +40,7 @@ class Meta:
 
 class CandidateURLSerializer(serializers.ModelSerializer):
     excluded = serializers.BooleanField(required=False)
-    document_type_display = serializers.CharField(
-        source="get_document_type_display", read_only=True
-    )
+    document_type_display = serializers.CharField(source="get_document_type_display", read_only=True)
     url = serializers.CharField(required=False)
     generated_title_id = serializers.SerializerMethodField(read_only=True)
     match_pattern_type = serializers.SerializerMethodField(read_only=True)
@@ -95,10 +89,9 @@ class Meta:
 
 
 class CandidateURLAPISerializer(serializers.ModelSerializer):
-    document_type = serializers.CharField(
-        source="get_document_type_display", read_only=True
-    )
+    document_type = serializers.CharField(source="get_document_type_display", read_only=True)
     title = serializers.CharField(source="scraped_title")
+    excluded = serializers.SerializerMethodField()
 
     class Meta:
         model = CandidateURL
@@ -107,13 +100,15 @@ class Meta:
             "title",
             "document_type",
             "hash",
+            "excluded",
         )
 
+    def get_excluded(self, obj):
+        return getattr(obj, "excluded", False)
+
 
 class BasePatternSerializer(serializers.ModelSerializer):
-    match_pattern_type_display = serializers.CharField(
-        source="get_match_pattern_type_display", read_only=True
-    )
+    match_pattern_type_display = serializers.CharField(source="get_match_pattern_type_display", read_only=True)
     candidate_urls_count = serializers.SerializerMethodField(read_only=True)
 
     def get_candidate_urls_count(self, instance):
@@ -161,9 +156,7 @@ def validate_match_pattern(self, value):
 
 
 class DocumentTypePatternSerializer(BasePatternSerializer, serializers.ModelSerializer):
-    document_type_display = serializers.CharField(
-        source="get_document_type_display", read_only=True
-    )
+    document_type_display = serializers.CharField(source="get_document_type_display", read_only=True)
     document_type = serializers.ChoiceField(
         choices=DocumentTypes.choices
         + [