Python 对pdf文件切分合并

今天有一个问题，就是需要实现提取某个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真的很强大！