temp save

Author: sebastianMindee

.pre-commit-config.yaml

@@ -41,7 +41,7 @@ repos:
         exclude: "tests/|examples/|docs/"
         additional_dependencies:
           - toml
-          - pikepdf
+          - pypdfium2
           - types-pytz
           - types-requests
           - types-setuptools

mindee/input/sources.py

@@ -7,7 +7,7 @@
 from pathlib import Path
 from typing import BinaryIO, Optional, Sequence, Tuple, Union
 
-import pikepdf
+import pypdfium2 as pdfium
 
 from mindee.error.mimetype_error import MimeTypeError
 from mindee.error.mindee_error import MindeeError, MindeeSourceError
@@ -117,14 +117,14 @@ def count_doc_pages(self) -> int:
         :return: the number of pages.
         """
         self.file_object.seek(0)
-        with pikepdf.open(self.file_object) as pdf:
-            return len(pdf.pages)
+        pdf = pdfium.PdfDocument(self.file_object)
+        return len(pdf)
 
     def process_pdf(
-        self,
-        behavior: str,
-        on_min_pages: int,
-        page_indexes: Sequence,
+            self,
+            behavior: str,
+            on_min_pages: int,
+            page_indexes: Sequence,
     ) -> None:
         """Run any required processing on a PDF file."""
         if self.is_pdf_empty():
@@ -163,14 +163,13 @@ def merge_pdf_pages(self, page_numbers: set) -> None:
         :return: None
         """
         self.file_object.seek(0)
-        new_pdf = pikepdf.Pdf.new()
-        with pikepdf.open(self.file_object) as pdf:
-            for page_id in page_numbers:
-                page = pdf.pages[page_id]
-                new_pdf.pages.append(page)
+        new_pdf = pdfium.PdfDocument.new()
+        pdf = pdfium.PdfDocument(self.file_object)
+        new_pdf.import_pages(pdf, list(page_numbers))
         self.file_object.close()
-        self.file_object = io.BytesIO()
-        new_pdf.save(self.file_object)
+        bytes_io = io.BytesIO()
+        new_pdf.save(bytes_io)
+        self.file_object = bytes_io
 
     def is_pdf_empty(self) -> bool:
         """
@@ -179,24 +178,17 @@ def is_pdf_empty(self) -> bool:
         :return: ``True`` if the PDF is empty
         """
         self.file_object.seek(0)
-        with pikepdf.open(self.file_object) as pdf:
-            for page in pdf.pages:
-                # mypy incorrectly identifies the "/Length" key's value as
-                # an object rather than an int.
-                try:
-                    total_size = page["/Contents"]["/Length"]
-                except ValueError:
-                    total_size = 0  # type: ignore
-                    for content in page["/Contents"]:  # type: ignore
-                        total_size += content["/Length"]
-                has_data = total_size > 1000  # type: ignore
-
-                has_font = "/Font" in page["/Resources"].keys()
-                has_xobj = "/XObject" in page["/Resources"].keys()
-
-                if has_font or has_xobj or has_data:
-                    return False
-            return True
+        pdf = pdfium.PdfDocument(self.file_object)
+        for i in range(len(pdf)):
+            page = pdf.get_page(i)
+
+            has_objects = False
+            for _ in page.get_objects():
+                has_objects = True
+                break
+            if has_objects:
+                return False
+        return True
 
     def read_contents(self, close_file: bool) -> Tuple[str, bytes]:
         """

setup.cfg

@@ -32,8 +32,7 @@ packages = find:
 include_package_data = True
 python_requires = >=3.7
 install_requires =
-    pikepdf~=8.6;python_version>="3.8"
-    pikepdf==6.2.9;python_version<"3.8"
+    pypdfium2~=4.7
     pytz>=2023.3
     requests~=2.31