Python实现给单个PDF文件添加水印的方法.rar

Python实现给单个PDF文件添加水印的方法 from PyPDF2 import PdfFileReader, PdfFileWriter from reportlab.lib.units import cm from reportlab.pdfgen import canvas def create_watermark(content): """水印信息""" # 默认大小为21cm*29.7cm file_name = "C:\\Users\\Administrator\\Desktop\\bbb\\bbb.pdf" #最后的带水印的文件保存位置，要修改 c = canvas.Canvas(file_name, pagesize=(30 * cm, 30 * cm)) # 移动坐标原点(坐标系左下为(0,0)) c.translate(10 * cm, 5 * cm) # 设置字体 c.setFont("Helvetica", 30) # 指定描边的颜色 c.setStrokeColorRGB(0, 1, 0) # 指定填充颜色 c.setFillColorRGB(0, 1, 0) # 旋转45度,坐标系被旋转 c.rotate(30) # 指定填充颜色 c.setFillColorRGB(0, 0, 0, 0.1) # 设置透明度,1为不透明 # c.setFillAlpha(0.1) # 画几个文本,注意坐标系旋转的影响 for i in range(5): for j in range(10): a = 10 * (i - 1) b = 5 * (j - 2) c.drawString(a * cm, b * cm, content) c.setFillAlpha(0.1) # 关闭并保存pdf文件 c.save() return file_name def add_watermark(pdf_file_in, pdf_file_mark, pdf_file_out): """把水印添加到pdf中""" pdf_output = PdfFileWriter() input_stream = open(pdf_file_in, 'rb') pdf_input = PdfFileReader(input_stream, strict=False) # 获取PDF文件的页数 pageNum = pdf_input.getNumPages() # 读入水印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) pdf_output.write(open(pdf_file_out, 'wb')) if __name__ == '__main__': pdf_file_in = 'C:\\Users\\Administrator\\Desktop\\aaa\\aaa.pdf' #要加水印的pdf文件，要修改 pdf_file_out = 'C:\\Users\\Administrator\\Desktop\\bbb\\bbb.pdf' #最后的带水印的文件保存位置，要修改 pdf_file_mark = create_watermark('pythonbug.cn') add_watermark(pdf_file_in, pdf_file_mark, pdf_file_out)