Как использовать PyPDF2 для вставки файла изображения * .png в файл .pdf

Question

вот мой код:

input_file = "example.pdf"
output_file = "example-drafted.pdf"
watermark_file = "draft.pdf"

with open(input_file, "rb") as filehandle_input:
    pdf = PyPDF2.PdfFileReader(filehandle_input)

    with open(watermark_file, "rb") as filehandle_watermark:
         watermark = PyPDF2.PdfFileReader(filehandle_watermark)
         first_page = pdf.getPage(0)
         first_page_watermark = watermark.getPage(0)
         first_page.mergePage(first_page_watermark)
         pdf_writer = PyPDF2.PdfFileWriter()       
         pdf_writer.addPage(first_page)

         with open(output_file, "wb") as filehandle_output:
              pdf_writer.write(filehandle_output)

в моем коде выше: watermark_file = "draft.pdf".

я хочу, чтобы этот watermark_file использовался как "draft.png"

Answer 1

Я не знаю, хорошо ли я понял вашу проблему, но у вас, похоже, проблемы с добавлением изображения PNG в PDF?

Возможно, вы решите использовать PDFKit вместо PyPDF2, что обеспечиваетболее простой способ работы с PDF-файлами с личной точки зрения.

Например, вы можете создать страницу HTML, содержащую ваш draft.png, а затем добавить ее в файл PDF.

Я нашелхороший пример для этого метода в этом блоге, если вы заинтересованы:

http://yoroshikune.com/creating-pdf-files-python/