quickwatermark.py

'''
Usage:
python quickwatermark.py [path to folder that contains images to waternark]

This will go through all the files in that folder, try to open them and add 
the filename as text on top of the image. The watermarked images will be stored
in a subfolder of the chosen folder called "watermarked"

'''


import os
from PIL import Image, ImageDraw
import argparse

parser = argparse.ArgumentParser()
parser.add_argument('folder')
args = parser.parse_args()

folder = args.folder
if os.path.exists(folder):
    if not folder[-1]=="/":
        folder += "/"

    os.makedirs(folder+"watermarked",exist_ok=True)

    for filename in os.listdir(folder):
        if not os.path.isdir(folder+filename):
            try:
                im = Image.open(folder+filename)
                draw = ImageDraw.Draw(im)
                font = draw.getfont()
                size = font.getsize(filename)
            
                draw.text((int((im.size[0]-size[0])/2),int((im.size[1]-size[1])/2)), filename, font=font, fill=(255,255,255,128))
                im.save(folder+"watermarked/"+filename)
            except Exception as e:
                print(e)