diff options
| author | VG <vg@devys.org> | 2017-06-18 19:31:26 +0200 |
|---|---|---|
| committer | VG <vg@devys.org> | 2017-06-18 19:31:26 +0200 |
| commit | b981356bc365eb1857696926842f306a6d30b5a2 (patch) | |
| tree | 16fdf7f095a60efd8c6ad1c2076a5fd86e34083c | |
| parent | 40bc663ec6cec90f31565036623d2cbb586c1454 (diff) | |
| download | scripts-b981356bc365eb1857696926842f306a6d30b5a2.tar.gz scripts-b981356bc365eb1857696926842f306a6d30b5a2.tar.bz2 scripts-b981356bc365eb1857696926842f306a6d30b5a2.zip | |
add static gallery generator script
| -rwxr-xr-x | scripts/mkgallery.py | 111 |
1 files changed, 111 insertions, 0 deletions
diff --git a/scripts/mkgallery.py b/scripts/mkgallery.py new file mode 100755 index 0000000..130cbce --- /dev/null +++ b/scripts/mkgallery.py @@ -0,0 +1,111 @@ +#!/usr/bin/env python3 +# Copyright 2017 vg@devys.org +# SPDX-License-Identifier: MIT + + +''' +Make a html gallery of given directory. Supports image and videos. + +Usage: mkgallery.py [options] [--] DIRECTORY + mkgallery.py -h|--help + +Options: + -h, --help Display this help message + -g GALLERY Where to generate gallery +''' + + +import os +import urllib +import math + + +import docopt +import cv2 + + +class Gallery: + 'Gallery generator class' + + def __init__(self, args): + self.gallery_dir = args.get('GALLERY', 'gallery') + self.source_dir = args['DIRECTORY'] + + def generate_thumbnail(self, original_filename): + 'generate a thumbnail of original_filename in a gallery subdir' + #sizes = [(120, 120), (720, 720), (1600, 1600)] + #image = cv2.imread("input.jpg") + #for size in sizes: + # resized_image = cv2.resize(image, size) + # cv2.imwrite("thumbnail_%d.jpg" % size[0], resized_image) + image = cv2.imread(original_filename) + sizes = [(128, 128)] + for size in sizes: + p = os.path + thumbnail_dir = p.join(self.gallery_dir, + p.dirname(original_filename)) + os.makedirs(thumbnail_dir, exist_ok=True) + thumbnail_basename = '%s_%dx%d.jpg' % (p.splitext(p.basename(original_filename))[0], size[0], size[1]) + thumbnail_filename = p.join(thumbnail_dir, thumbnail_basename) + image_height, image_width = image.shape[:2] + aspect_ratio = image_width / image_height + area = size[0] * size[1] + height = math.sqrt(area / aspect_ratio) + width = height * aspect_ratio + aspect_ratio_size = (int(width), int(height)) + print('debug thumbnail size:', aspect_ratio_size) + resized_image = cv2.resize(image, aspect_ratio_size, cv2.INTER_AREA) + cv2.imwrite(thumbnail_filename, resized_image) + return thumbnail_filename + + def generate_entry(self, filename): + 'Generate html code for specific image source filename.' + + thumbnail_filename = self.generate_thumbnail(filename) + + return '''\ +<p><a href="{encoded_filename}"><img src="{thumbnail_filename}"></a></p> +'''.format( + encoded_filename=urllib.parse.quote(os.path.abspath(filename)), + thumbnail_filename=urllib.parse.quote(thumbnail_filename[len(self.gallery_dir):].lstrip('/')), +) + + def content_generator(self): + 'generate <img> entry for each source image' + for dirpath, _, filenames in os.walk(self.source_dir): + for filename in filenames: + yield self.generate_entry(os.path.join(dirpath, filename)) + + def generate_html_file(self, output_stream): + 'generate a single html file' + output_stream.write('''\ +<html> +<head> + <title>Python gallery</title> +</head> +<body> +''') + for content in self.content_generator(): + output_stream.write(content) + output_stream.write('''\ +</body> +</html> +''') + + def mkgallery(self): + 'global call for generating a gallery' + os.makedirs(self.gallery_dir, exist_ok=True) + filename = os.path.join(self.gallery_dir, 'index.html') + with open(filename, 'w', encoding='utf8') as outstream: + self.generate_html_file(outstream) + + +def main(): + 'function called only when script invoked directly on command line' + args = docopt.docopt(__doc__) + gallery = Gallery(args) + gallery.mkgallery() + + +if __name__ == '__main__': + main() |