Python實(shí)現(xiàn)批量向PDF文件添加中文水印
前言
可以通過設(shè)置批量PDF文件所在的路徑及需要添加的水印名稱可以實(shí)現(xiàn)批量添加PDF水印的效果。
實(shí)現(xiàn)思路是這樣的,通過在批量PDF文件路徑下面生成一個帶有水印的PDF模板。最后,將批量文件的每個PDF頁面和水印模板進(jìn)行合并完成批量添加水印的效果。
需要注意的是批量PDF文件必須和PDF模板水印文件的大小尺寸保持一致,這個可以在代碼里面調(diào)節(jié)一下就成了。
實(shí)現(xiàn)步驟
首先將需要添加水印的PDF文件準(zhǔn)備好放在一個文件夾下面。
在代碼中設(shè)置好PDF批量文件的路徑及水印名稱。
if?__name__?==?'__main__': ????main('C:/pdf',?'我是一個水印')
內(nèi)部實(shí)現(xiàn)過程都封裝在main()函數(shù)里面了,這里改一下水印名稱和批量PDF文件路徑直接執(zhí)行就好了。
啟動以后,出現(xiàn)如下面的結(jié)果說明已經(jīng)執(zhí)行完成了。
為了不覆蓋原來的PDF文件,合并后的文件都是添加了"已合并"字樣的PDF文件。
說完了怎么操作,看一下主要的代碼塊部分有哪些吧。
其中用到的第三方庫有下面這些,里面我寫了相關(guān)的注釋。
import?os??#?應(yīng)用文件操作 # reportlab是Python的一個標(biāo)準(zhǔn)庫,可以畫圖、畫表格、編輯文字,最后可以輸出PDF格式。 from?reportlab.pdfgen?import?canvas from?reportlab.lib.units?import?cm from?reportlab.pdfbase?import?pdfmetrics from?reportlab.pdfbase.ttfonts?import?TTFont pdfmetrics.registerFont(TTFont('songti',?'C:/Windows/Fonts/simsun.ttc'))??#?加載宋體 # PyPDF2模塊主要的功能是分割或合并PDF文件,裁剪或轉(zhuǎn)換PDF文件中的頁面。 from?PyPDF2?import?PdfFileWriter,?PdfFileReader import?logging??#?日志打印庫
日志模塊的初始化也比較簡單,前面的文章中都有過相關(guān)的調(diào)用。
#?初始化日志設(shè)置 logger?=?logging.getLogger('批量添加水印') logging.basicConfig(format='%(asctime)s?%(levelname)-8s:?%(message)s') logger.setLevel(logging.DEBUG)
日志初始化完成后在后面需要打印日志的地方調(diào)用就可以了。
實(shí)現(xiàn)過程主要有三個函數(shù)來實(shí)現(xiàn)的,一個是為了生成水印模板、另一個是使水印模板和批量PDF文件執(zhí)行合并從而實(shí)現(xiàn)添加水印的功能、還有一個就是逐個遍歷批量PDF文件使其能夠逐個實(shí)現(xiàn)水印合并。
水印模板生成函數(shù)。
def?generate_water_pdf(content): ????''' ????生成帶有水印的PDF ????:param?content:?水印名稱 ????:return: ????''' ????cans?=?canvas.Canvas('water_back.pdf',?pagesize=(21?*?cm,?29.7?*?cm)) ????cans.translate(10?*?cm, ???????????????????12?*?cm)??#?移動原點(diǎn)坐標(biāo) ????cans.setFont('songti',?23)??#?設(shè)置字體為宋體、大小為23號 ????cans.setFillColorRGB(0.5,?0.5, ?????????????????????????0.5)??#?設(shè)置字體背景顏色 ????cans.rotate(45)??#?設(shè)置字體傾斜45度 ????cans.drawString(-7?*?cm,?0?*?cm,?content) ????cans.drawString(7?*?cm,?0?*?cm,?content) ????cans.drawString(0?*?cm,?7?*?cm,?content) ????cans.drawString(0?*?cm,?-7?*?cm,?content) ????cans.save()??#?保存水印的PDF文件
水印合成實(shí)現(xiàn)函數(shù)。
def?insert_water_to_pdf(input_pdf,?output_pdf,?water_pdf): ????''' ????合并水印到PDF文件中 ????:param?input_pdf:?輸入文件路徑 ????:param?output_pdf:?輸出文件路徑 ????:param?water_pdf:?水印文件路徑 ????:return: ????''' ????water?=?PdfFileReader(water_pdf)??#?讀取水印PDF ????water_page?=?water.getPage(0)??#?獲取水印PDF的第一頁 ????pdf?=?PdfFileReader(input_pdf,?strict=False)??#?讀取需要添加水印的文件 ????pdf_writer?=?PdfFileWriter()??#?創(chuàng)建PDF文件寫入對象 ????for?page?in?range(pdf.getNumPages()):??#?遍歷每一頁P(yáng)DF對象 ????????pdf_page?=?pdf.getPage(page)??#?獲取PDF的當(dāng)前頁對象 ????????pdf_page.mergePage(water_page)??#?將水印頁合并到當(dāng)前頁中 ????????pdf_writer.addPage(pdf_page)??#?將合并后的PDF對象頁添加到PDF寫入對象中 ????output_file?=?open(output_pdf,?'wb')??#?打開PDF輸出文件 ????pdf_writer.write(output_file)??#?將文件寫入到輸出文件 ????output_file.close()??#?關(guān)閉寫入流
批量PDF文件遍歷調(diào)用合成函數(shù)。
def?main(diretory,?current): ????if?os.path.isdir(diretory): ????????logger.info('文件夾['?+?diretory?+?']校驗(yàn)成功!') ????????os.chdir(diretory) ????????logger.info('當(dāng)前路徑為['?+?os.getcwd()?+?']') ????????generate_water_pdf(current) ????????logger.info('水印PDF文件生成成功!') ????????for?file_path,?dir_names,?file_names?in?os.walk(r''?+?os.getcwd()): ????????????for?file_name?in?file_names: ????????????????try: ????????????????????name?=?file_name.split('.')[0] ????????????????????if?name?==?'water_back': ????????????????????????continue ????????????????????else: ????????????????????????file_name_path?=?os.path.join(file_path,?file_name) ????????????????????????output_file_path?=?file_name_path.split('.')[0]?+?'_已添加水印.pdf' ????????????????????????insert_water_to_pdf(file_name_path,?output_file_path,?'water_back.pdf') ????????????????????????logger.info('['?+?file_name_path?+?']完成水印合并!') ????????????????except?Exception?as?e: ????????????????????logger.error('['?+?file_name_path?+?']發(fā)生異常,執(zhí)行下一個!') ????????????????????logger.error('異常信息:'?+?repr(e)) ????else: ????????logger.info('文件夾['?+?diretory?+?']校驗(yàn)失??!')
主要實(shí)現(xiàn)過程就是通過上面三個函數(shù)來完成的,最后調(diào)用后臺入口函數(shù)將mian()函數(shù)調(diào)用執(zhí)行就可以了。
完整代碼
# -*- coding:utf-8 -*- # @author Python 集中營 # @date 2022/1/27 # @file test4.py # done # 批量向PDF文件添加中文水印 import os # 應(yīng)用文件操作 # reportlab是Python的一個標(biāo)準(zhǔn)庫,可以畫圖、畫表格、編輯文字,最后可以輸出PDF格式。 from reportlab.pdfgen import canvas from reportlab.lib.units import cm from reportlab.pdfbase import pdfmetrics from reportlab.pdfbase.ttfonts import TTFont pdfmetrics.registerFont(TTFont('songti', 'C:/Windows/Fonts/simsun.ttc')) # 加載宋體 # PyPDF2模塊主要的功能是分割或合并PDF文件,裁剪或轉(zhuǎn)換PDF文件中的頁面。 from PyPDF2 import PdfFileWriter, PdfFileReader import logging # 日志打印庫 # 初始化日志設(shè)置 logger = logging.getLogger('批量添加水印') logging.basicConfig(format='%(asctime)s %(levelname)-8s: %(message)s') logger.setLevel(logging.DEBUG) def generate_water_pdf(content): ''' 生成帶有水印的PDF :param content: 水印名稱 :return: ''' cans = canvas.Canvas('water_back.pdf', pagesize=(21 * cm, 29.7 * cm)) cans.translate(10 * cm, 12 * cm) # 移動原點(diǎn)坐標(biāo) cans.setFont('songti', 23) # 設(shè)置字體為宋體、大小為23號 cans.setFillColorRGB(0.5, 0.5, 0.5) # 設(shè)置字體背景顏色 cans.rotate(45) # 設(shè)置字體傾斜45度 cans.drawString(-7 * cm, 0 * cm, content) cans.drawString(7 * cm, 0 * cm, content) cans.drawString(0 * cm, 7 * cm, content) cans.drawString(0 * cm, -7 * cm, content) cans.save() # 保存水印的PDF文件 def insert_water_to_pdf(input_pdf, output_pdf, water_pdf): ''' 合并水印到PDF文件中 :param input_pdf: 輸入文件路徑 :param output_pdf: 輸出文件路徑 :param water_pdf: 水印文件路徑 :return: ''' water = PdfFileReader(water_pdf) # 讀取水印PDF water_page = water.getPage(0) # 獲取水印PDF的第一頁 pdf = PdfFileReader(input_pdf, strict=False) # 讀取需要添加水印的文件 pdf_writer = PdfFileWriter() # 創(chuàng)建PDF文件寫入對象 for page in range(pdf.getNumPages()): # 遍歷每一頁P(yáng)DF對象 pdf_page = pdf.getPage(page) # 獲取PDF的當(dāng)前頁對象 pdf_page.mergePage(water_page) # 將水印頁合并到當(dāng)前頁中 pdf_writer.addPage(pdf_page) # 將合并后的PDF對象頁添加到PDF寫入對象中 output_file = open(output_pdf, 'wb') # 打開PDF輸出文件 pdf_writer.write(output_file) # 將文件寫入到輸出文件 output_file.close() # 關(guān)閉寫入流 def main(diretory, current): if os.path.isdir(diretory): logger.info('文件夾[' + diretory + ']校驗(yàn)成功!') os.chdir(diretory) logger.info('當(dāng)前路徑為[' + os.getcwd() + ']') generate_water_pdf(current) logger.info('水印PDF文件生成成功!') for file_path, dir_names, file_names in os.walk(r'' + os.getcwd()): for file_name in file_names: try: name = file_name.split('.')[0] if name == 'water_back': continue else: file_name_path = os.path.join(file_path, file_name) output_file_path = file_name_path.split('.')[0] + '_已添加水印.pdf' insert_water_to_pdf(file_name_path, output_file_path, 'water_back.pdf') logger.info('[' + file_name_path + ']完成水印合并!') except Exception as e: logger.error('[' + file_name_path + ']發(fā)生異常,執(zhí)行下一個!') logger.error('異常信息:' + repr(e)) else: logger.info('文件夾[' + diretory + ']校驗(yàn)失敗!') if __name__ == '__main__': main('C:/pdf', '我是一個水印')
到此這篇關(guān)于Python實(shí)現(xiàn)批量向PDF文件添加中文水印的文章就介紹到這了,更多相關(guān)Python PDF添加水印內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
django中使用POST方法獲取POST數(shù)據(jù)
這篇文章主要介紹了django中使用POST方法獲取POST數(shù)據(jù),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-08-08Python類中的裝飾器在當(dāng)前類中的聲明與調(diào)用詳解
這篇文章主要介紹了Python類中的裝飾器在當(dāng)前類中的聲明與調(diào)用詳解,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-04-04Python中解析JSON并同時進(jìn)行自定義編碼處理實(shí)例
這篇文章主要介紹了Python中解析JSON并同時進(jìn)行自定義編碼處理實(shí)例,需要的朋友可以參考下2015-02-02python實(shí)現(xiàn)文件快照加密保護(hù)的方法
這篇文章主要介紹了python實(shí)現(xiàn)文件快照加密保護(hù)的方法,涉及Python文件加密的技巧,可有效防止文件被篡改,需要的朋友可以參考下2015-06-06Python pymysql連接數(shù)據(jù)庫并將查詢結(jié)果轉(zhuǎn)化為Pandas dataframe
這篇文章主要為大家介紹了Python pymysql連接數(shù)據(jù)庫并將結(jié)果轉(zhuǎn)化為Pandas dataframe實(shí)現(xiàn)方法詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-05-05使用OpenCV獲取圖片連通域數(shù)量,并用不同顏色標(biāo)記函
這篇文章主要介紹了使用OpenCV獲取圖片連通域數(shù)量,并用不同顏色標(biāo)記函,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06Python寫捕魚達(dá)人的游戲?qū)崿F(xiàn)
這篇文章主要介紹了Python寫捕魚達(dá)人的游戲?qū)崿F(xiàn),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-03-03