winzer/winziprint
2
0
Fork 0
Code Issues Pull Requests Activity

winziprint.py: Fix 'list assignment index out of range' error 2

Browse Source
This commit is contained in:
Lorenz Stechauner
2024-03-09 22:06:08 +01:00
parent 1a425915d7
commit 3e0d004e82
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/winziprint.py
View File

@ -16,7 +16,7 @@ import weasyprint
import pypdf import pypdf
VERSION = __version__ = '0.2.7' VERSION = __version__ = '0.2.8'
SOCKET_ADDRESS = ('127.0.0.1', 30983) SOCKET_ADDRESS = ('127.0.0.1', 30983)
BATCH_SIZE = 10 BATCH_SIZE = 10
@ -75,12 +75,12 @@ def convert(input_files: list[str],
if letterhead and file_name.startswith('#'): if letterhead and file_name.startswith('#'):
merger.insert_page(letterhead[0], index=letterhead[1]) merger.insert_page(letterhead[0], index=letterhead[1])
merger.insert_blank_page(index=letterhead[1] + 1) merger.insert_blank_page(index=letterhead[1] + 1)
page_nums[letterhead[1]] = 1 page_nums[letterhead[2]] = 1
letterhead = None letterhead = None
if file_name.endswith('.pdf'): if file_name.endswith('.pdf'):
if padding and file_name.startswith('#'): if padding and file_name.startswith('#'):
r = pypdf.PdfReader(file_name.lstrip('!#')) r = pypdf.PdfReader(file_name.lstrip('!#'))
letterhead = (r.pages[0], p0) letterhead = (r.pages[0], p0, n)
del r del r
else: else:
merger.append(file_name.lstrip('!#')) merger.append(file_name.lstrip('!#'))
@ -90,7 +90,7 @@ def convert(input_files: list[str],
tmp_file_name = tmp_file_names[i // BATCH_SIZE] tmp_file_name = tmp_file_names[i // BATCH_SIZE]
if padding and file_name.startswith('#'): if padding and file_name.startswith('#'):
r = pypdf.PdfReader(tmp_file_name) r = pypdf.PdfReader(tmp_file_name)
letterhead = (r.pages[page_start], p0) letterhead = (r.pages[page_start], p0, n)
del r del r
else: else:
merger.append(tmp_file_name, pages=(page_start, page_start + tmp_page_nums[i])) merger.append(tmp_file_name, pages=(page_start, page_start + tmp_page_nums[i]))
@ -107,11 +107,10 @@ def convert(input_files: list[str],
if len(merger.pages) <= letterhead[1]: if len(merger.pages) <= letterhead[1]:
merger.add_page(letterhead[0]) merger.add_page(letterhead[0])
merger.add_blank_page() merger.add_blank_page()
page_nums.append(1)
else: else:
merger.insert_page(letterhead[0], index=letterhead[1]) merger.insert_page(letterhead[0], index=letterhead[1])
merger.insert_blank_page(index=letterhead[1] + 1) merger.insert_blank_page(index=letterhead[1] + 1)
page_nums[letterhead[1]] = 1 page_nums[letterhead[2]] = 1
merger.write(output_file) merger.write(output_file)
merger.close() merger.close()
del merger del merger