Python给PDF添加水印(极速版)

使用PyMuPDF给PDF文件添加水印,秒级处理!!!直接上代码

python 复制代码
from PyPDF2 import PdfReader, PdfWriter

def add_watermark_pymupdf(input_pdf, watermark_pdf, output_pdf,
                          position='full', opacity=0.3):
    """
    使用PyMuPDF给PDF添加水印文件

    Args:
        input_pdf: 输入PDF路径
        watermark_pdf: 水印PDF文件路径
        output_pdf: 输出PDF路径
        position: 水印位置 ('full', 'center', 'bottom-right', 'top-left')
        opacity: 水印透明度 (0.0-1.0)

    Returns:
        处理后的PDF路径
    """
    start_time = time.time()

    # 打开源PDF和水印PDF
    src_doc = fitz.open(input_pdf)
    watermark_doc = fitz.open(watermark_pdf)

    # 获取水印页面(假设水印PDF只有一页)
    watermark_page = watermark_doc[0]

    # 获取水印页面尺寸
    watermark_rect = watermark_page.rect
    watermark_width = watermark_rect.width
    watermark_height = watermark_rect.height

    total_pages = len(src_doc)
    print(f"开始处理 {total_pages} 页PDF...")

    # 处理每一页
    for page_num in range(total_pages):
        if page_num % 10 == 0:
            print(f"处理进度: {page_num + 1}/{total_pages}")

        page = src_doc[page_num]
        page_rect = page.rect
        page_width = page_rect.width
        page_height = page_rect.height

        # 全页面覆盖
        scale_x = page_width / watermark_width
        scale_y = page_height / watermark_height
        scale = min(scale_x, scale_y)

        new_width = watermark_width * scale
        new_height = watermark_height * scale

        x = (page_width - new_width) / 2
        y = (page_height - new_height) / 2

        clip_rect = fitz.Rect(0, 0, watermark_width, watermark_height)

        # 创建要插入的矩形区域
        target_rect = fitz.Rect(x, y, x + new_width, y + new_height)

        # 插入水印
        page.show_pdf_page(
            target_rect,  # 在页面上的位置
            watermark_doc,  # 水印文档
            0,  # 水印页码
            clip=clip_rect,  # 裁剪区域
            keep_proportion=True,  # 保持比例
            overlay=True,  # 覆盖模式(作为水印)
            oc=0  # 可选内容组(用于透明度)
        )

    # 保存输出
    src_doc.save(output_pdf)
    src_doc.close()
    watermark_doc.close()

    elapsed = time.time() - start_time
    # print(f"处理完成!耗时: {elapsed:.2f}秒,平均每页: {elapsed / total_pages:.3f}秒")

    return output_pdf

注意 :水印可以直接插入到PDF中,然后使用脚本将水印PDF和目标PDF合并

相关推荐
恩喜玛生物17 分钟前
AIDD系列02:ProteinMPNN序列设计,从已知骨架到可检验的肽候选
python·ai
FelixZhang02844 分钟前
不懂金融或不会编程,如何开始量化?从策略搭建到回测验证
人工智能·python·算法·数学建模·均值算法·自然语言处理·金融
打工仔折腾 AI1 小时前
后端部署避坑:吃透IPv4地址、网络接口与CIDR前缀的实际应用
arm开发·人工智能·后端·python·性能优化
开开心心就好1 小时前
免安装PDF盖章工具,透明度和位置可调
运维·服务器·前端·javascript·pdf·ocr·1024程序员节
wangchunyu11411 小时前
JAVA CompletableFuture异步编程实战
java·开发语言·python
2501_9445215913 小时前
Immer 跨端三方库仓颉适配全流程
python·eclipse·django
信誓旦旦的程序猿13 小时前
【Python量化实战宝典 #26】止损与回撤控制手册:移动止损、最大回撤(Python 可跑)
python·1024程序员节·股票数据api·股票数据api接口·股票api数据接口·股票量化数据api·股票量化数据接口
2501_9445215913 小时前
Deepmerge 跨端三方库仓颉适配全流程
java·python·eclipse·github
Patrick在香港13 小时前
同一行 strftime,两台机器跑出两种日期
python·datetime·时区·locale·日期处理
2601_9679873714 小时前
用Python+OpenCV+YOLO写个游戏脚本:从自动瞄准到自动采集的保姆级教程
python·opencv·yolo·目标检测·游戏自动化