Python分割PDF

忽然发现需要有一个PDF分割器的功能,以前有Adobe Acrobat安装时,自带PDF打印机。现在卸载了Acrobat,没有PDF打印机,不能实现打印多少页的功能。

自然想到Python的PyPDF2模块。

在原先的合并PDF中的代码,修改一下,应该就可以。比如我要拷贝PDF文件中的前面4页。

# -*- coding:utf-8*-
# 利用PyPDF2模块合并同一文件夹下的所有PDF文件
# 只需修改存放PDF文件的文件夹变量:file_dir 和 输出文件名变量: outfile import os from PyPDF2 import PdfFileReader, PdfFileWriter import time # 使用os模块的walk函数,搜索出指定目录下的全部PDF文件 # 获取同一目录下的所有PDF文件的绝对路径 def getFileName(filedir): file_list = [os.path.join(root, filespath) \ for root, dirs, files in os.walk(filedir) \ for filespath in files \ if str(filespath).endswith('pdf') ] return file_list if file_list else [] # 合并同一目录下的所有PDF文件 def MergePDF(filepath, outfile): output = PdfFileWriter() pdf_fileName = getFileName(filepath) if pdf_fileName: for pdf_file in pdf_fileName: print("path:%s"%pdf_file) # 读取源PDF文件 input = PdfFileReader(open(pdf_file, "rb")) # 获得源PDF文件中页面总数 pageCount = input.getNumPages() # 分别将page添加到输出output中 for iPage in range(pageCount):
        # 添加条件判断
        if iPage <=3: #这样就可以保存前面4页了,或者有其他的组合,按实际需要   output.addPage(input.getPage(iPage)) # 写入到目标PDF文件 outputStream = open(os.path.join(filepath, outfile), "wb") output.write(outputStream) outputStream.close() print("PDF split complete!") else: print("nothing to split") # 主函数 def main(): time1 = time.time() file_dir = r'd:\downloads\pdf\' # 存放PDF的原文件夹 outfile = "split.pdf" # 输出的PDF文件的名称 MergePDF(file_dir, outfile) time2 = time.time() print('time consumed %s s.' %(time2 - time1)) main()

需要注意的是,这个文件夹下不能有其他的PDF文件。

猜你喜欢

转载自www.cnblogs.com/darkshitlord/p/12059987.html