【DuodooBMS】给PDF附件加“受控”水印的完整Python实现

给PDF附件加“受控”水印的完整Python实现

功能需求
在实际工作中，许多文件需要添加水印以标识其状态，比方“受控”“秘密”等。对于PDF文件，添加水印不仅可以增强文件的可识别性，还可以防止未经授权的使用。本代码的功能需求是：

• 修复PDF文件：在添加水印之前，确保PDF文件是完整且可读的，避免因文件损坏导致操纵失败。
  
• 添加水印：在PDF的每一页上添加指定的水印图像或笔墨，水印可以设置位置、角度和透明度。
  
• 生存输出：将添加水印后的PDF文件生存到指定路径，并返回其二进制数据以便后续处理。
  

实现过程

• 修复PDF文件
  
  
  
  • 使用PyMuPDF库打开PDF文件，并实验修复。如果文件损坏，PyMuPDF可以实验修复并生存为一个新的二进制流。
    
  • 如果修复失败，则直接返回原始的PDF二进制数据。
    
  Python复制
  
  1. def repair_pdf(self, input_pdf_binary):
  2.     try:
  3.         # 使用 PyMuPDF 打开并修复 PDF
  4.         doc = fitz.open(stream=input_pdf_binary, filetype="pdf")
  5.         repaired_pdf_binary = BytesIO()
  6.         doc.save(repaired_pdf_binary)
  7.         doc.close()
  8.         repaired_pdf_binary.seek(0)
  9.         return repaired_pdf_binary.read()
  10.     except Exception as e:
  11.         print(f"Error repairing PDF: {e}")
  12.         return input_pdf_binary

  复制代码
  
  
• 添加水印
  
  
  
  • 使用reportlab库创建一个临时的PDF文件作为水印。水印可以是图像或笔墨，支持设置位置、角度和透明度。
    
  • 使用PyPDF2库将水印PDF与原始PDF合并。通过merge_page方法，将水印添加到每一页。
    
  Python复制
  
  1. def add_watermark(self, input_pdf_binary, output_pdf, watermark_image, x_position=30, y_position=50, opacity=1):
  2.     # 尝试修复 PDF
  3.     repaired_pdf_binary = self.repair_pdf(input_pdf_binary)
  4.     input_pdf_obj = PdfReader(BytesIO(repaired_pdf_binary))  # 从二进制数据中读取 PDF
  5.     output_pdf_obj = PdfWriter()
  6.     output_buffer = BytesIO()
  8.     # 创建一个临时的 PDF 作为水印
  9.     page_width, page_height = A4[1], A4[0]
  10.     c = canvas.Canvas(output_buffer, pagesize=(page_width, page_height))
  11.     try:
  12.         c.setFillColor(colors.white)  # 将背景设置为白色
  13.         c.setFillColor(colors.red)  # 设置字体颜色为红色
  14.         c.setFont("Helvetica", 12)  # 设置字体和字体大小
  15.         c.setFillAlpha(opacity)  # 设置透明度
  16.         c.setStrokeColor(colors.transparent)  # 设置笔触颜色为透明
  17.         img = ImageReader(watermark_image)
  18.         if x_position is not None and y_position is not None:
  19.             c.saveState()
  20.             c.translate(x_position, y_position)
  21.             c.rotate(20)
  22.             c.drawImage(img, 0, 0, width=60, height=25)
  23.             c.restoreState()
  24.         else:
  25.             x = (page_width - 60) / 2
  26.             y = (page_height - 25) / 2
  27.             c.saveState()
  28.             c.translate(x, y)
  29.             c.rotate(20)
  30.             c.drawImage(img, 0, 0, width=60, height=25)
  31.             c.restoreState()
  32.     except Exception as e:
  33.         raise ValueError(f"Error drawing image: {e}")
  34.     c.showPage()
  35.     c.save()
  37.     # 将水印 PDF 与原始 PDF 合并
  38.     watermark_pdf = PdfReader(output_buffer)
  39.     for page in input_pdf_obj.pages:
  40.         page.merge_page(watermark_pdf.pages[0])
  41.         output_pdf_obj.add_page(page)
  43.     # 保存输出 PDF
  44.     final_output_buffer = BytesIO()
  45.     output_pdf_obj.write(final_output_buffer)
  46.     binary_data = final_output_buffer.getvalue()
  48.     with open(output_pdf, 'wb') as f:
  49.         output_pdf_obj.write(f)
  50.     return binary_data

  复制代码
  
  
