Python3,2段代码,给pdf文件添加水印,原来watermark还可以这么玩。

简介: Python3,2段代码,给pdf文件添加水印,原来watermark还可以这么玩。

1、 引言


小屌丝:鱼哥,新年快乐!

小鱼:无事不登三宝殿,有啥事,你直说吧…

小屌丝:别说的这么直接,这大过年的…

小鱼:别整没用的,就你那点小心思,我还能不知道。

小屌丝:…


小屌丝:鄙视就鄙视,只要能帮我解决问题,我然你鄙视三连!

小鱼:…还可以这样,那你说吧,啥事?

小屌丝:就是…就是…

小鱼:能不能不这么磨磨唧唧,赶紧立马撒冷的!!

小屌丝:就是,能不能帮我给pdf文件加个水印…

小鱼:就这…

小屌丝:对,就这!


小鱼:别整这么无辜又可爱的小表情。

小屌丝:谁曾经还不是个小可爱 ^ ^

小鱼:我擦,破防了,我整还不行嘛!

小屌丝:奈斯 ~ ~


今天我们就来分享 pdf文档如何添加水印。

小鱼分享过图片添加水印的方法,很简单,直接watermark模块的add_watermark 就可以,


如果不知道的,可以看这篇《Python3,2行代码添加水印,发朋友圈,图片再也不怕被盗了!!!》


但是,pdf文件添加水印,就需要用到另外两个库,即:


reportlab

pikepdf

这两个库有什么特殊的"魔法",可以给pdf添加水印呢?

别走开,稍后回来!


2、指定水印内容输出到pdf文件


2.1 模块安装


因为reportlab库是python的第三方库,所以,

第一步,安装:


pip install reportlab


其他方式安装:


《Python3,选择Python自动安装第三方库,从此跟pip说拜拜!!》

《Python3:我低调的只用一行代码,就导入Python所有库!!》


2.2 思路

1、要设置水印字体填充:

所以,我们就需要对字体设置一些基本信息,例如:


content: 水印文本内容

filename: 导出的水印文件名

width: 画布宽度,单位:mm

height: 画布高度,单位:mm

font: 对应注册的字体代号

fontsize: 字号大小

angle: 旋转角度

text_stroke_color_rgb: 文字轮廓rgb色

text_fill_color_rgb: 文字填充rgb色

text_fill_alpha: 文字透明度

2、输出水印字体到pdf文档

canvas.Canvas.save()方法,把输出的字体保存到pdf文档


2.3 代码示例


直接上代码:


# -*- coding:utf-8 -*-
# @Time   : 2022-02-10
# @Author : carl_DJ
from typing import Union,Tuple
from reportlab.lib import units
from reportlab.pdfgen import canvas
from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import  TTFont
pdfmetrics.registerFont(TTFont('msyh',r'./msyh.ttc'))
'''
用于生成包含content文字内容的水印pdf文件
content: 水印文本内容
filename: 导出的水印文件名
width: 画布宽度,单位:mm
height: 画布高度,单位:mm
font: 对应注册的字体代号
fontsize: 字号大小
angle: 旋转角度
text_stroke_color_rgb: 文字轮廓rgb色
text_fill_color_rgb: 文字填充rgb色
text_fill_alpha: 文字透明度
'''
def create_wartmark(content:str,
                    filename:str,
                    width: Union[int, float],
                    height: Union[int, float],
                    font: str,
                    fontsize: int,
                    angle: Union[int, float] = 45,
                    text_stroke_color_rgb: Tuple[int, int, int] = (0, 0, 0),
                    text_fill_color_rgb: Tuple[int, int, int] = (0, 0, 0),
                    text_fill_alpha: Union[int, float] = 1) -> None:
    #创建PDF文件,指定文件名及尺寸,以像素为单位
    c = canvas.Canvas(f'{filename}.pdf',pagesize=(width*units.mm,height*units.mm))
    #画布平移保证文字完整性
    c.translate(0.1*width*units.mm,0.1*height*units.mm)
    #设置旋转角度
    c.rotate(angle)
    #设置字体大小
    c.setFont(font,fontsize)
    #设置字体轮廓彩色
    c.setStrokeColorRGB(*text_stroke_color_rgb)
    #设置填充色
    c.setFillColorRGB(*text_fill_color_rgb)
    #设置字体透明度
    c.setFillAlpha(text_fill_alpha)
    #绘制字体内容
    c.drawString(0,0,content)
    #保存文件
    c.save()
create_wartmark(content='关注carl_奕然,学习更多有趣的python知识',
                 filename='小鱼watermarkDemo',
                 width=200,
                 height=200,
                 font='msyh',
                 fontsize=35,
                 text_fill_alpha=0.3)

 

运行结果:

image.png


3、水印内容批量输出到pdf文件


3.1 模块安装


因为pikepdf库是python的第三方库,所以,

第一步,安装:


pip install pikepdf

1

其他方式安装:


《Python3,选择Python自动安装第三方库,从此跟pip说拜拜!!》

《Python3:我低调的只用一行代码,就导入Python所有库!!》


小屌丝:鱼哥,为什么我们还需要安装pikepdf这个模块?

小鱼:因为我们要把已生成的pdf水印覆盖到目标pdf文档。

小屌丝:你的意思是,上面的代码,就仅仅是生成一个水印文档,其他的没有了?

小鱼:对的,你可以使用上面已生成的pdf水印文档,也可以随便找一个pdf文档作为水印文档,去覆盖到目标pdf文档。

小屌丝:还可以这么玩?

小鱼:对的,待会给你演示。


3.2 思路


1、需要准备pdf文件:


目标pdf文件

已生成水印的pdf文件

2、我们把已生成的pdf文档来覆盖到目标pdf文档,同样,需要设置的参数:


