对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文件