from PyPDF2 import PdfFileReader,PdfFileWriter
import os
# 合并PDF
def Merge_pdf(src_dir,pdf_list,out):
if src_dir and os.path.isdir(src_dir):
if pdf_list:
merge = PdfFileWriter()
for pdf in pdf_list:
pdf_path = os.path.join(src_dir,pdf)
print(pdf_path)
if os.path.isfile(pdf_path):
pdf_reader = PdfFileReader(pdf_path)
for i in range(0,pdf_reader.getNumPages()):
merge.addPage(pdf_reader.getPage(i))
else:
merge = PdfFileWriter()
for pdf in os.listdir(src_dir):
if pdf.endswith('.pdf'):
pdf_path = os.path.join(src_dir,pdf)
print(pdf_path)
if os.path.isfile(pdf_path):
pdf_reader = PdfFileReader(pdf_path)
for i in range(0,pdf_reader.getNumPages()):
merge.addPage(pdf_reader.getPage(i))
with open(os.path.join(src_dir,out),'wb') as f:
merge.write(f)
src_dir = r'D:\工作\02.学习\书籍\pandas'
# pdf_list = ['Categorical data — pandas 2.0.1 documentation.pdf','Chart visualization — pandas 2.0.1 documentation.pdf']
out = 'Merge.pdf'
pdf_list = []
Merge_pdf(src_dir, pdf_list, out)