Python 处理PDF加水印

from PyPDF2 import PdfFileReader, PdfFileWriter
from reportlab.lib.units import cm
from reportlab.pdfgen import canvas
from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont
from reportlab.lib.styles import getSampleStyleSheet
import sys
import importlib
import redis
import json
import io

def create_watermark(file_name,x,y):
    """水印信息"""
    #模板保存名称
    file_name = file_name   
    # 加载字体
    pdfmetrics.registerFont(TTFont("SimSun", "SimSun.ttf"))
    # 创建画布，x,y为宽高
    c = canvas.Canvas(file_name, pagesize=(x, y))
    # 移动坐标原点(坐标系左下为(0,0))
    # c.translate(10*cm, 5*cm)
    # 设置字号
    c1=int(y/20)
    # 设置字体
    c.setFont("SimSun", c1)
    s="本标准仅供内部使用，不得翻印！"
    # 指定描边的颜色
    c.setStrokeColorRGB(0, 1, 0)
    # 指定填充颜色
    c.setFillColorRGB(0, 1, 0)
    # 画一个矩形
    # c.rect(cm, cm, 7*cm, 17*cm, fill=1)
    # 旋转45度,坐标系被旋转
    c.rotate(30)
    # 指定填充颜色 第四个参数 影响颜色深浅
    c.setFillColorRGB(0, 0, 0, 0.17)
    # 设置透明度,1为不透明
    # c.setFillAlpha(0.1)
    # 画几个文本,注意坐标系旋转的影响
    # c.drawString(3*cm, 0*cm, content)
    # print(int(y)*0.3)
    # sys.exit(0) 
    c.drawString(int(x/3),int(y/3),s)
    c.setFillAlpha(0.6)
    # 关闭并保存pdf文件
    c.save()
    return file_name


def add_watermark(pdf_file_in, pdf_mark_url, pdf_file_out):
    # """把水印添加到pdf中"""
    pdf_output = PdfFileWriter()
    input_stream = open(pdf_file_in, 'rb')
    # False
    pdf_input = PdfFileReader(input_stream, strict=True)
    # 获取PDF文件的页数
    pageNum = pdf_input.getNumPages()
    # 获取PDF尺寸
    x=pdf_input.getPage(0).mediaBox.getUpperRight_x()
    y=pdf_input.getPage(0).mediaBox.getUpperRight_y()
    # 生成水印模板
    pdf_file_mark=create_watermark(pdf_mark_url,x,y)
    # 读入水印pdf文件
    pdf_watermark = PdfFileReader(open(pdf_file_mark, 'rb'), strict=False)
    # 给每一页打水印
    for i in range(pageNum):
        page = pdf_input.getPage(i)
        page.mergePage(pdf_watermark.getPage(0))
        page.compressContentStreams()  # 压缩内容
        pdf_output.addPage(page)
    outputStream = open(pdf_file_out, "wb",errors=None)
    pdf_output.write(outputStream)
   

if __name__ == '__main__':
    importlib.reload(sys)
#初始化运行环境为UTF-8,否则姓名中文输出报错
    key = sys.argv[1]
    # key='1a81f53e1e3bf8b3443fb3eb0370c929'
    # print(key)
    re = redis.Redis(host='127.0.0.1', port=6379,db=0)
    info=re.get(key).decode(encoding='UTF-8',errors='strict')
    data=json.loads(info)
    # data={'file':"C:/Users/34008/Desktop/2.pdf",'suiyin':"C:/Users/34008/Desktop/33434342sy.pdf",'url':"C:/Users/34008/Desktop/123445.pdf"}
    # data={'file':"G:/wwwroot/127.0.0.1/public/标准文件/国内标准/国家标准/废止/GB/60000/GB 50405-2007/GB 50405-2007.pdf",'suiyin':"G:/wwwroot/127.0.0.1/public/uploads/viewpdf/83311980bed3350a52158c59ea270172sy.pdf",'url':"G:/wwwroot/127.0.0.1/public/uploads/viewpdf/83311980bed3350a52158c59ea270172.pdf"}
    add_watermark(data['file'],data['suiyin'], data['url'])