使用Python實(shí)現(xiàn)為PDF文件添加圖章

更新時(shí)間：2023年11月23日 09:09:16 作者：Python 集中營

在日常工作中,我們經(jīng)常需要給PDF文檔添加一些標(biāo)識,比如公司的圖章或水印圖章,所以本文就來為大家詳細(xì)介紹一下如何使用Python實(shí)現(xiàn)為PDF文件添加圖章,需要的可以參考下

在日常工作中，我們經(jīng)常需要給PDF文檔添加一些標(biāo)識，比如公司的圖章或水印圖章。

使用Python可以很方便地實(shí)現(xiàn)這個功能。添加圖章或水印是操作 PDF 文件的兩種常用方法。圖章是在文檔頂部添加內(nèi)容，水印是在文檔的背景中。

在這兩種情況下，您可能都希望確保原始內(nèi)容的媒體框/裁剪框保持不變。

本文將介紹如何使用PyPDF2、pathlib和typing模塊來給PDF文檔添加圖章或水印圖章。

PyPDF2是一個用于處理PDF文件的Python庫，它可以讀取、寫入和操作PDF文件。

pathlib是Python 3中用于處理文件和目錄路徑的模塊，它提供了一種簡單而直觀的方式來操作文件系統(tǒng)。

typing模塊是Python 3.5中引入的一個模塊，它提供了一種用于類型注釋的方式，可以增強(qiáng)代碼的可讀性和可維護(hù)性。

首先，我們需要安裝PyPDF2庫。可以使用pip命令來安裝：

pip install PyPDF2

接下來，我們需要創(chuàng)建一個Python腳本，并導(dǎo)入所需的模塊：

import PyPDF2
from pathlib import Path
from typing import Tuple

然后，我們定義一個函數(shù)stamp()來添加覆蓋原有文字的圖章。

from pathlib import Path
from typing import Union, Literal, List

from PyPDF2 import PdfWriter, PdfReader


def stamp(
    content_pdf: Path,
    stamp_pdf: Path,
    pdf_result: Path,
    page_indices: Union[Literal["ALL"], List[int]] = "ALL",
):
    reader = PdfReader(stamp_pdf)
    image_page = reader.pages[0]

    writer = PdfWriter()

    reader = PdfReader(content_pdf)
    if page_indices == "ALL":
        page_indices = list(range(0, len(reader.pages)))
    for index in page_indices:
        content_page = reader.pages[index]
        mediabox = content_page.mediabox
        content_page.merge_page(image_page)
        content_page.mediabox = mediabox
        writer.add_page(content_page)

    with open(pdf_result, "wb") as fp:
        writer.write(fp)

下面是將圖章覆蓋模式添加到文字上面的效果圖，具體可以根據(jù)自身的需求進(jìn)行調(diào)整。

現(xiàn)在，我們可以調(diào)用watermark()這個函數(shù)來給PDF文檔添加水印圖章。

下面是一個示例：

from pathlib import Path
from typing import Union, Literal, List

from PyPDF2 import PdfWriter, PdfReader


def watermark(
    content_pdf: Path,
    stamp_pdf: Path,
    pdf_result: Path,
    page_indices: Union[Literal["ALL"], List[int]] = "ALL",
):
    reader = PdfReader(content_pdf)
    if page_indices == "ALL":
        page_indices = list(range(0, len(reader.pages)))

    writer = PdfWriter()
    for index in page_indices:
        content_page = reader.pages[index]
        mediabox = content_page.mediabox

        # You need to load it again, as the last time it was overwritten
        reader_stamp = PdfReader(stamp_pdf)
        image_page = reader_stamp.pages[0]

        image_page.merge_page(content_page)
        image_page.mediabox = mediabox
        writer.add_page(image_page)

    with open(pdf_result, "wb") as fp:
        writer.write(fp)

處理過程相差不大，最后通過PdfWriter將結(jié)果寫入到文檔中，下面是水印圖章的效果圖：