pdfkit,wkhtmltopdf 添加包含logo和文字的页眉和水印

本文接另一篇：记一次将使用python将html转pdf的辛酸史_contentdocument.documentelement.outerhtml_一心萝卜的博客-CSDN博客

在使用python将html转为pdf时，我采用了pdfkit模块。

现在有个需求,要求生成的pdf带水印和页眉logo。

1.添加页眉：左侧logo，右侧文字

options = {
    '--enable-local-file-access': '--enable-local-file-access',
    '--header-html': 'http://127.0.0.1:8000/admin/admin-add-logo/', # 该url获取到仅包含logo图片的html页面
    '--header-right': '页眉右侧文字',
    '--header-line': '--header-line', # 页眉下划线
    '--header-spacing': 5, # 页眉距离内容高度
}

2.添加水印：

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 = "mark.pdf"
    c = canvas.Canvas(file_name, pagesize=(21 * cm, 29.7 * cm))
    # 移动坐标原点(坐标系左下为(0,0))
    c.translate(5 * cm, 10 * cm)

    # 设置字体
    c.setFont("Helvetica", 80)
    # 指定描边的颜色
    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.1)
    # 设置透明度,1为不透明
    # c.setFillAlpha(0.1)
    # 画几个文本,注意坐标系旋转的影响
    c.drawString(3 * cm, 0 * cm, content)
    c.setFillAlpha(0.6)
    # 关闭并保存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 = '../media/reports/69/report.pdf'
    pdf_file_out = 'watermark.pdf'
    pdf_file_mark = create_watermark('Report')
    add_watermark(pdf_file_in, pdf_file_mark, pdf_file_out)

 欢迎关注wx公众号：python web小栈，共同探讨学习