add initial curatedurl apis

Author: CarsonDavis

sde_collections/serializers.py

@@ -155,7 +155,46 @@ class Meta:
             "url",
             "title",
             "document_type",
-            "hash",
+            "file_extension",
+            "tree_root",
+        )
+
+    def get_document_type(self, obj):
+        if obj.document_type is not None:
+            return obj.get_document_type_display()
+        elif obj.collection.document_type is not None:
+            return obj.collection.get_document_type_display()
+        else:
+            return "Unknown"
+
+    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
+
+    def get_tree_root(self, obj):
+        if obj.collection.is_multi_division:
+            if obj.division:
+                return f"/{obj.get_division_display()}/{obj.collection.name}/"
+            else:
+                return f"/{obj.collection.get_division_display()}/{obj.collection.name}/"
+        else:
+            return obj.collection.tree_root
+
+
+class CuratedURLAPISerializer(serializers.ModelSerializer):
+    document_type = serializers.SerializerMethodField()
+    title = serializers.SerializerMethodField()
+    file_extension = serializers.SerializerMethodField()
+    tree_root = serializers.SerializerMethodField()
+
+    class Meta:
+        model = CuratedUrl
+        fields = (
+            "url",
+            "title",
+            "document_type",
             "file_extension",
             "tree_root",
         )

sde_collections/urls.py

@@ -60,5 +60,11 @@
         view=views.DeltaURLAPIView.as_view(),
         name="delta-url-api",
     ),
+    path("curated-urls-api/<str:config_folder>/", view=views.CuratedURLAPIView.as_view(), name="curated-url-api"),
+    path(
+        "candidate-url-api/<str:config_folder>/",
+        view=views.CuratedURLAPIView.as_view(),
+        name="candidate-url-api",
+    ),
     path("titles-and-errors/", views.TitlesAndErrorsView.as_view(), name="titles-and-errors-list"),
 ]

sde_collections/views.py

@@ -42,6 +42,7 @@
 from .serializers import (
     CollectionReadSerializer,
     CollectionSerializer,
+    CuratedURLAPISerializer,
     CuratedURLSerializer,
     DeltaURLAPISerializer,
     DeltaURLBulkCreateSerializer,
@@ -360,6 +361,23 @@ def get_queryset(self):
         return queryset
 
 
+class CuratedURLAPIView(ListAPIView):
+    serializer_class = CuratedURLAPISerializer
+
+    def get(self, request, *args, **kwargs):
+        config_folder = kwargs.get("config_folder")
+        self.config_folder = config_folder
+        return super().get(request, *args, **kwargs)
+
+    def get_queryset(self):
+        queryset = (
+            CuratedUrl.objects.filter(collection__config_folder=self.config_folder)
+            .with_exclusion_status()
+            .filter(excluded=False)
+        )
+        return queryset
+
+
 class ExcludePatternViewSet(CollectionFilterMixin, viewsets.ModelViewSet):
     queryset = DeltaExcludePattern.objects.all()
     serializer_class = ExcludePatternSerializer