今天有一个问题,就是需要实现提取某个pdf文档的某几页,合并到另外一个pdf文档中,想用我的pdf编辑工具实现此功能,发现需要收费,那怎么能行。想到之前用Python处理过word和excel,此时萌生了另外一个想法,那就是万能的Python有没有对应的库呢?网上一查还真的有,那就是PyPDF2 直接pip install pypdf2。很快便实现了我想要的结果,特此记录!
#coding=utf-8
"""
利用python对pdf进行切分和合并
2019/4/18
@Zjh
"""
from PyPDF2 import PdfFileReader, PdfFileWriter
#拆分一个pdf文档,提取其中特定的页数,input为输入文件的路径,out为输出
def split_pdf(input,out):
pdf_open = open(input, 'rb')
read_pdf = PdfFileReader(pdf_open)
n1 = read_pdf.getNumPages() # 计算此PDF文件中的页数
print("文档共有%s页"%n1)
writer = PdfFileWriter()
start=1 #起始页数
end=10 #终止页数 可修改为自己想要的
#page1 = read_pdf.getPage(0) ##首页下标为0
#page1.extractText()
#print(page1)
for i in range(start,end):
writer.addPage(read_pdf.getPage(i))
pdf_write = open(out, "wb")
writer.write(pdf_write)
pdf_open.close()
pdf_write.close()
print("切分完毕!提取源文档第{}-{}页".format(start,end))
#合并两个pdf文档,input2加在input1后面
def merge(input1,input2,out):
pdf_open1 = open(input1, 'rb')
pdf_open2 = open(input2, 'rb')
read_pdf1 = PdfFileReader(pdf_open1)
read_pdf2 = PdfFileReader(pdf_open2)
n1 = read_pdf1.getNumPages() # 计算此PDF文件中的页数
n2 = read_pdf2.getNumPages() # 计算此PDF文件中的页数
writer = PdfFileWriter()
for i in range(n1):
writer.addPage(read_pdf1.getPage(i))
for j in range(n2):
writer.addPage(read_pdf2.getPage(j))
writer.encrypt("666666") #为pdf设置密码
pdf_write = open(out, "wb")
writer.write(pdf_write)
pdf_open1.close()
pdf_open2.close()
pdf_write.close()
if __name__ == '__main__':
split_pdf('test.pdf','out.pdf')
#merge('1.pdf','2.pdf','result.pdf')
本人在调试程序中还是遇到了一些小问题,如下图所示:
网上查了下,是编码问题,定位到对应的utils.py文件
源文件如下:
修改为:
重新运行,问题解决!不得不说,Python真的很强大!