完整代码如下
import os
from PyPDF2 import PdfFileReader, PdfFileWriter
def GetFileName(dir_path):
file_list = [os.path.join(dirpath, filesname) \
for dirpath, dirs, files in os.walk(dir_path) \
for filesname in files]
return file_list
def MergePDF(dir_path, file_name):
output = PdfFileWriter()
outputPages = 0
file_list = GetFileName(dir_path)
for pdf_file in file_list:
print("文件:%s" % pdf_file.split('\\')[-1], end=' ')
input = PdfFileReader(open(pdf_file, "rb"))
pageCount = input.getNumPages()
outputPages += pageCount
print("页数:%d" % pageCount)
for iPage in range(pageCount):
output.addPage(input.getPage(iPage))
print("\n合并后的总页数:%d" % outputPages)
print("PDF文件正在合并,请稍等......")
with open(os.path.join(dir_path, file_name), "wb") as outputfile:
output.write(outputfile)
print("PDF文件合并完成")
if __name__ == '__main__':
dir_path = r'C:\Users\David\Desktop\merge'
file_name = "9 临床实验室质量保证体系-选择题.pdf"
MergePDF(dir_path, file_name)