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

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

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

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

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

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

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

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

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

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

首先，我們需要安裝PyPDF2庫(kù)?？梢允褂胮ip命令來安裝：

pip install PyPDF2

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

import PyPDF2
from pathlib import Path
from typing import Tuple

然后，我們定義一個(gè)函數(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()這個(gè)函數(shù)來給PDF文檔添加水印圖章。

下面是一個(gè)示例：

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é)果寫入到文檔中，下面是水印圖章的效果圖：