from PyPDF2 import PdfFileWriter, PdfFileReader,PdfFileMerger
def pdf_split(pdf_in):
"""拆分PDF成单页"""
with open(pdf_in,'rb') as in_pdf:
pdf_file = PdfFileReader(in_pdf)
if pdf_file.isEncrypted:
pdf_file.decrypt('')
page_num = pdf_file.getNumPages()
for i in range(page_num):
pdf_out = 'after_split_b_' + str(i) +'.pdf'
output = PdfFileWriter()
output.addPage(pdf_file.getPage(i))
with open(pdf_out,'ab') as out_pdf:
output.write(out_pdf)
if __name__ == '__main__':
"""拆分PDF"""
pdf_split("pdf_in")
"""合并PDF"""
merger = PdfFileMerger()
for i in range(72):
with open('after_split_b_' + str(i) + '.pdf' , 'rb') as pdf:
merger.append(PdfFileReader(pdf))
with open('after_split_a_' + str(i+1) + '.pdf' , 'rb') as pdf:
merger.append(PdfFileReader(pdf))
merger.append(PdfFileReader('after_split_a_73.pdf'))
merger.append(PdfFileReader('after_split_a_74.pdf'))
merger.write('merged.pdf')