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

winziprint: delete temp files in finally block

Browse Source
This commit is contained in:
Lorenz Stechauner
2023-10-19 21:21:21 +02:00
parent 43c1c27382
commit 5c924c944c
1 changed files with 29 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
winziprint/winziprint.py
View File

@ -19,6 +19,7 @@ def convert(input_file_names: list[str], output_file_name: str, encoding: str =
page_nums = [] page_nums = []
tmp_file_names = [] tmp_file_names = []
try:
for i in range(0, len(input_file_names), BATCH_SIZE): for i in range(0, len(input_file_names), BATCH_SIZE):
batch = input_file_names[i:i + BATCH_SIZE] batch = input_file_names[i:i + BATCH_SIZE]
documents = [] documents = []
@ -39,9 +40,13 @@ def convert(input_file_names: list[str], output_file_name: str, encoding: str =
merger = pypdf.PdfWriter() merger = pypdf.PdfWriter()
for pdf in tmp_file_names: for pdf in tmp_file_names:
merger.append(pdf) merger.append(pdf)
os.remove(pdf)
merger.write(output_file_name) merger.write(output_file_name)
merger.close() merger.close()
del merger
finally:
for pdf in tmp_file_names:
if os.path.isfile(pdf):
os.remove(pdf)
return page_nums return page_nums
@ -59,6 +64,8 @@ def _wrapper_convert(args: list[str], encoding: str = None) -> None:
msg = str(e).replace('\n', ' ') msg = str(e).replace('\n', ' ')
print(f'error: {msg}', flush=True) print(f'error: {msg}', flush=True)
traceback.print_exception(e) traceback.print_exception(e)
finally:
gc.collect()
def usage() -> None: def usage() -> None: