Python给PDF添加水印的代码步骤

pip install PyPDF2 reportlab

import io
from PyPDF2 import PdfWriter, PdfReader
from reportlab.lib import pagesizes
from reportlab.lib.units import cm
from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont
from reportlab.pdfgen import canvas

# 注册字体，以便在 PDF 中使用中文
pdfmetrics.registerFont(TTFont('SimHei', 'SimHei.ttf'))

# 生成水印文件的函数
def create_water_mark(text):
    packet = io.BytesIO()
    # 创建一个新的 PDF 页面作为水印
    my_canvas = canvas.Canvas(packet, pagesizes.A4)
    # 设置水印字体和大小
    my_canvas.setFont("SimHei", 20)
    # 设置填充色为黑色
    my_canvas.setFillColorRGB(0, 0, 0)
    # 设置透明度（0.0 完全透明，1.0 完全不透明）
    my_canvas.setFillAlpha(0.1)
    # 旋转字体以增加水印效果
    my_canvas.rotate(30)
    # 在页面上多次绘制水印文本，以形成水印图案
    for i in range(3, 24, 10):
        for j in range(-5, 30, 5):
            my_canvas.drawString(i * cm, j * cm, text)
    my_canvas.save()
    packet.seek(0)
    # 返回水印页面的 PdfReader 对象
    return PdfReader(packet)

# 将水印添加到 PDF 文件的函数
def add_watermark(input_pdf_path, output_pdf_path, watermark_text):
    # 创建水印
    watermark = create_water_mark(watermark_text)
    # 读取输入 PDF 文件
    pdf_reader = PdfReader(input_pdf_path)
    pdf_writer = PdfWriter()
    # 遍历输入 PDF 的每一页，并添加水印
    for page in pdf_reader.pages:
        page.merge_page(watermark.pages[0])
        pdf_writer.add_page(page)
    # 将带有水印的 PDF 写入输出文件
    with open(output_pdf_path, "wb") as output_pdf:
        pdf_writer.write(output_pdf)

# 主程序
if __name__ == '__main__':
    # 输入和输出 PDF 文件的路径
    input_pdf = r"input.pdf"  # 替换为你的输入 PDF 路径
    output_pdf = r"output_with_watermark.pdf"  # 期望的输出 PDF 路径
    watermark_text = "添加水印测试文本"  # 水印文本内容
    # 调用函数添加水印
    add_watermark(input_pdf, output_pdf, watermark_text)

python your_script_name.py