博客
关于我
python 图片添加水印 pdf 添加水印
阅读量:503 次
发布时间:2019-03-07

本文共 2691 字,大约阅读时间需要 8 分钟。

Python添加PDF水印,使用pyPdf和ReportLab库实现

本示例将介绍如何为PDF文件添加文字水印的实现方法。我们将使用两个第三方库:pyPdf和ReportLab。以下是实现代码的详细说明。

环境准备

  • Python 2.7
  • pyPdf库(获取地址:http://pybrary.net/pyPdf)
  • ReportLab库(获取地址:http://www.reportlab.com)
  • PIL库(Python Imaging Library,安装命令:pip install PIL
  • 代码解释

    1. 导入所需库

    from pyPdf import PdfFileWriter, PdfFileReader
    from reportlab.pdfgen import canvas
    from PIL import Image, ImageDraw, ImageFont
    import os

    2. 定义常用变量

    # 指定字体和字号
    font = ImageFont.truetype('/usr/share/fonts/truetype/dejavu/DejaVuSerif.ttf', 45)

    3. 添加文字水印到图片的函数

    def add_text_to_image(image_name, text, font=font):
    # 打开目标图片
    image = Image.open(image_name)
    # 转换为RGBA格式
    rgba_image = image.convert('RGBA')
    # 创建透明的文本覆盖层
    text_overlay = Image.new('RGBA', rgba_image.size, (255, 255, 255, 0))
    # 创建绘图对象
    image_draw = ImageDraw.Draw(text_overlay)
    # 计算文本大小
    text_size_x, text_size_y = image_draw.textsize(text, font=font)
    # 确定文本位置(这里采用底部居中)
    text_xy = ((rgba_image.size[0] - text_size_x) // 2, (rgba_image.size[1] - text_size_y) // 2)
    # 绘制文本
    image_draw.text(text_xy, text, font=font, fill=(0, 0, 0))
    # 合成图片
    image_with_text = Image.alpha_composite(rgba_image, text_overlay)
    return image_with_text

    4. 为PDF文件添加文字水印的函数

    def add_mark(info_name, text):
    # 创建水印PDF文件
    c = canvas.Canvas("watermark.pdf")
    c.setFont("Courier", 35)
    c.setFillGray(0.0, 1) # 设置灰度
    c.saveState() # 保存绘图状态
    c.translate(280, 750) # 平移坐标
    c.rotate(0) # 旋转角度
    c.drawCentredString(0, 0, text) # 中心绘制文本
    c.restoreState() # 恢复绘图状态
    c.save() # 保存PDF文件

    5. 主函数实现

    def run():
    # 创建结果目录
    os.mkdir("results")
    # 获取当前目录下的文件列表
    file_name = "info"
    files = os.listdir(file_name)
    r = 0 # 初始化计数器
    for t_file in files:
    file_all_img = os.path.join(file_name, t_file)
    if os.path.isfile(file_all_img):
    continue
    # 遍历文件夹中的所有图片文件
    for file_img in os.listdir(file_all_img):
    file_img_all_img_all = os.path.join(file_all_img, file_img)
    if file_img_all_img_all.endswith('pdf'):
    # 为PDF文件添加水印
    output = add_mark(file_img_all_img_all, t_file)
    # 保存结果
    with open(os.path.join('results', f'{r}.pdf'), 'wb') as outStream:
    output.write(outStream)
    # 删除临时水印文件
    os.remove("watermark.pdf")
    else:
    # 为图片文件添加文字水印
    im_after = add_text_to_image(file_img_all_img_all, t_file)
    im_after.save(os.path.join('results', f'{r}.png'))
    r += 1

    使用说明

  • 将代码保存为.py文件,并确保所有依赖库已安装。
  • 运行代码时,传递要处理的文件路径作为输入。
  • 生成的结果文件将存放在results目录下。
  • 通过以上方法,您可以轻松为PDF文件和图片文件添加文字水印,实现文件的版权保护。

    转载地址:http://ljmcz.baihongyu.com/

    你可能感兴趣的文章
    mysql中like % %模糊查询
    查看>>
    MySql中mvcc学习记录
    查看>>
    mysql中null和空字符串的区别与问题!
    查看>>
    MySQL中ON DUPLICATE KEY UPDATE的介绍与使用、批量更新、存在即更新不存在则插入
    查看>>
    MYSQL中TINYINT的取值范围
    查看>>
    MySQL中UPDATE语句的神奇技巧,让你操作数据库如虎添翼!
    查看>>
    Mysql中varchar类型数字排序不对踩坑记录
    查看>>
    MySQL中一条SQL语句到底是如何执行的呢?
    查看>>
    MySQL中你必须知道的10件事,1.5万字!
    查看>>
    MySQL中使用IN()查询到底走不走索引?
    查看>>
    Mysql中使用存储过程插入decimal和时间数据递增的模拟数据
    查看>>
    MySql中关于geometry类型的数据_空的时候如何插入处理_需用null_空字符串插入会报错_Cannot get geometry object from dat---MySql工作笔记003
    查看>>
    mysql中出现Incorrect DECIMAL value: '0' for column '' at row -1错误解决方案
    查看>>
    mysql中出现Unit mysql.service could not be found 的解决方法
    查看>>
    mysql中出现update-alternatives: 错误: 候选项路径 /etc/mysql/mysql.cnf 不存在 dpkg: 处理软件包 mysql-server-8.0的解决方法(全)
    查看>>
    Mysql中各类锁的机制图文详细解析(全)
    查看>>