Adding ownerpw param (#208)

* add owp flag

* add opw

* added ownerpw param

* update

* added test case

Co-authored-by: Neeraj <[email protected]>

Author: Code0987

pdf2image/pdf2image.py

@@ -44,6 +44,7 @@ def convert_from_path(
     jpegopt=None,
     thread_count=1,
     userpw=None,
+    ownerpw=None,
     use_cropbox=False,
     strict=False,
     transparent=False,
@@ -69,6 +70,7 @@ def convert_from_path(
             jpegopt -> jpeg options `quality`, `progressive`, and `optimize` (only for jpeg format)
             thread_count -> How many threads we are allowed to spawn for processing
             userpw -> PDF's password
+            ownerpw -> PDF's owner password
             use_cropbox -> Use cropbox instead of mediabox
             strict -> When a Syntax Error is thrown, it will be raised as an Exception
             transparent -> Output with a transparent background instead of a white one.
@@ -95,7 +97,7 @@ def convert_from_path(
     if isinstance(poppler_path, pathlib.PurePath):
         poppler_path = poppler_path.as_posix()
 
-    page_count = pdfinfo_from_path(pdf_path, userpw, poppler_path=poppler_path)["Pages"]
+    page_count = pdfinfo_from_path(pdf_path, userpw, ownerpw, poppler_path=poppler_path)["Pages"]
 
     # We start by getting the output format, the buffer processing function and if we need pdftocairo
     parsed_fmt, final_extension, parse_buffer_func, use_pdfcairo_format = _parse_format(
@@ -169,6 +171,7 @@ def convert_from_path(
             jpegopt,
             thread_output_file,
             userpw,
+            ownerpw,
             use_cropbox,
             transparent,
             single_file,
@@ -237,6 +240,7 @@ def convert_from_bytes(
     jpegopt=None,
     thread_count=1,
     userpw=None,
+    ownerpw=None,
     use_cropbox=False,
     strict=False,
     transparent=False,
@@ -262,6 +266,7 @@ def convert_from_bytes(
             jpegopt -> jpeg options `quality`, `progressive`, and `optimize` (only for jpeg format)
             thread_count -> How many threads we are allowed to spawn for processing
             userpw -> PDF's password
+            ownerpw -> PDF's owner password
             use_cropbox -> Use cropbox instead of mediabox
             strict -> When a Syntax Error is thrown, it will be raised as an Exception
             transparent -> Output with a transparent background instead of a white one.
@@ -290,6 +295,7 @@ def convert_from_bytes(
                 jpegopt=jpegopt,
                 thread_count=thread_count,
                 userpw=userpw,
+                ownerpw=ownerpw,
                 use_cropbox=use_cropbox,
                 strict=strict,
                 transparent=transparent,
@@ -317,6 +323,7 @@ def _build_command(
     jpegopt,
     output_file,
     userpw,
+    ownerpw,
     use_cropbox,
     transparent,
     single_file,
@@ -354,6 +361,9 @@ def _build_command(
     if userpw is not None:
         args.extend(["-upw", userpw])
 
+    if ownerpw is not None:
+        args.extend(["-opw", ownerpw])
+
     if grayscale:
         args.append("-gray")
 
@@ -440,14 +450,17 @@ def _get_poppler_version(command, poppler_path=None, timeout=None):
 
 
 def pdfinfo_from_path(
-    pdf_path, userpw=None, poppler_path=None, rawdates=False, timeout=None
+    pdf_path, userpw=None, ownerpw=None, poppler_path=None, rawdates=False, timeout=None
 ):
     try:
         command = [_get_command_path("pdfinfo", poppler_path), pdf_path]
 
         if userpw is not None:
             command.extend(["-upw", userpw])
 
+        if ownerpw is not None:
+            command.extend(["-opw", ownerpw])
+
         if rawdates:
             command.extend(["-rawdates"])
 
@@ -491,15 +504,15 @@ def pdfinfo_from_path(
 
 
 def pdfinfo_from_bytes(
-    pdf_file, userpw=None, poppler_path=None, rawdates=False, timeout=None
+    pdf_file, userpw=None, ownerpw=None, poppler_path=None, rawdates=False, timeout=None
 ):
     fh, temp_filename = tempfile.mkstemp()
     try:
         with open(temp_filename, "wb") as f:
             f.write(pdf_file)
             f.flush()
-        return pdfinfo_from_path(temp_filename, userpw=userpw, rawdates=rawdates,
-                                 poppler_path=poppler_path)
+        return pdfinfo_from_path(temp_filename, userpw=userpw, ownerpw=ownerpw, 
+                                 rawdates=rawdates, poppler_path=poppler_path)
     finally:
         os.close(fh)
         os.remove(temp_filename)

tests.py

@@ -621,6 +621,23 @@ def test_locked_pdf_with_ownerpw_and_userpw(self):
             )
         )
 
+    @profile
+    @unittest.skipIf(not POPPLER_INSTALLED, "Poppler is not installed!")
+    def test_locked_pdf_with_ownerpw_and_userpw_forgotten(self):
+        start_time = time.time()
+        with TemporaryDirectory() as path:
+            with open("./tests/test_locked_both_user_forgotten.pdf", "rb") as pdf_file:
+                images_from_bytes = convert_from_bytes(
+                    pdf_file.read(), output_folder=path, fmt=".jpg", ownerpw="pdf2image"
+                )
+                self.assertTrue(len(images_from_bytes) == 1)
+                [im.close() for im in images_from_bytes]
+        print(
+            "test_locked_pdf_with_ownerpw_and_userpw_forgotten: {} sec".format(
+                time.time() - start_time
+            )
+        )
+
     ## Tests cropbox
 
     @profile