3 files changed, 89 insertions, 3 deletions

diff --git a/pyblackbird_cc/resources/forms.py b/pyblackbird_cc/resources/forms.py
index ed2c1a6..5a4a606 100644
--- a/pyblackbird_cc/resources/forms.py
+++ b/pyblackbird_cc/resources/forms.py
@@ -181,6 +181,46 @@ class ResourceUpdateMetadataForm(forms.ModelForm):
         ]
 
 
+class ResourceUpdatePDFsForm(forms.Form):
+    def __init__(self, *args, **kwargs):
+        try:
+            self.resource = kwargs.pop("resource")
+        except KeyError:
+            pass
+        super().__init__(*args, **kwargs)
+        self.helper = FormHelper(self)
+        self.helper.add_input(Submit("submit", "Submit"))
+
+    pdf_files = forms.FileField(
+        widget=forms.TextInput(
+            attrs={
+                "multiple": True,
+                "type": "File",
+                "required": True,
+            },
+        ),
+        required=False,
+        help_text="You can multi-select up to 20 .pdf files here.",
+        label="PDF files",
+    )
+
+    pdf_files.widget.attrs.update({"class": "file_upload", "accept": ".pdf"})
+
+    def clean_pdf_files(self):
+        pdf_files = self.files.getlist("pdf_files")
+        if not pdf_files:
+            raise forms.ValidationError("Please select at least one PDF file.")
+        acceptable = ["application/pdf"]
+        for f in pdf_files:
+            content_type = magic.from_buffer(f.file.read(), mime=True)
+            f.file.seek(0)
+            if content_type not in acceptable:
+                raise forms.ValidationError("Please select only PDF files.")
+        if len(pdf_files) > ALLOWED_PDFS:
+            raise forms.ValidationError("Please select up to 20 PDF files.")
+        return pdf_files
+
+
 class ResourceUpdateThumbnailsForm(forms.Form):
     def __init__(self, *args, **kwargs):
         try:
@@ -218,4 +258,3 @@ class ResourceUpdateThumbnailsForm(forms.Form):
         if len(thumbnail_files) > ALLOWED_THUMBNAILS:
             raise forms.ValidationError("Please select up to 5 files.")
         return thumbnail_files
-
diff --git a/pyblackbird_cc/resources/urls.py b/pyblackbird_cc/resources/urls.py
index 0babe83..45bfd59 100644
--- a/pyblackbird_cc/resources/urls.py
+++ b/pyblackbird_cc/resources/urls.py
@@ -14,6 +14,7 @@ urlpatterns = [
         name="resource_update_metadata",
     ),
     path("resource/update-thumbnails/<int:pk>", views.update_resource_thumbnails, name="resource_update_thumbnails"),
+    path("resource/update-pdfs/<int:pk>", views.update_resource_pdfs, name="resource_update_pdfs"),
 ]
 
 htmx_patterns = [
diff --git a/pyblackbird_cc/resources/views.py b/pyblackbird_cc/resources/views.py
index 4158a16..8958339 100644
--- a/pyblackbird_cc/resources/views.py
+++ b/pyblackbird_cc/resources/views.py
@@ -14,12 +14,12 @@ from django.shortcuts import redirect
 from django.shortcuts import render
 
 from . import services
-from .forms import ResourceCreateForm, ResourceUpdateThumbnailsForm
+from .forms import ResourceCreateForm, ResourceUpdateThumbnailsForm, ResourceUpdatePDFsForm
 from .forms import ResourceUpdateMetadataForm
 from .models import PDFPageSnapshot, ResourceSubcategory
 from .models import PDFResource
 from .models import Resource
-from .s3 import get_presigned_obj_url, upload_files_to_s3, upload_to_s3
+from .s3 import get_presigned_obj_url, upload_files_to_s3, upload_to_s3, upload_snapshotted_pages_to_s3
 
 logger = logging.getLogger(__name__)
 
@@ -284,6 +284,7 @@ def resource_detail(request, resource_id):
     return render(request, "resources/resource_detail.html", {"resource": resource})
 
 
+@login_required()
 def update_resource_thumbnails(request, pk):
     resource = get_object_or_404(Resource, pk=pk)
     if request.method == "POST":
@@ -337,3 +338,48 @@ def update_resource_metadata(request, pk):  # Change resource_id to pk
         "resources/resource_metadata_update.html",
         {"form": form, "resource": resource},
     )
+
+
+@login_required()
+def update_resource_pdfs(request, pk):
+    resource = get_object_or_404(Resource, pk=pk)
+    if request.method == "POST":
+        form = ResourceUpdatePDFsForm(request.POST, request.FILES)
+        if form.is_valid():
+            pdf_files = form.cleaned_data["pdf_files"]
+
+            metadata_generator = create_metadata(pdf_files)
+
+            snapshotted_pages = []
+
+            for metadata, snapshot_images in metadata_generator:
+                # TODO replace or add? This needs to be decided here
+                pdf_resource = PDFResource.objects.create(
+                    resource=resource,
+                    file_name=os.path.basename(metadata.file_name),
+                    file_size=metadata.file_size,
+                )
+
+                for snapshot_image in snapshot_images:
+                    PDFPageSnapshot.objects.create(
+                        name="test",
+                        file_name=os.path.basename(snapshot_image),
+                        pdf_file=pdf_resource,
+                    )
+
+                snapshotted_pages.append(snapshot_images)
+
+                # Reset the file pointers for pdf_files
+                for pdf_file in pdf_files:
+                    pdf_file.seek(0)
+
+                upload_files_to_s3(pdf_files, "pdfuploads")
+                if not upload_snapshotted_pages_to_s3(snapshotted_pages):
+                    raise Exception("Error uploading snapshotted pages to S3")
+
+                return redirect("resources:resource_detail", resource_id=resource.id)
+
+    else:
+        form = ResourceUpdatePDFsForm(resource=pk)
+
+    return render(request, "resources/update_pdfs.html", {"form": form, "resource": resource})