2行Python程式碼實現給pdf檔案新增水印

pip install reportlab

# -*- 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) 

pip install pikepdf

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