Python 在PDF中生成水印

前言

在PDF中插入水印是比较常用的一种功能。一般在生成比较重要的，或者需要注明版权、作者的文档时使用比较多。

这里我将分享一个通过python代码为PDF文档添加水印的办法（包括文本水印和图像水印）。这种方法也适用于批量添加水印的情况。

所需工具：

这个方法将用到以下程序和组件

• Visual Studio Code
• Spire.PDF for Python组件。

下载链接：Spire.PDF for Python

试用版链接：30天免费试用

安装方法：

首先，在VS Code中安装Spire.PDF for Python和plum-dispatch v1.7.4。

在这之前请先确认是否已经安装Python。

安装好后，可以直接进行以下步骤：

1. 打开VS Code，在中搜索到Python并安装。
2. 依次选择Explorer - NO FOLRDER OPENED- Open Folder
3. 选择一个文件夹，然后在里面创建一个“ .py ”文件。
4. 创建后，点击Terminal- New Terminal
5. 输入命令： pip install Spire.PDF

示例代码

在PDF中生成文本水印

完整代码：

from spire.pdf import *
from spire.pdf.common import *
import math

#创建PdfDocument类的对象并加载一个PDF文档
pdf = PdfDocument()
pdf.LoadFromFile("C:/Users/Administrator/Desktop/Sample.pdf")

#创建一个PdfTrueTypeFont类的对象
font = PdfTrueTypeFont("Arial", 48.0, 0, True)

#设置水印文本
text = "CONFIDENTIAL"

#计算水印位置
set1 = float (font.MeasureString(text).Width * math.sqrt(2) / 4)
set2 = float (font.MeasureString(text).Height * math.sqrt(2) / 4)

#遍历所有页面
for i in range(pdf.Pages.Count):

    #获取页面
    page = pdf.Pages.get_Item(i)

    #设置水印透明度
    page.Canvas.SetTransparency(0.5)

    #将页面坐标系平移到指定位置
    page.Canvas.TranslateTransform(page.Canvas.Size.Width / 2 - set1 - set2, 
                                   page.Canvas.Size.Height / 2 + set1 - set2)

    #逆时针旋转坐标系45度
    page.Canvas.RotateTransform(-45.0)

    #绘制水印
    page.Canvas.DrawString(text, font, PdfBrushes.get_LightPink(), 0.0, 0.0)

#保存结果文档
pdf.SaveToFile("C:/Users/Administrator/Desktop/TextWatermark.pdf")
pdf.Close()

代码解释：

以上代码中，主要用到的是PdfPageBase.Canvas.DrawString()方法来绘制水印。绘制时，我们可以根据需求，利用PdfCanvas下的方法来设置水印的大小，位置，透明度，旋转角度，颜色等。具体解释可以参考一下代码注释。

水印效果如图所示：

在PDF中生成图片水印

完整代码：

from spire.pdf import *
from spire.pdf.common import *

#创建PdfDocument类的对象并加载一个PDF文档
pdf = PdfDocument()
pdf.LoadFromFile("C:/Users/Administrator/Desktop/Sample.pdf")

#加载图片
image = PdfImage.FromFile("C:/Users/Administrator/Desktop/image.png")

#获取图片的高和宽
imageWidth = float(300)
imageHeight = float(300)

#遍历所有页面
for i in range(pdf.Pages.Count):

    #获取页面
    page = pdf.Pages.get_Item(i)

    #设置水印透明度
    page.Canvas.SetTransparency(0.3)

    #获取页面的高和宽
    pageWidth = page.ActualSize.Width
    pageHeight = page.ActualSize.Height

    #在页面绘制图片水印
    page.Canvas.DrawImage(image, pageWidth/2 - imageWidth/2, pageHeight/2 - imageHeight/2, imageWidth, imageHeight)

#保存结果文档
pdf.SaveToFile("C:/Users/Administrator/Desktop/ImageWatermark.pdf")
pdf.Close()

代码解释：

与添加文本水印的代码相似，绘制时也可以根据需求，利用PdfCanvas下的方法来设置水印的大小，位置等信息。但在绘制时，使用的是PdfPageBase.Canvas.DrawImage()方法来将加载的图片作为水印添加到PDF页面中。详细解释参考代码注释。

水印效果如图所示：

如果需要其他的示例代码的话，可以直接下载这个库，里面包含了大部分比较常用的demo。