详解Python如何实现批量为PDF添加水印

我们有时候需要把一些机密文件发给多个客户，为了避免客户泄露文件，会在机密文件中添加水印。每个客户收到的文件内容相同，但是水印都不相同。这样一来，如果资料泄露了，通过水印就知道是从谁手上泄露的。

今天，一个做市场的朋友找我咨询PDF加水印的问题，如下图所示：

他有一个Excel文件，文件里面有10000个经销商的名字，他要把价目表PDF发给这些经销商，每个经销商收到的PDF文件上面的水印都是这个经销商自己的名字。

这个需求手动操作肯定要累死人。但是如果用Python来做，就非常简单。代码不超过30行。

网址:yii666.com

准备环境

reportlabpikepdf

.ttf.ttc

/System/Library/Fonts.ttc/System/Library/Fonts/Supplemental/Songti.ttcSongti.ttc

获得经销商名字对应的列表

由于这位朋友不会使用pandas，那么我们就尽量使用Python原生的方法来获得经销商名字列表。假设经销商信息对应的Excel如下图所示：

文章来源地址:https://www.yii666.com/blog/286513.html

我们首先把这个Excel文件导出成csv文件：

网址:yii666.com<

然后，我们用Python读取这个csv文件，获得经销商名字列表：

运行效果如下图所示：

生成水印PDF

一般来说，我们不能直接把一段文字作为水印添加到另一个PDF文件中。我们只有先把这段文字生成图片或者生成水印PDF文件，然后把这个图片或者水印PDF作为『图层』覆盖到目标PDF上面。

因此，现在需要给每一个经销商生成对应的水印PDF文件。这个PDF中只含有水印文字。效果如下图所示：

对应的代码create_watermark.py如下：

water_pdf

合并水印与目标PDF

最后一步，把每一个经销商的水印PDF与目标PDF进行合并。水印PDF作为一个图层覆盖到目标PDF上面。

pikepdf

result

这里有必要对代码中的一些地方进行解释。带上行号的代码如下图所示：

for

row

page.trimbox[2]page.trimbox[3]

总结

大家注意在这篇文章中，我把任务分成了3个部分，分别是：

• Excel转CSV，让Python方便读取
• Python读取CSV生成水印PDF
• 水印PDF与目标PDF文件合并

.py

在计算机领域，所有问题都可以通过把问题拆分成多个部分分别单独运行或者增加若干个中间层来解决。今天用的方法就是把问题拆分的方法。对于初学者来说，每一步都是相对独立的，都能立刻看到效果。第二步只需要依赖第一步的结果，第三步只需要依赖第二步的结果，这样每一步的输入输出非常清楚，可以显著降低问题的复杂度。如果报错了，也更容易知道是哪个地方有问题。