-- coding: utf-8 --

import fitz
import os
from django.conf import settings
from PIL import Image as pilImage

author = ‘JayChen’

class ProcessPDF:
def init(self, pdf_path, pdf_name, pdf_out_path, pdf_out_name, seal, clarity=1.34):
self.pdf_path = pdf_path
self.pdf_name = pdf_name
self.pdf_out_path = pdf_out_path
self.pdf_out_name = pdf_out_name
self.clarity = clarity
self.pdf = r"D:\2-python\8-河源\90、PDF添加公章\temp\宗地图.pdf"
self.pdf_out = r"D:\2-python\8-河源\90、PDF添加公章\temp\宗地图1.pdf"
self.seal = seal

def merge_img(self):"""图片叠加"""img1 = pilImage.open(self.img_path)  # PDF图片seal_img = pilImage.open(self.seal)  # 公司印章图片layer = pilImage.new('RGBA', img1.size, (0, 0, 0, 0))print(img1.size[0] - 200, img1.size[1] - 250)layer.paste(seal_img, (0, 500))out = pilImage.composite(layer, img1, layer)self.merge_path = os.path.join(self.pdf_path, 'out.png')out.save(self.merge_path)def pdf_to_img(self):"""PDF转化为图片"""doc = fitz.open(self.pdf)for pg in range(doc.pageCount):page = doc[pg]rotate = int(0)zoom_x = self.clarityzoom_y = self.claritytrans = fitz.Matrix(zoom_x, zoom_y).preRotate(rotate)pm = page.getPixmap(matrix=trans, alpha=False)path = r"D:\2-python\8-河源\90、PDF添加公章\1.png"pm.writePNG(path)self.img_path = r"D:\2-python\8-河源\90、PDF添加公章\1.png"def img_to_pdf(self):"""图片转化为PDF"""doc = fitz.open()imgdoc = fitz.open(self.merge_path)pdfbytes = imgdoc.convertToPDF()  # 使用图片创建单页的 PDFimgpdf = fitz.open("pdf", pdfbytes)doc.insertPDF(imgpdf)  # 将当前页插入文档doc.save(self.pdf_out)  # 保存pdf文件doc.close()def out(self):self.pdf_to_img()  # PDF转化为图片self.merge_img()  # 合并图片self.img_to_pdf()  # 图片转化为PDF

def run():
“”"
pdf_path: PDF文件的路径
pdf_name：PDF文件名
pdf_out_path：PDF输出路径
pdf_out_name：PDF输出文件名
seal: 公章路径
clarity：可选参数，可以调整pdf清晰度，默认1.34，数值越大，清晰度越高
“”"
pdf_path = r"D:\2-python\8-河源\90、PDF添加公章\temp"
pdf_name = ‘宗地图.pdf’
pdf_out_path = r"D:\2-python\8-河源\90、PDF添加公章\temp\宗地图1.pdf"
pdf_out_name = ‘out.pdf’
seal = r"D:\2-python\8-河源\90、PDF添加公章\aaa.png"
pdf = ProcessPDF(pdf_path=pdf_path,
pdf_name=pdf_name,
pdf_out_path=pdf_out_path,
pdf_out_name=pdf_out_name,
seal=seal,
clarity=1.34)
pdf.out()