From 2a519916179a550ad1852d476eac6cb6822b45ee Mon Sep 17 00:00:00 2001 From: Lorenz Stechauner Date: Thu, 19 Oct 2023 22:53:55 +0200 Subject: [PATCH] winziprint: update progress more often --- winziprint/winziprint.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/winziprint/winziprint.py b/winziprint/winziprint.py index 7430731..f9986f9 100755 --- a/winziprint/winziprint.py +++ b/winziprint/winziprint.py @@ -19,7 +19,7 @@ def convert(input_files: list[str], output_files: str, encoding: str = None, pro page_nums = [] tmp_file_names = [] - steps = len(input_files) // BATCH_SIZE + 1 + steps = len(input_files) + len(input_files) // BATCH_SIZE + 1 try: for i in range(0, len(input_files), BATCH_SIZE): @@ -30,8 +30,10 @@ def convert(input_files: list[str], output_files: str, encoding: str = None, pro doc = html.render() documents.append(doc) del html + if progress: + print(f'progress: {i + n + i // BATCH_SIZE + 1}/{steps}', flush=True) all_pages = [p for doc in documents for p in doc.pages] - tmp_file_name = f'{output_files}.part.{i:0000}' + tmp_file_name = f'{output_files}.part.{i:04}' documents[0].copy(all_pages).write_pdf(tmp_file_name) tmp_file_names.append(tmp_file_name) page_nums += [len(doc.pages) for doc in documents] @@ -39,7 +41,7 @@ def convert(input_files: list[str], output_files: str, encoding: str = None, pro del all_pages gc.collect() if progress: - print(f'progress: {i // BATCH_SIZE + 1}/{steps}', flush=True) + print(f'progress: {i + BATCH_SIZE + i // BATCH_SIZE + 1}/{steps}', flush=True) merger = pypdf.PdfWriter() for pdf in tmp_file_names: