import os

import fitz


def add_watermask():
    file_path = "./pdfs/girl.pdf"
    water_path = "./waters/logo.png"
    baseaname = os.path.basename(file_path)
    name, suffix = os.path.splitext(baseaname)
    out_path = os.path.join(os.path.dirname(file_path), name + "_watered" + suffix)
    pdf_file = fitz.open(file_path)
    page = pdf_file[0]
    print(page.bound())
    # page.insert_image(page.bound(), filename=water_path)
    rect = (800, -100, 900, 0)
    page.insert_image(rect, filename=water_path)
    pdf_file.save(out_path)


add_watermask()