winzer/elwig-misc
3
1
Fork 0
Code Pull Requests Activity

winziprint: update progress more often

Browse Source
This commit is contained in:
Lorenz Stechauner
2023-10-19 22:53:55 +02:00
parent 499cd2032e
commit 2a51991617
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
winziprint/winziprint.py
View File

@ -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: