?

本文完整示例代碼及文件已上傳至我的Github倉庫https://github.com/CNFeffery/PythonPracticalSkills

?

這是我的系列文章「Python實用秘技」的第4期，本系列立足于筆者日常工作中使用Python積累的心得體會，每一期為大家?guī)硪粋€3分鐘即可學會的簡單小技巧。

作為系列第4期，我們即將學習的是：為pdf文件批量添加文字水印。

有些情況下我們需要為單個或多個pdf文件添加文字水印，尤其是那種需要在每一頁按照一定間距鋪滿的文字水印。而借助reportlab和pikepdf這兩個實用的pdf文件操作庫，我們就可以很方便地實現(xiàn)批量文字水印添加工作。

利用pip install reportlab pikepdf完成安裝后，我們就可以按照步驟來實現(xiàn)需要的功能：

• 生成指定的文本水印pdf文件

為了向目標pdf文件添加水印，我們首先需要有單獨的pdf格式的文本水印文件，我用reportlab編寫了一個方便易用的函數(shù)來生成水印文件，你可以通過注釋來仔細學習其中的步驟，也可以直接調(diào)用即可：

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

#?注冊字體，這里的字體是我從windows的字體目錄下復(fù)制過來的
pdfmetrics.registerFont(TTFont('msyh',?r'./msyh.ttc'))

def?create_watermark(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:
????'''
????用于生成包含content文字內(nèi)容的水印pdf文件
????content:?水印文本內(nèi)容
????filename:?導(dǎo)出的水印文件名
??? width:?畫布寬度，單位：mm
??? height:?畫布高度，單位：mm
????font:?對應(yīng)注冊的字體代號
????fontsize:?字號大小
????angle:?旋轉(zhuǎn)角度
????text_stroke_color_rgb:?文字輪廓rgb色
????text_fill_color_rgb:?文字填充rgb色
????text_fill_alpha:?文字透明度
????'''

????#?創(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)
????
????#?設(shè)置旋轉(zhuǎn)角度
????c.rotate(angle)
????
????#?設(shè)置字體及字號大小
????c.setFont(font,?fontsize)
????
????#?設(shè)置文字輪廓色彩
????c.setStrokeColorRGB(*text_stroke_color_rgb)
????
????#?設(shè)置文字填充色
????c.setFillColorRGB(*text_fill_color_rgb)
????
????#?設(shè)置文字填充色透明度
????c.setFillAlpha(text_fill_alpha)
????
????#?繪制文字內(nèi)容
????c.drawString(0,?0,?content)
????
????#?保存水印pdf文件
????c.save()

下面我們就利用這個函數(shù)來生成水印文件：

#?制造示例文字水印pdf文件
create_watermark(content='公眾號【Python大數(shù)據(jù)分析】作者：費弗里',?
?????????????????filename='水印示例',?
?????????????????width=200,
?????????????????height=200,?
?????????????????font='msyh',?
?????????????????fontsize=35,
?????????????????text_fill_alpha=0.3)

看看效果，非常的不錯，具體使用時，你可以自己動手調(diào)參以找到大小以及畫幅都令你滿意的水印導(dǎo)出結(jié)果：

• 將水印文件批量覆蓋到目標pdf文件中

搞定了文本水印文件的生成之后，接下來我們就可以把現(xiàn)成的水印文件插入到目標pdf文件中，這里我們使用pikepdf中的相關(guān)功能就可以輕松實現(xiàn)，我寫了一個簡單的函數(shù)，大家在調(diào)用時只需要傳入幾個必要參數(shù)即可：

from?typing?import?List
from?pikepdf?import?Pdf,?Page,?Rectangle

def?add_watermark(target_pdf_path:?str,
??????????????????watermark_pdf_path:?str,
??????????????????nrow:?int,
??????????????????ncol:?int,
??????????????????skip_pages:?List[int]?=?[])?->?None:
????'''
????向目標pdf文件中添加平鋪水印
????target_pdf_path:?目標pdf文件的路徑+文件名
????watermark_pdf_path:?水印pdf文件的路徑+文件名
????nrow:?水印平鋪的行數(shù)
??? ncol：水印平鋪的列數(shù)
????skip_pages:?需要跳過不添加水印的頁面序號（從0開始）
????'''
????
????#?讀入需要添加水印的pdf文件
????target_pdf?=?Pdf.open(target_pdf_path)
????
????#?讀入水印pdf文件并提取水印頁
????watermark_pdf?=?Pdf.open(watermark_pdf_path)
????watermark_page?=?watermark_pdf.pages[0]
????
????#?遍歷目標pdf文件中的所有頁（排除skip_pages指定的若干頁）
????for?idx,?target_page?in?enumerate(target_pdf.pages):
????????
????????if?idx?not?in?skip_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))
????????????????????
????#?將添加完水印后的結(jié)果保存為新的pdf
????target_pdf.save(target_pdf_path[:-4]+'_已添加水印.pdf')

下面我們直接調(diào)用這個函數(shù)，對示例文件【吳恩達】機器學習訓練秘籍-中文版.pdf中除了封面以外的每一頁，按照3行2列的平鋪密度，添加上我們的示例水印：

add_watermark(target_pdf_path='./【吳恩達】機器學習訓練秘籍-中文版.pdf',
??????????????watermark_pdf_path='./水印示例.pdf',
??????????????nrow=3,
??????????????ncol=2,
??????????????skip_pages=[0])

效果杠杠的，讀者朋友們可以自己多試試，得到更多心得體會~

本期分享結(jié)束，咱們下回見~??