Added views and serializers for DeltaUrl

Author: dhanur-sharma

sde_collections/serializers.py

@@ -9,7 +9,7 @@
     DeltaIncludePattern,
     DeltaTitlePattern,
 )
-from .models.delta_url import DeltaUrl
+from .models.delta_url import CuratedUrl, DeltaUrl
 
 
 class CollectionSerializer(serializers.ModelSerializer):
@@ -94,6 +94,46 @@ class Meta:
         )
 
 
+class CuratedURLSerializer(serializers.ModelSerializer):
+    excluded = serializers.BooleanField(required=False)
+    document_type_display = serializers.CharField(source="get_document_type_display", read_only=True)
+    division_display = serializers.CharField(source="get_division_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)
+    curated_urls_count = serializers.SerializerMethodField(read_only=True)
+
+    def get_curated_urls_count(self, obj):
+        titlepattern = obj.deltatitlepattern_curated_urls.last()
+        return titlepattern.curated_urls.count() if titlepattern else 0
+
+    def get_generated_title_id(self, obj):
+        titlepattern = obj.deltatitlepattern_curated_urls.last()
+        return titlepattern.id if titlepattern else None
+
+    def get_match_pattern_type(self, obj):
+        titlepattern = obj.deltatitlepattern_curated_urls.last()
+        return titlepattern.match_pattern_type if titlepattern else None
+
+    class Meta:
+        model = CuratedUrl
+        fields = (
+            "id",
+            "excluded",
+            "url",
+            "scraped_title",
+            "generated_title",
+            "generated_title_id",
+            "match_pattern_type",
+            "curated_urls_count",
+            "document_type",
+            "document_type_display",
+            "division",
+            "division_display",
+            "visited",
+        )
+
+
 class DeltaURLBulkCreateSerializer(serializers.ModelSerializer):
     class Meta:
         model = DeltaUrl

sde_collections/urls.py

@@ -9,6 +9,7 @@
 router.register(r"collections", views.CollectionViewSet, basename="collection")
 router.register(r"collections-read", views.CollectionReadViewSet, basename="collection-read")
 router.register(r"delta-urls", views.DeltaURLViewSet)
+router.register(r"curated-urls", views.CuratedURLViewSet)
 router.register(r"exclude-patterns", views.ExcludePatternViewSet)
 router.register(r"include-patterns", views.IncludePatternViewSet)
 router.register(r"title-patterns", views.TitlePatternViewSet)

sde_collections/views.py

@@ -33,10 +33,16 @@
     DeltaIncludePattern,
     DeltaTitlePattern,
 )
-from .models.delta_url import DeltaResolvedTitle, DeltaResolvedTitleError, DeltaUrl
+from .models.delta_url import (
+    CuratedUrl,
+    DeltaResolvedTitle,
+    DeltaResolvedTitleError,
+    DeltaUrl,
+)
 from .serializers import (
     CollectionReadSerializer,
     CollectionSerializer,
+    CuratedURLSerializer,
     DeltaURLAPISerializer,
     DeltaURLBulkCreateSerializer,
     DeltaURLSerializer,
@@ -284,6 +290,36 @@ def update_division(self, request, pk=None):
         return Response(status=status.HTTP_400_BAD_REQUEST, data={"error": "Division is required."})
 
 
+class CuratedURLViewSet(CollectionFilterMixin, viewsets.ModelViewSet):
+    queryset = CuratedUrl.objects.all()
+    serializer_class = CuratedURLSerializer
+
+    def _filter_by_is_excluded(self, queryset, is_excluded):
+        if is_excluded == "false":
+            queryset = queryset.filter(excluded=False)
+        elif is_excluded == "true":
+            queryset = queryset.exclude(excluded=False)
+        return queryset
+
+    def get_queryset(self):
+        queryset = super().get_queryset()
+        if self.request.method == "GET":
+            # Filter based on exclusion status
+            is_excluded = self.request.GET.get("is_excluded")
+            if is_excluded:
+                queryset = self._filter_by_is_excluded(queryset, is_excluded)
+        return queryset.order_by("url")
+
+    def update_division(self, request, pk=None):
+        delta_url = get_object_or_404(CuratedUrl, pk=pk)
+        division = request.data.get("division")
+        if division:
+            delta_url.division = division
+            delta_url.save()
+            return Response(status=status.HTTP_200_OK)
+        return Response(status=status.HTTP_400_BAD_REQUEST, data={"error": "Division is required."})
+
+
 class DeltaURLBulkCreateView(generics.ListCreateAPIView):
     queryset = DeltaUrl.objects.all()
     serializer_class = DeltaURLBulkCreateSerializer