PDF添加水印

给一个文件夹内的所有pdf添加水印

from reportlab.pdfgen import canvas  # 提供pdf文件
from reportlab.pdfbase import pdfmetrics  # 注册字体
from reportlab.pdfbase.ttfonts import TTFont  # 提供字体对象
from PyPDF2 import PdfFileReader, PdfFileWriter


def water_write(water_path, path, file, num=-1):
    """
    pdf 加水印
    :param water_path:水印文件的路径
    :param path:需要加水印文件的路径
    :param file:加水印后保存的位置
    :param num: 前几页加水印 默认全部页都加水印
    :return:
    """
    water_reader = PdfFileReader(open(water_path, mode='rb'))
    file_reader = PdfFileReader(open(path, mode='rb'))
    water_page = water_reader.getPage(0)
    page_count = file_reader.numPages
    page_num = page_count if num < 0 or num > page_count else num
    writer = PdfFileWriter()
    for i in range(page_num