2行Python代碼實現給pdf文件添加水印
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)
運行結果:
3、水印內容批量輸出到pdf文件
3.1 模塊安裝
因為pikepdf庫是python的第三方庫,所以,
第一步,安裝:
pip install pikepdf
其他方式安裝:
《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文檔:
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])
運行結果:
4、總結
寫到這裡,今天的分享就差不多快結束瞭。
今天主要是對reportlab 庫和pikepdf庫進行瞭拓展,讓添加水印,不在花錢。
以上就是2行Python代碼實現給pdf文件添加水印的詳細內容,更多關於Python pdf添加水印的資料請關註WalkonNet其它相關文章!
推薦閱讀:
- Python對PDF文件的常用操作方法詳解
- vue實現頁面添加水印效果
- Python 操作pdf pdfplumber讀取PDF寫入Exce
- 用python實現PDF解密打印文件
- python使用reportlab生成pdf實例