腳本之家服務器常用軟件

快捷導航

2行Python代碼實現(xiàn)給pdf文件添加水印

更新時間：2022年02月11日 09:57:29 作者：Carl_奕然

你們在給PDF文件添加水印時，還在手動一頁頁添加嗎？本文小編為大家?guī)砹艘粋€更方便的方法，即用Python的2行代碼來實現(xiàn)，感興趣的小伙伴可以學習一下

1、引言

小屌絲：魚哥，新年快樂！

小魚：無事不登三寶殿，有啥事，你直說吧…

小屌絲：別說的這么直接，這大過年的…

小魚：別整沒用的，就你那點小心思，我還能不知道。

小屌絲：…

小屌絲：鄙視就鄙視，只要能幫我解決問題，我然你鄙視三連！

小魚：…還可以這樣，那你說吧，啥事？

小屌絲：就是…就是…

小魚：能不能不這么磨磨唧唧，趕緊立馬撒冷的！！

小屌絲：就是，能不能幫我給pdf文件加個水印…

小魚：就這…

小屌絲：對，就這!

小魚：別整這么無辜又可愛的小表情。

小屌絲：誰曾經(jīng)還不是個小可愛 ^ ^

小魚：我擦，破防了，我整還不行嘛！

小屌絲：奈斯 ~ ~

今天我們就來分享 pdf文檔如何添加水印。

小魚分享過圖片添加水印的方法，很簡單，直接watermark模塊的add_watermark 就可以，

如果不知道的，可以看這篇《Python3，2行代碼添加水印，發(fā)朋友圈，圖片再也不怕被盜了?。?！》

但是，pdf文件添加水印，就需要用到另外兩個庫，即：

reportlab
pikepdf

這兩個庫有什么特殊的"魔法"，可以給pdf添加水印呢？

別走開，稍后回來！

2、指定水印內(nèi)容輸出到pdf文件

2.1 模塊安裝

因為reportlab庫是python的第三方庫，所以，

第一步，安裝：

pip install reportlab

其他方式安裝：

《Python3，選擇Python自動安裝第三方庫，從此跟pip說拜拜?。　?/a>

《Python3：我低調(diào)的只用一行代碼，就導入Python所有庫?。　?/a>

2.2 思路

1、要設置水印字體填充：

所以，我們就需要對字體設置一些基本信息，例如：

content: 水印文本內(nèi)容
filename: 導出的水印文件名
width: 畫布寬度，單位：mm
height: 畫布高度，單位：mm
font: 對應注冊的字體代號
fontsize: 字號大小
angle: 旋轉(zhuǎn)角度
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文字內(nèi)容的水印pdf文件

content: 水印文本內(nèi)容
filename: 導出的水印文件名
width: 畫布寬度，單位：mm
height: 畫布高度，單位：mm
font: 對應注冊的字體代號
fontsize: 字號大小
angle: 旋轉(zhuǎn)角度
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:


    #創(chuàng)建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)

    #設置旋轉(zhuǎn)角度
    c.rotate(angle)

    #設置字體大小
    c.setFont(font,fontsize)

    #設置字體輪廓彩色
    c.setStrokeColorRGB(*text_stroke_color_rgb)

    #設置填充色
    c.setFillColorRGB(*text_fill_color_rgb)

    #設置字體透明度
    c.setFillAlpha(text_fill_alpha)

    #繪制字體內(nèi)容
    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)

運行結(jié)果：

3、水印內(nèi)容批量輸出到pdf文件

3.1 模塊安裝

因為pikepdf庫是python的第三方庫，所以，

第一步，安裝：

pip install pikepdf

其他方式安裝：

《Python3，選擇Python自動安裝第三方庫，從此跟pip說拜拜??！》

《Python3：我低調(diào)的只用一行代碼，就導入Python所有庫??！》

小屌絲：魚哥，為什么我們還需要安裝pikepdf這個模塊？

小魚：因為我們要把已生成的pdf水印覆蓋到目標pdf文檔。

小屌絲：你的意思是，上面的代碼，就僅僅是生成一個水印文檔，其他的沒有了？

小魚：對的，你可以使用上面已生成的pdf水印文檔，也可以隨便找一個pdf文檔作為水印文檔，去覆蓋到目標pdf文檔。

小屌絲：還可以這么玩？

小魚：對的，待會給你演示。

3.2 思路

1、需要準備pdf文件：

目標pdf文件
已生成水印的pdf文件

2、我們把已生成的pdf文檔來覆蓋到目標pdf文檔，同樣，需要設置的參數(shù)：

target_pdf_path:目標pdf文件路徑+文件名
watermark_pad_path:水印pdf文件路徑+文件名
nrow:水印平鋪的行數(shù)
ncol:水印平鋪的列數(shù)
skip_pages:需要跳過不添加水印的頁數(shù)

目標pdf文檔：

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:水印平鋪的行數(shù)
ncol:水印平鋪的列數(shù)
skip_pages:需要跳過不添加水印的頁數(shù)

'''

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])

運行結(jié)果：