Python批量给PDF加水印,再也不用一个个手动加了!

pip install reportlab pypdf

import osimport iofrom reportlab.pdfgen import canvasfrom reportlab.pdfbase import pdfmetricsfrom reportlab.pdfbase.ttfonts import TTFontfrom pypdf import PdfReader, PdfWriter# ============ 配置区 ============INPUT_DIR = "./pdfs"           # 原PDF文件夹OUTPUT_DIR = "./output"        # 输出文件夹WATERMARK_TEXT = "机密文件 请勿外传"  # 水印文字FONT_PATH = "C:/Windows/Fonts/simhei.ttf"  # 中文字体路径FONT_SIZE = 40                 # 字号OPACITY = 0.15                 # 透明度（0-1，越小越淡）ANGLE = 45                     # 旋转角度# ================================# 注册中文字体pdfmetrics.registerFont(TTFont("SimHei", FONT_PATH))def create_watermark_page(width, height, text):    """生成水印PDF页面，返回BytesIO"""    packet = io.BytesIO()    can = canvas.Canvas(packet, pagesize=(width, height))    can.setFont("SimHei", FONT_SIZE)    can.setFillAlpha(OPACITY)    can.setFillColorRGB(0.5, 0.5, 0.5)  # 灰色    # 平移到页面中心并旋转    can.translate(width / 2, height / 2)    can.rotate(ANGLE)    # 文字居中    text_width = pdfmetrics.stringWidth(text, "SimHei", FONT_SIZE)    can.drawString(-text_width / 2, 0, text)    can.save()    packet.seek(0)    return packetdef add_watermark(input_path, output_path, text):    """给单个PDF添加水印"""    reader = PdfReader(input_path)    writer = PdfWriter()    for page in reader.pages:        width = float(page.mediabox.width)        height = float(page.mediabox.height)        # 生成水印页        watermark_packet = create_watermark_page(width, height, text)        watermark_page = PdfReader(watermark_packet).pages[0]        # 合并        page.merge_page(watermark_page)        writer.add_page(page)    with open(output_path, "wb") as f:        writer.write(f)def batch_process():    """批量处理文件夹下所有PDF"""    if not os.path.exists(OUTPUT_DIR):        os.makedirs(OUTPUT_DIR)    pdf_files = [f for f in os.listdir(INPUT_DIR) if f.lower().endswith(".pdf")]    if not pdf_files:        print(f"❌ 在 {INPUT_DIR} 中没有找到PDF文件")        return    print(f"📄 共找到 {len(pdf_files)} 个PDF文件\n")    for i, filename in enumerate(pdf_files, 1):        input_path = os.path.join(INPUT_DIR, filename)        output_path = os.path.join(OUTPUT_DIR, filename)        try:            add_watermark(input_path, output_path, WATERMARK_TEXT)            print(f"✅ [{i}/{len(pdf_files)}] {filename} 处理成功")        except Exception as e:            print(f"❌ [{i}/{len(pdf_files)}] {filename} 处理失败：{e}")    print(f"\n🎉 全部完成！输出目录：{os.path.abspath(OUTPUT_DIR)}")if __name__ == "__main__":    batch_process()

📄 共找到 5 个PDF文件✅ [1/5] 报告A.pdf 处理成功✅ [2/5] 报告B.pdf 处理成功✅ [3/5] 合同C.pdf 处理成功✅ [4/5] 简历D.pdf 处理成功✅ [5/5] 说明E.pdf 处理成功🎉 全部完成！输出目录：/Users/xxx/output

def create_watermark_page(width, height, text):    packet = io.BytesIO()    can = canvas.Canvas(packet, pagesize=(width, height))    can.setFont("SimHei", FONT_SIZE)    can.setFillAlpha(OPACITY)    can.setFillColorRGB(0.5, 0.5, 0.5)    step_x, step_y = 250, 180  # 间距    for x in range(0, int(width) + step_x, step_x):        for y in range(0, int(height) + step_y, step_y):            can.saveState()            can.translate(x, y)            can.rotate(ANGLE)            can.drawString(0, 0, text)            can.restoreState()    can.save()    packet.seek(0)    return packet

from datetime import datetimetext = f"机密文件 {datetime.now().strftime('%Y-%m-%d')}"