前言

本文针对 仅有 PDF 文件，而无相关 PDF 编辑器的情况下，给 PDF 添加水印。

一、前期准备

安装 PyPDF2 ，命令提示框输入：

pip install PyPDF2

新建 watermark.pdf 文件 实际的水印，可以在此文件里修改水印文字的字体和位置。

实现步骤：

新建 watermark.word ，【设计】 → \to→ 【水印】

【自定义水印】 → \to→ 【文字水印】

输入【文字】，修改【字体】、【自号】、【颜色】和【板式】

效果

另存为 watermark.pdf

watermark.pdf 与 .py 文件放在同一目录下

二、代码实现

// =============================================

# @Time : 2021-01-24

# @Author : AXYZdong

# @CSDN : https://axyzdong.blog.csdn.net/

# @FileName: PDFset.py

# @Software: Python IDE

// =============================================

from PyPDF2 import PdfFileReader, PdfFileWriter

def add_watermark(pdf_file_in, pdf_file_mark, pdf_file_out):

"""把水印添加到pdf中"""

pdf_output = PdfFileWriter()

input_stream = open(pdf_file_in, 'rb')

pdf_input = PdfFileReader(input_stream, strict=False)

# 获取PDF文件的页数

pageNum = pdf_input.getNumPages()

# 读入水印pdf文件

pdf_watermark = PdfFileReader(open(pdf_file_mark, 'rb'), strict=False)

# 给每一页打水印

for i in range(pageNum):

page = pdf_input.getPage(i)

page.mergePage(pdf_watermark.getPage(0))

page.compressContentStreams() # 压缩内容

pdf_output.addPage(page)

pdf_output.write(open(pdf_file_out, 'wb'))

if __name__ == '__main__':

pdf_file_in = '基于matlab的控制系统与仿真-1_AXYZdong的博客-CSDN博客.pdf'

pdf_file_out = 'watermarked.pdf'

pdf_file_mark = 'watermark.pdf'

add_watermark(pdf_file_in, pdf_file_mark, pdf_file_out)

效果

三、其他方法

// =============================================

# @Time : 2021-01-24

# @Author : AXYZdong

# @CSDN : https://axyzdong.blog.csdn.net/

# @FileName: PDFSet.py

# @Software: Python IDE

// =============================================

from PyPDF2 import PdfFileReader, PdfFileWriter

from reportlab.lib.units import cm

from reportlab.pdfgen import canvas

def create_watermark(content):

"""水印信息"""

# 默认大小为21cm*29.7cm

file_name = "mark.pdf"

c = canvas.Canvas(file_name, pagesize=(30*cm, 30*cm))

# 移动坐标原点(坐标系左下为(0,0))

c.translate(10*cm, 5*cm)

# 设置字体

c.setFont("Helvetica", 30)

# 指定描边的颜色

c.setStrokeColorRGB(0, 1, 0)

# 指定填充颜色

c.setFillColorRGB(0, 1, 0)

# 旋转45度,坐标系被旋转

c.rotate(30)

# 指定填充颜色

c.setFillColorRGB(0, 0, 0, 0.1)

# 设置透明度,1为不透明

# c.setFillAlpha(0.1)

# 画几个文本,注意坐标系旋转的影响

for i in range(5):

for j in range(10):

a=10*(i-1)

b=5*(j-2)

c.drawString(a*cm, b*cm, content)

c.setFillAlpha(0.1)

# 关闭并保存pdf文件

c.save()

return file_name

def add_watermark(pdf_file_in, pdf_file_mark, pdf_file_out):

"""把水印添加到pdf中"""

pdf_output = PdfFileWriter()

input_stream = open(pdf_file_in, 'rb')

pdf_input = PdfFileReader(input_stream, strict=False)

# 获取PDF文件的页数

pageNum = pdf_input.getNumPages()

# 读入水印pdf文件

pdf_watermark = PdfFileReader(open(pdf_file_mark, 'rb'), strict=False)

# 给每一页打水印

for i in range(pageNum):

page = pdf_input.getPage(i)

page.mergePage(pdf_watermark.getPage(0))

page.compressContentStreams() # 压缩内容

pdf_output.addPage(page)

pdf_output.write(open(pdf_file_out, 'wb'))

if __name__ == '__main__':

pdf_file_in = '基于matlab的控制系统与仿真-1_AXYZdong的博客-CSDN博客.pdf'

pdf_file_out = 'watermarked.pdf'

pdf_file_mark = create_watermark('CSDN@AXYZdong')

add_watermark(pdf_file_in, pdf_file_mark, pdf_file_out)

效果

四、总结 Python通过PyPDF2实现PDF添加水印，实际上就是将两个PDF文件合并实现添加水印；

第一种方法的水印文件是通过 word 另存为 PDF，第二种方法是自动创建一个 PDF 文件；

第二种方法与第一种相比，水印的字体不能随意改变，不支持中文，省去了方法一中手动生成 PDF 水印的步骤；

第一种方法可以任意修改水印的位置和字体。

作者：AXYZdong

[拿走不谢！Python 3.9 官方中文文档，限时领！] (http://dwz.date/dE6v)

[限时！速领！14张高清Python速查表，效率提升必备！] (http://dwz.date/dE6w)

[GitHub标星3W+，80个Python案例，带你轻松玩转Python学习！] (http://dwz.date/dE64)