[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。