用python写一段代码,实现对多个pdf打水印的工作,每一个pdf的水印都不一样
时间: 2024-02-15 10:02:56 浏览: 74
python给pdf添加水印
好的,以下是一个使用 PyPDF2 库实现对多个 PDF 文件打不同水印的示例代码:
```python
import PyPDF2
# 定义水印内容和字体大小
watermark_text = 'Confidential'
font_size = 30
# 定义要加水印的 PDF 文件列表和对应的水印内容
pdf_files = {'file1.pdf': 'Watermark 1',
'file2.pdf': 'Watermark 2',
'file3.pdf': 'Watermark 3'}
# 遍历 PDF 文件列表
for file_name, watermark_text in pdf_files.items():
# 打开 PDF 文件
pdf_file = open(file_name, 'rb')
pdf_reader = PyPDF2.PdfFileReader(pdf_file)
pdf_writer = PyPDF2.PdfFileWriter()
# 加载水印字体
watermark_font = PyPDF2.pdf.encrypt.StandardEncryption.load_pubkey('Helvetica')
# 遍历 PDF 文件中的每一页
for page_num in range(pdf_reader.getNumPages()):
# 获取当前页
page = pdf_reader.getPage(page_num)
# 创建水印对象
watermark = PyPDF2.pdf.ContentStream(
[PyPDF2.pdf.TextObject(watermark_text, fontName='Helvetica', fontSize=font_size).getObject()],
pdf_writer)
# 获取当前页的大小
page_width = page.mediaBox.getWidth()
page_height = page.mediaBox.getHeight()
# 将水印添加到当前页的中心位置
x = (page_width - watermark.bbox[2]) / 2
y = (page_height - watermark.bbox[3]) / 2
watermark.moveTo(x, y)
watermark_text = watermark
# 将水印添加到当前页
page.mergePage(watermark_text)
# 将当前页添加到 PDF 输出对象中
pdf_writer.addPage(page)
# 保存添加水印后的 PDF 文件
output_file_name = 'watermarked_' + file_name
output_file = open(output_file_name, 'wb')
pdf_writer.write(output_file)
# 关闭文件
pdf_file.close()
output_file.close()
```
此代码将会依次遍历指定的多个PDF文件,根据对应的水印内容在每一页的中心位置添加水印,并将添加水印后的结果保存为新的 PDF 文件。
阅读全文