Python实现对PDF批量添加水印

对pdf文件添加水印

下面附上详细代码和代码解释

# 导入库
from PyPDF2 import PdfFileWriter, PdfFileReader
import os
# 将水印页面读出来
water_page = PdfFileReader('files/jc_water.pdf').getPage(0)

# 遍历文件夹里的文件
all_files = os.listdir('files/inputs')   # 返回在files/inputs下的所有文件
for file_name in all_files:             # 遍历all_files的所有文件
   pash = 'files/inputs/'+file_name          # 通过file_name和files/inputs/ 拼接出文件完整路径
   # print(pash)
   reader = PdfFileReader(pash)           # reader保存读出需要添加水印的页面 PdfFileReader()
   number = reader.getNumPages()        # 获取pdf文件的页数
   writer = PdfFileWriter()         # 创建writer对象
   for x in range(number):             # for in  遍历每个页面
      page = reader.getPage(x)      # page保存读出的每个页面
      page.mergePage(water_page)    # 对读出的每个页面粘贴水印
   writer.write(open('files/outputs/'+file_name,'wb'))     # 保存批量添加水印的pdf文件