mirror of
https://github.com/ocrmypdf/OCRmyPDF.git
synced 2026-02-06 07:07:59 +00:00
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
# SPDX-FileCopyrightText: 2022 James R. Barlow
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import patch
|
|
|
|
from ocrmypdf import hookimpl
|
|
from ocrmypdf.builtin_plugins import ghostscript
|
|
from ocrmypdf.subprocess import run_polling_stderr
|
|
|
|
|
|
def run_rig_args(args, **kwargs):
|
|
# Remove the two arguments that tell ghostscript to create a PDF/A
|
|
# Does not remove the Postscript definition file - not necessary
|
|
# to cause PDF/A creation failure
|
|
new_args = [
|
|
arg for arg in args if not arg.startswith('-dPDFA') and not arg.endswith('.ps')
|
|
]
|
|
proc = run_polling_stderr(new_args, **kwargs)
|
|
return proc
|
|
|
|
|
|
@hookimpl
|
|
def generate_pdfa(pdf_pages, pdfmark, output_file, compression, pdf_version, pdfa_part):
|
|
with patch('ocrmypdf._exec.ghostscript.run_polling_stderr') as mock:
|
|
mock.side_effect = run_rig_args
|
|
ghostscript.generate_pdfa(
|
|
pdf_pages=pdf_pages,
|
|
pdfmark=pdfmark,
|
|
output_file=output_file,
|
|
compression=compression,
|
|
pdf_version=pdf_version,
|
|
pdfa_part=pdfa_part,
|
|
progressbar_class=None,
|
|
)
|
|
mock.assert_called()
|
|
return output_file
|