Python使用PyPDF2库实现向PDF文件中插入内容

pip install PyPDF2

from PyPDF2 import PdfFileReader, PdfFileWriter  
  
# 创建PDF写入对象  
output = PdfFileWriter()  
  
# 读取第一个PDF文件  
input1 = PdfFileReader(open("file1.pdf", "rb"))  
  
# 读取第二个PDF文件  
input2 = PdfFileReader(open("file2.pdf", "rb"))  
  
# 将第一个PDF文件的所有页面添加到输出文件中  
for i in range(input1.getNumPages()):  
    output.addPage(input1.getPage(i))  
  
# 将第二个PDF文件的所有页面添加到输出文件中  
for i in range(input2.getNumPages()):  
    output.addPage(input2.getPage(i))  
  
# 将合并后的PDF文件写入到新的文件中  
with open("output.pdf", "wb") as outputStream:  
    output.write(outputStream)

from PyPDF2 import PdfFileReader, PdfFileWriter  
  
# 定义要插入的起始页码  
insert_start_page = 3  # 假设要在第一个PDF文件的第3页后插入第二个PDF文件的内容  
  
# 创建PDF写入对象  
output = PdfFileWriter()  
  
# 读取第一个PDF文件  
input1 = PdfFileReader(open("file1.pdf", "rb"))  
  
# 读取第二个PDF文件  
input2 = PdfFileReader(open("file2.pdf", "rb"))  
  
# 将第一个PDF文件的前insert_start_page-1页添加到输出文件中  
for i in range(insert_start_page - 1):  
    output.addPage(input1.getPage(i))  
  
# 将第二个PDF文件的所有页面添加到输出文件中  
for i in range(input2.getNumPages()):  
    output.addPage(input2.getPage(i))  
  
# 将第一个PDF文件的剩余页面添加到输出文件中  
for i in range(insert_start_page - 1, input1.getNumPages()):  
    output.addPage(input1.getPage(i))  
  
# 将合并后的PDF文件写入到新的文件中  
with open("output.pdf", "wb") as outputStream:  
    output.write(outputStream)

# author:mlnt
# createdate:2022/8/16
import PyPDF2    # 导入PyPDF2模块

# 官方文档：https://pypi.org/project/PyPDF2/
# 1.打开PDF文件
pdf = open(file='test.pdf', mode='rb')  # 以二进制方式打开

# 2.获取PDF文件的页数
# 打开PDF文件成功后，可使用PdfFileReader()方法读取PDF内容
pdf_reader = PyPDF2.PdfFileReader(pdf)  # 读取PDF内容
print(f'PDF页数为：{pdf_reader.numPages}')
print(f'PDF页数为：{len(pdf_reader.pages)}')

# 3.读取PDF页面内容
"""
- 使用PdfFileReader()方法读取PDF文件后，可使用getPage(n)（或pages[n]）获取第n页的PDF内容
- PDF页面从第0页开始计算
- 页面内容被读入后，可使用extractText()取得该页的字符串内容
"""
for i in range(pdf_reader.numPages):
    pageObj = pdf_reader.getPage(i)  # 读取第i页内容
    # pageObj = pdf_reader.pages[i]  # 读取第i页内容
    page_content = pageObj.extractText()  # 提取页面内容
    print(page_content)

# author:mlnt
# createdate:2022/8/16

import PyPDF2
from PyPDF2 import PdfReader, PdfWriter


def checkEncrypted(filename):
    """检查文件是否加密"""
    pdfObj = open(file=filename, mode='rb')  # 以二进制方式打开
    pdfRd = PyPDF2.PdfFileReader(pdfObj)  # 读取PDF内容
    # 判断文件是否加密
    if pdfRd.isEncrypted:
        print(f'{filename}属于加密文件')
    else:
        print(f'{filename}未加密')


def pdfEncrypt(filename):
    """pdf加密"""
    reader = PdfReader(filename)  # 创建RdfReader对象
    writer = PdfWriter()  # 创建PdfWriter对象

    # 将所有页面添加到writer
    for page in reader.pages:
        # PDF页面旋转
        page.rotate_clockwise(90)  # 旋转90度
        page.rotate_clockwise(-180)  # 逆时针旋转180度
        writer.add_page(page)

    # 添加密码
    writer.encrypt('123456')

    new_name = filename[:filename.rfind('.pdf')] + '-encrypted.pdf'
    # 保存pdf
    with open(new_name, 'wb') as f:
        writer.write(f)


def pdfDecrypt(filename):
    """pdf解密"""
    reader = PdfReader(filename)
    writer = PdfWriter()

    # 判断是否加密
    if reader.isEncrypted:
        reader.decrypt('123456')

    for page in reader.pages:
        writer.add_page(page)

    new_name = filename[:filename.rfind('.pdf')] + '-decrypted.pdf'
    # 保存pdf
    with open(new_name, 'wb') as f:
        writer.write(f)


checkEncrypted('test.pdf')
pdfEncrypt('test.pdf')
pdfDecrypt('test-encrypted.pdf')

# author:mlnt
# createdate:2022/8/16
from PyPDF2 import PdfMerger

merger = PdfMerger()  # 创建PdfMerger对象

pdf1 = open('test.pdf', 'rb')

pdf2 = open('watermark.pdf', 'rb')

# 将整个文档添加到merger
merger.append(fileobj=pdf1)

# 将pdf2插入到merger，从第0页开始
merger.merge(position=0, fileobj=pdf2)

with open('merger-test.pdf', 'wb') as f:
    merger.write(f)

merger.close()