mirror of
https://github.com/ocrmypdf/OCRmyPDF.git
synced 2026-02-07 21:03:59 -05:00
4c7086c609
Migrate watcher.py and pdf_text_diff.py from typer to cyclopts for CLI argument parsing. Update pyproject.toml to reflect the dependency change in the watcher optional feature.
58 lines
1.4 KiB
Python
58 lines
1.4 KiB
Python
# SPDX-FileCopyrightText: 2025 James R. Barlow
|
|
# SPDX-License-Identifier: MPL-2.0
|
|
|
|
"""Compare text in PDFs."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
from subprocess import run
|
|
from tempfile import NamedTemporaryFile
|
|
from typing import Annotated
|
|
|
|
import cyclopts
|
|
|
|
app = cyclopts.App()
|
|
|
|
|
|
@app.default
|
|
def main(
|
|
pdf1: Annotated[Path, cyclopts.Parameter()],
|
|
pdf2: Annotated[Path, cyclopts.Parameter()],
|
|
*,
|
|
engine: Annotated[str, cyclopts.Parameter()] = 'pdftotext',
|
|
):
|
|
"""Compare text in PDFs."""
|
|
with open(pdf1, 'rb') as f1, open(pdf2, 'rb') as f2:
|
|
text1 = run(
|
|
['pdftotext', '-layout', '-', '-'],
|
|
stdin=f1,
|
|
capture_output=True,
|
|
check=True,
|
|
)
|
|
text2 = run(
|
|
['pdftotext', '-layout', '-', '-'],
|
|
stdin=f2,
|
|
capture_output=True,
|
|
check=True,
|
|
)
|
|
|
|
with NamedTemporaryFile() as t1, NamedTemporaryFile() as t2:
|
|
t1.write(text1.stdout)
|
|
t1.flush()
|
|
t2.write(text2.stdout)
|
|
t2.flush()
|
|
diff = run(
|
|
['diff', '--color=always', '--side-by-side', t1.name, t2.name],
|
|
capture_output=True,
|
|
)
|
|
run(['less', '-R'], input=diff.stdout, check=True)
|
|
if text1.stdout.strip() != text2.stdout.strip():
|
|
return 1
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app()
|