Restrict merge to PDFs

src/documents/bulk_edit.py

@@ -170,6 +170,11 @@ def rotate(doc_ids: list[int], degrees: int):
 
     rotate_tasks = []
     for doc in qs:
+        if doc.mime_type != "application/pdf":
+            logger.warning(
+                f"Document {doc.id} is not a PDF, skipping rotation.",
+            )
+            continue
         try:
             with pikepdf.open(doc.source_path, allow_overwriting_input=True) as pdf:
                 for page in pdf.pages:

src/documents/tests/test_bulk_edit.py

@@ -328,13 +328,28 @@ class TestPDFActions(DirectoriesMixin, TestCase):
             / "0000003.pdf",
             sample3,
         )
-        self.doc1 = Document.objects.create(checksum="A", title="A", filename=sample1)
+        self.doc1 = Document.objects.create(
+            checksum="A",
+            title="A",
+            filename=sample1,
+            mime_type="application/pdf",
+        )
         self.doc1.archive_filename = sample1_archive
         self.doc1.save()
-        self.doc2 = Document.objects.create(checksum="B", title="B", filename=sample2)
+        self.doc2 = Document.objects.create(
+            checksum="B",
+            title="B",
+            filename=sample2,
+            mime_type="application/pdf",
+        )
         self.doc2.archive_filename = sample2_archive
         self.doc2.save()
-        self.doc3 = Document.objects.create(checksum="C", title="C", filename=sample3)
+        self.doc3 = Document.objects.create(
+            checksum="C",
+            title="C",
+            filename=sample3,
+            mime_type="application/pdf",
+        )
         img_doc = os.path.join(self.dirs.scratch_dir, "sample_image.jpg")
         shutil.copy(
             os.path.join(
@@ -348,6 +363,7 @@ class TestPDFActions(DirectoriesMixin, TestCase):
             checksum="D",
             title="D",
             filename=img_doc,
+            mime_type="image/jpeg",
         )
 
     @mock.patch("documents.tasks.consume_file.delay")
@@ -471,8 +487,8 @@ class TestPDFActions(DirectoriesMixin, TestCase):
         mock_chord.assert_called_once()
         self.assertEqual(result, "OK")
 
-    @mock.patch("documents.tasks.bulk_update_documents.delay")
-    @mock.patch("documents.tasks.update_document_archive_file.delay")
+    @mock.patch("documents.tasks.bulk_update_documents.s")
+    @mock.patch("documents.tasks.update_document_archive_file.s")
     @mock.patch("pikepdf.Pdf.save")
     def test_rotate_with_error(
         self,
@@ -498,3 +514,30 @@ class TestPDFActions(DirectoriesMixin, TestCase):
             expected_str = "Error rotating document"
             self.assertIn(expected_str, error_str)
             mock_update_archive_file.assert_not_called()
+
+    @mock.patch("documents.tasks.bulk_update_documents.s")
+    @mock.patch("documents.tasks.update_document_archive_file.s")
+    @mock.patch("celery.chord.delay")
+    def test_rotate_non_pdf(
+        self,
+        mock_chord,
+        mock_update_document,
+        mock_update_documents,
+    ):
+        """
+        GIVEN:
+            - Existing documents
+        WHEN:
+            - Rotate action is called with 2 documents, one of which is not a PDF
+        THEN:
+            - Rotate action should be performed 1 time, with the non-PDF document skipped
+        """
+        with self.assertLogs("paperless.bulk_edit", level="INFO") as cm:
+            result = bulk_edit.rotate([self.doc2.id, self.img_doc.id], 90)
+            output_str = cm.output[1]
+            expected_str = "Document 4 is not a PDF, skipping rotation"
+            self.assertIn(expected_str, output_str)
+            self.assertEqual(mock_update_document.call_count, 1)
+            mock_update_documents.assert_called_once()
+            mock_chord.assert_called_once()
+            self.assertEqual(result, "OK")