From 67553fc5c6b4e2f05cd7bb171da378c4e54dddbf Mon Sep 17 00:00:00 2001 From: "James R. Barlow" Date: Thu, 17 Sep 2020 01:20:50 -0700 Subject: [PATCH] Display page numbers in log messages when grafting --- src/ocrmypdf/_graft.py | 2 +- src/ocrmypdf/_sync.py | 22 +++++++++++++--------- 2 files changed, 14 insertions(+), 10 deletions(-) diff --git a/src/ocrmypdf/_graft.py b/src/ocrmypdf/_graft.py index 85caeaba..59607348 100644 --- a/src/ocrmypdf/_graft.py +++ b/src/ocrmypdf/_graft.py @@ -130,7 +130,7 @@ class OcrGrafter: text_rotation = autorotate_correction text_misaligned = (text_rotation - content_rotation) % 360 log.debug( - f"Rotations for page {pageno}: [text, auto, misalign, content] = " + f"Rotations for page: [text, auto, misalign, content] = " f"{text_rotation}, {autorotate_correction}, " f"{text_misaligned}, {content_rotation}" ) diff --git a/src/ocrmypdf/_sync.py b/src/ocrmypdf/_sync.py index fc01ba65..cf0bfd61 100644 --- a/src/ocrmypdf/_sync.py +++ b/src/ocrmypdf/_sync.py @@ -252,15 +252,19 @@ def exec_concurrent(context: PdfContext): ocrgraft = OcrGrafter(context) def update_page(result: PageResult, pbar): - sidecars[result.pageno] = result.text - pbar.update() - ocrgraft.graft_page( - pageno=result.pageno, - image=result.pdf_page_from_image, - textpdf=result.ocr, - autorotate_correction=result.orientation_correction, - ) - pbar.update() + try: + tls.pageno = result.pageno + 1 + sidecars[result.pageno] = result.text + pbar.update() + ocrgraft.graft_page( + pageno=result.pageno, + image=result.pdf_page_from_image, + textpdf=result.ocr, + autorotate_correction=result.orientation_correction, + ) + pbar.update() + finally: + tls.pageno = None exec_progress_pool( use_threads=context.options.use_threads,