Loading ...

Applicare un watermark su PDF in Python

Applicare un watermark su PDF in Python

Ho scoperto per Python la libreria PyPDF2, che ci permette di fare praticamente qualsiasi operazionei sui PDF.

Oggi vediamo come applicare un watermark.

Prima di tutto installiamola:

# pip3 install PyPDF2

Il watermark da applicare deve essere sempre un PDF (o almeno io non sono riuscito ad applicare una immagine direttamente).

Quindi, al momento, ho preso una immagine qualsiasi e l'ho convertita in PDF.

Detto ciò, vediamo il codice:

from PyPDF2 import PdfFileReader, PdfFileWriter

def mettiWatermark(input, output, watermark):
    wm = PdfFileReader(watermark)
    wmPage = wm.getPage(0)

    pdfReader = PdfFileReader(input)
    pdfWriter = PdfFileWriter()

    for page in range(pdfReader.getNumPages()):
        p = pdfReader.getPage(page)
        p.mergePage(wmPage)
        pdfWriter.addPage(p)

    with open(output, 'wb') as fh:
        pdfWriter.write(fh)

input = '/home/matte-server/Scrivania/test.pdf'
output = '/home/matte-server/Scrivania/watermarked.pdf'
watermark = '/home/matte-server/Scrivania/img.pdf'

mettiWatermark(input, output, watermark)

Enjoy!