pdf_watermark.py 1.1 KB

1234567891011121314151617181920212223242526272829303132333435
  1. import PyPDF2
  2. import os
  3. base_dir = os.path.dirname(__file__)
  4. input_file = base_dir + "/data/Rechnung Miete_Ulmen.pdf"
  5. output_file = base_dir + "/data/example-gc.pdf"
  6. watermark_file = base_dir + "/data/Briefbogen_GC_Jahnstr_GmbH.pdf"
  7. with open(input_file, "rb") as filehandle_input:
  8. # read content of the original file
  9. pdf = PyPDF2.PdfFileReader(filehandle_input)
  10. with open(watermark_file, "rb") as filehandle_watermark:
  11. # read content of the watermark
  12. watermark = PyPDF2.PdfFileReader(filehandle_watermark)
  13. # get first page of the original PDF
  14. first_page = pdf.getPage(0)
  15. # get first page of the watermark PDF
  16. first_page_watermark = watermark.getPage(0)
  17. # merge the two pages
  18. first_page.mergePage(first_page_watermark)
  19. # create a pdf writer object for the output file
  20. pdf_writer = PyPDF2.PdfFileWriter()
  21. # add page
  22. pdf_writer.addPage(first_page)
  23. with open(output_file, "wb") as filehandle_output:
  24. # write the watermarked file to the new file
  25. pdf_writer.write(filehandle_output)