• 调用示例
  
  
  
  • 准备一个PDF文件和一个水印图像文件。
    
  • 调用add_watermark方法，指定输入PDF、输出路径、水印图像路径等参数。
    
  Python复制
  
  1. if __name__ == "__main__":
  2.     from io import BytesIO
  3.     from PyPDF2 import PdfReader, PdfWriter
  4.     from reportlab.pdfgen import canvas
  5.     from reportlab.lib.pagesizes import A4
  6.     from reportlab.lib import colors
  7.     from reportlab.lib.utils import ImageReader
  8.     import fitz
  10.     class WatermarkPDF:
  11.         def repair_pdf(self, input_pdf_binary):
  12.             try:
  13.                 doc = fitz.open(stream=input_pdf_binary, filetype="pdf")
  14.                 repaired_pdf_binary = BytesIO()
  15.                 doc.save(repaired_pdf_binary)
  16.                 doc.close()
  17.                 repaired_pdf_binary.seek(0)
  18.                 return repaired_pdf_binary.read()
  19.             except Exception as e:
  20.                 print(f"Error repairing PDF: {e}")
  21.                 return input_pdf_binary
  23.         def add_watermark(self, input_pdf_binary, output_pdf, watermark_image, x_position=30, y_position=50, opacity=1):
  24.             repaired_pdf_binary = self.repair_pdf(input_pdf_binary)
  25.             input_pdf_obj = PdfReader(BytesIO(repaired_pdf_binary))
  26.             output_pdf_obj = PdfWriter()
  27.             output_buffer = BytesIO()
  29.             page_width, page_height = A4[1], A4[0]
  30.             c = canvas.Canvas(output_buffer, pagesize=(page_width, page_height))
  31.             try:
  32.                 c.setFillColor(colors.white)
  33.                 c.setFillColor(colors.red)
  34.                 c.setFont("Helvetica", 12)
  35.                 c.setFillAlpha(opacity)
  36.                 c.setStrokeColor(colors.transparent)
  37.                 img = ImageReader(watermark_image)
  38.                 if x_position is not None and y_position is not None:
  39.                     c.saveState()
  40.                     c.translate(x_position, y_position)
  41.                     c.rotate(20)
  42.                     c.drawImage(img, 0, 0, width=60, height=25)
  43.                     c.restoreState()
  44.                 else:
  45.                     x = (page_width - 60) / 2
  46.                     y = (page_height - 25) / 2
  47.                     c.saveState()
  48.                     c.translate(x, y)
  49.                     c.rotate(20)
  50.                     c.drawImage(img, 0, 0, width=60, height=25)
  51.                     c.restoreState()
  52.             except Exception as e:
  53.                 raise ValueError(f"Error drawing image: {e}")
  54.             c.showPage()
  55.             c.save()
  57.             watermark_pdf = PdfReader(output_buffer)
  58.             for page in input_pdf_obj.pages:
  59.                 page.merge_page(watermark_pdf.pages[0])
  60.                 output_pdf_obj.add_page(page)
  62.             final_output_buffer = BytesIO()
  63.             output_pdf_obj.write(final_output_buffer)
  64.             binary_data = final_output_buffer.getvalue()
  66.             with open(output_pdf, 'wb') as f:
  67.                 output_pdf_obj.write(f)
  68.             return binary_data
  70.     # 示例调用
  71.     watermark_pdf = WatermarkPDF()
  72.     with open("example.pdf", "rb") as f:
  73.         input_pdf_binary = f.read()
  74.     watermark_image = "watermark.png"
  75.     output_pdf = "output_with_watermark.pdf"
  76.     watermark_pdf.add_watermark(input_pdf_binary, output_pdf, watermark_image)

  复制代码
  
  

实现总结
本代码通过PyMuPDF修复PDF文件，使用reportlab创建水印PDF，并通过PyPDF2将水印合并到原始PDF中。整个过程支持自界说水印的位置、角度和透明度，可以或许灵活地满足差别场景的需求。代码结构清晰，易于扩展和维护，得当在实际项目中使用。
 
让转型不迷航——邹工转型手札
 

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。