Initial commit

env/lib/python3.10/site-packages/wagtail/documents/api/v2/serializers.py

@@ -0,0 +1,23 @@
+from rest_framework.fields import Field
+
+from wagtail.api.v2.serializers import BaseSerializer
+from wagtail.api.v2.utils import get_full_url
+
+
+class DocumentDownloadUrlField(Field):
+    """
+    Serializes the "download_url" field for documents.
+
+    Example:
+    "download_url": "http://api.example.com/documents/1/my_document.pdf"
+    """
+
+    def get_attribute(self, instance):
+        return instance
+
+    def to_representation(self, document):
+        return get_full_url(self.context["request"], document.url)
+
+
+class DocumentSerializer(BaseSerializer):
+    download_url = DocumentDownloadUrlField(read_only=True)

env/lib/python3.10/site-packages/wagtail/documents/api/v2/views.py

@@ -0,0 +1,23 @@
+from wagtail.api.v2.filters import FieldsFilter, OrderingFilter, SearchFilter
+from wagtail.api.v2.views import BaseAPIViewSet
+
+from ... import get_document_model
+from .serializers import DocumentSerializer
+
+
+class DocumentsAPIViewSet(BaseAPIViewSet):
+    base_serializer_class = DocumentSerializer
+    filter_backends = [FieldsFilter, OrderingFilter, SearchFilter]
+    body_fields = BaseAPIViewSet.body_fields + ["title"]
+    meta_fields = BaseAPIViewSet.meta_fields + ["tags", "download_url"]
+    listing_default_fields = BaseAPIViewSet.listing_default_fields + [
+        "title",
+        "tags",
+        "download_url",
+    ]
+    nested_default_fields = BaseAPIViewSet.nested_default_fields + [
+        "title",
+        "download_url",
+    ]
+    name = "documents"
+    model = get_document_model()