target_pdf_path:目标pdf文件路径+文件名

watermark_pad_path:水印pdf文件路径+文件名

nrow:水印平铺的行数

ncol:水印平铺的列数

skip_pages:需要跳过不添加水印的页数

目标pdf文档:

image.png

3.3 代码示例


# -*- coding:utf-8 -*-
# @Time   : 2022-02-10
# @Author : carl_DJ
from typing import List
from pikepdf import Pdf,Page,Rectangle
'''
向目标pdf文件批量添加水印
target_pdf_path:目标pdf文件路径+文件名
watermark_pad_path:水印pdf文件路径+文件名
nrow:水印平铺的行数
ncol:水印平铺的列数
skip_pages:需要跳过不添加水印的页数
'''
def add_watemark(target_pdf_path:str,
                 watermark_pdf_path:str,
                 nrow:int,
                 ncol:int,
                 skip_pages:List[int] = []) -> None:
    #选择需要添加水印的pdf文件
    target_pdf = Pdf.open(target_pdf_path)
    #读取水印pdf文件并提取水印
    watermark_pdf = Pdf.open(watermark_pdf_path)
    watermark_page = watermark_pdf.pages[0]
    #遍历目标pdf文件中的所有页,批量添加水印
    for idx,target_page in enumerate(target_pdf.pages):
        for x in range(ncol):
            for y in range(nrow):
                #向目标页指定范围添加水印
                target_page.add_overlay(watermark_page,
                                        Rectangle(target_page.trimbox[2] * x / ncol,
                                        target_page.trimbox[3] * y / nrow,
                                        target_page.trimbox[2] * (x + 1) / ncol,
                                        target_page.trimbox[3] * (y + 1) / nrow
                                        ))
    #保存PDF文件,同时对pdf文件进行重命名,从文件名第7位置写入后缀名
    target_pdf.save(target_pdf_path[:6] + '_已添加水印.pdf')
add_watemark(target_pdf_path='跟小鱼学水印.pdf',
             #把生成的水印示例,添加到目标水印文件中
             watermark_pdf_path='小鱼watermarkDemo.pdf',
             nrow = 3,
             ncol = 2 ,
             skip_pages= [0])


运行结果:

image.png


4、总结


写到这里,今天的分享就差不多快结束了。

今天主要是对reportlab 库和pikepdf库进行了拓展,让添加水印,不在花钱。


关注小鱼博客,学习更多更有趣python知识。


目录
相关文章
|
6天前
|
测试技术 开发者 Python
Python单元测试入门:3个核心断言方法,帮你快速定位代码bug
本文介绍Python单元测试基础,详解`unittest`框架中的三大核心断言方法:`assertEqual`验证值相等,`assertTrue`和`assertFalse`判断条件真假。通过实例演示其用法,帮助开发者自动化检测代码逻辑,提升测试效率与可靠性。
65 1
|
9天前
|
机器学习/深度学习 算法 调度
基于多动作深度强化学习的柔性车间调度研究(Python代码实现)
基于多动作深度强化学习的柔性车间调度研究(Python代码实现)
|
7天前
|
IDE 开发工具 开发者
Python类型注解:提升代码可读性与健壮性
Python类型注解:提升代码可读性与健壮性
167 102
|
5天前
|
存储 大数据 Unix
Python生成器 vs 迭代器:从内存到代码的深度解析
在Python中,处理大数据或无限序列时,迭代器与生成器可避免内存溢出。迭代器通过`__iter__`和`__next__`手动实现,控制灵活;生成器用`yield`自动实现,代码简洁、内存高效。生成器适合大文件读取、惰性计算等场景,是性能优化的关键工具。
72 2
|
9天前
|
安全 大数据 程序员
Python operator模块的methodcaller:一行代码搞定对象方法调用的黑科技
`operator.methodcaller`是Python中处理对象方法调用的高效工具,替代冗长Lambda,提升代码可读性与性能。适用于数据过滤、排序、转换等场景,支持参数传递与链式调用,是函数式编程的隐藏利器。
47 4
|
10天前
|
机器学习/深度学习 数据采集 并行计算
多步预测系列 | LSTM、CNN、Transformer、TCN、串行、并行模型集合研究(Python代码实现)
多步预测系列 | LSTM、CNN、Transformer、TCN、串行、并行模型集合研究(Python代码实现)
105 2
|
10天前
|
机器学习/深度学习 数据采集 算法
独家原创 | CEEMDAN-CNN-GRU-GlobalAttention + XGBoost组合预测研究(Python代码实现)
独家原创 | CEEMDAN-CNN-GRU-GlobalAttention + XGBoost组合预测研究(Python代码实现)
|
2月前
|
C#
【PDF提取内容改名】批量提取PDF指定区域内容重命名PDF文件,PDF自动提取内容命名的方案和详细步骤
本工具可批量提取PDF中的合同编号、日期、发票号等关键信息,支持PDF自定义区域提取并自动重命名文件,适用于合同管理、发票处理、文档归档和数据录入场景。基于iTextSharp库实现,提供完整代码示例与百度、腾讯网盘下载链接,助力高效处理PDF文档。
410 40
|
5月前
|
存储 JSON API
如何将 Swagger 文档导出为 PDF 文件
你会发现自己可能需要将 Swagger 文档导出为 PDF 或文件,以便于共享和存档。在这篇博文中,我们将指导你完成将 Swagger 文档导出为 PDF 格式的过程。
|
2月前
|
编译器 Python
如何利用Python批量重命名PDF文件
本文介绍了如何使用Python提取PDF内容并用于文件重命名。通过安装Python环境、PyCharm编译器及Jupyter Notebook,结合tabula库实现PDF数据读取与处理,并提供代码示例与参考文献。

推荐镜像

更多