simple-s3-gallery/GenThumb.py

#!/usr/bin/env python3
# GenThumb.py - Part of the simple s3 gallery
# Usage: ./GenThumb.py

from pathlib import Path
from PIL import Image
import logging
import _thread
import time

logging.basicConfig(filename='GenThumb.log', level=logging.DEBUG,
                    format='%(asctime)s [%(levelname)s] %(message)s',
                    datefmt='%m/%d/%Y %I:%M:%S %p')
logging.info("---------------")

baseuri = "https://s3.wasabisys.com/c0de-photography/"
s3_path = "/mnt/photos/"
browse_path = "EOS 30D/MFF 2016"
thumb_path = "./thumbs"
thumbsize = (250, 250)

logging.info("Getting Paths (Can take time on large directories)")
filelist = list(Path(s3_path + browse_path).rglob("*.[jJ][pP][gG]"))
thumblist = list(Path(thumb_path).rglob("*.[jJ][pP][gG]"))

def gen_thumb(image, imagepath):
    Path(thumb_path + '/' + imagepath).mkdir(mode=0o755, parents=True, exist_ok=True) # Generate the path if it doesn't already exist
    thumbname = "%s/%s/%s" % (thumb_path, imagepath, image.name)

    if image.name in "%s" % thumblist:
        logging.debug("Skipping %s" % image)
    else:
        logging.info("Generating thumbnail for New Image: %s" % (image))

        img = Image.open(image)
        img.thumbnail(thumbsize)
        img.save(thumbname)

        logging.info("Saved Thumbnail as %s" % thumbname)

with open('pathlist_%s.txt' % browse_path.replace('/', ':'), 'w') as pathlist:
    for image in filelist:
        imagepath = "%s" % image.parent
        imagepath = imagepath.strip(s3_path)
        # The pathlist file is used by GenTemplate.py to generate a specific gallery for a specific directory and its children
        pathlist.write(baseuri + imagepath + '/' + image.name + '\n')
        try:
            # WARNING: Each file found will become a thread! Be sure your computer can handle the load
            _thread.start_new_thread(gen_thumb, (image, imagepath,))
            time.sleep(0.15)
        except:
            logging.error("unable to start thread")

logging.info("Wait for all threads to save")

while(1):
    pass