73 lines
2.4 KiB
Python
Executable File
73 lines
2.4 KiB
Python
Executable File
#!/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
|
|
import getopt
|
|
import sys
|
|
import os
|
|
|
|
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 = os.environ.get('BASEURI',"https://s3.wasabisys.com/c0de-photography/")
|
|
s3_path = os.environ.get('S3MOUNT', "/mnt/photos/")
|
|
thumb_path = os.environ.get('THUMBNAILS', "./thumbs")
|
|
thumbsize = (250, 250)
|
|
|
|
try:
|
|
opts, args = getopt.getopt(sys.argv[1:],"hb:",["browse="])
|
|
except getopt.GetoptError:
|
|
print ('GenThumb.py -b <browse path>')
|
|
sys.exit(2)
|
|
for opt, arg in opts:
|
|
if opt == '-h':
|
|
print ('GenThumb.py -b <browse path>')
|
|
sys.exit()
|
|
elif opt in ("-b", "--browse"):
|
|
browse_path = arg
|
|
|
|
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
|
|
|