Python实现给PDF添加水印的方法
(编辑：jimmy 日期: 2024/11/2 浏览：3 次 )
前言
利用 PyPDF2 处理 PDF 文件，相关文档：https://pythonhosted.org/PyPDF2/
本文针对 仅有 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 水印的步骤；
- 第一种方法可以任意修改水印的位置和字体。
参考文献
[1]：python pdf加水印
[2]：Python中通过PyPDF2实现PDF添加水印
下一篇：Python Process创建进程的2种方法详解