media-larlet-fr/boop.py

#!/usr/bin/env python3

import os
import shutil
import re

from operator import itemgetter

from jinja2 import Environment, PackageLoader, select_autoescape
from PIL import Image, ExifTags

from configuration import SITE_TITLE, THEME


PICTURES_DIR_NAME = "photos"
OUTPUT_DIR_NAME = "output"


def list_galleries_in(path):
    gallery_dirs = [f for f in os.scandir(path) if f.is_dir()]
    for gallery_dir in gallery_dirs:
        photos = list(list_photos_in(gallery_dir))
        if len(photos) == 0:
            continue

        photos.sort(key=itemgetter("name"))
        cover_index = (len(gallery_dir.name) + 42) % len(photos)
        gallery = {
            "name": gallery_dir.name,
            "path": gallery_dir.path,
            "url": gallery_dir.name,
            "num_photos": len(photos),
            "photos": photos,
            "cover_photo": photos[cover_index],
        }
        yield gallery


def list_photos_in(gallery_dir):
    photo_files = [
        f for f in os.scandir(gallery_dir) if re.match(".+\.jpg", f.name, re.I)
    ]
    for photo_file in photo_files:
        url = os.path.join(gallery_dir.name, photo_file.name)
        thumb_url = os.path.join(gallery_dir.name, f"thumb_{photo_file.name}")
        photo = {
            "name": photo_file.name,
            "path": photo_file.path,
            "url": url,
            "thumb_url": thumb_url,
        }
        yield photo


def generate_output_dir():
    output_path = os.path.join(os.curdir, OUTPUT_DIR_NAME)
    if not os.path.isdir(output_path):
        os.mkdir(output_path)
    return output_path


def generate_style(output_path):
    style_path = os.path.join(os.curdir, THEME, "style")
    style_output_path = os.path.join(output_path, "style")
    if os.path.isdir(style_output_path):
        shutil.rmtree(style_output_path)
    shutil.copytree(style_path, style_output_path)


def generate_index(output_path, galleries):
    index_path = os.path.join(output_path, "index.html")

    theme_path = os.path.join(os.curdir, THEME)
    jinja_env = Environment(
        loader=PackageLoader("boop", theme_path), autoescape=select_autoescape(["html"])
    )
    index_template = jinja_env.get_template("index.html.j2")

    with open(index_path, "w") as index_file:
        index_file.write(
            index_template.render(galleries=galleries, site_title=SITE_TITLE)
        )


def generate_gallery(output_path, gallery):
    generate_gallery_index(output_path, gallery)
    generate_gallery_dir(output_path, gallery)


def generate_gallery_index(output_path, gallery):
    gallery_index_path = os.path.join(output_path, f"{gallery['name']}.html")

    theme_path = os.path.join(os.curdir, THEME)
    jinja_env = Environment(
        loader=PackageLoader("boop", theme_path), autoescape=select_autoescape(["html"])
    )
    gallery_template = jinja_env.get_template("gallery.html.j2")

    with open(gallery_index_path, "w") as gallery_file:
        gallery_file.write(
            gallery_template.render(gallery=gallery, site_title=SITE_TITLE)
        )


def generate_gallery_dir(output_path, gallery):
    gallery_output_path = os.path.join(output_path, gallery["name"])
    if not os.path.isdir(gallery_output_path):
        os.mkdir(gallery_output_path)

    for photo in gallery["photos"]:
        photo_output_path = os.path.join(output_path, photo["url"])
        shutil.copyfile(photo["path"], photo_output_path)

        thumb_output_path = os.path.join(output_path, photo["thumb_url"])
        generate_thumb_file(thumb_output_path, photo)


def generate_thumb_file(output_path, photo):
    orientation_key = get_orientation_exif_key()
    size = (440, 264)

    with Image.open(photo["path"]) as image:
        # First, make sure image is correctly oriented
        exif = image._getexif()
        if exif[orientation_key] == 3:
            image = image.rotate(180, expand=True)
        elif exif[orientation_key] == 6:
            image = image.rotate(270, expand=True)
        elif exif[orientation_key] == 8:
            image = image.rotate(90, expand=True)

        w, h = image.size
        if w > size[0] and h > size[1]:
            # If the original file is larger in width AND height, we resize
            # first the image to the lowest size accepted (both width and
            # height stays greater or equal to requested size).
            # E.g. 1200x900 is resized to 440x330
            #      1200x600 is resized to 528x264
            if size[0] / size[1] <= w / h:
                w = int(max(size[1] * w / h, 1))
                h = 264
            else:
                h = int(max(size[0] * h / w, 1))
                w = 440
            new_size = (w, h)

            image.draft(None, new_size)
            image = image.resize(new_size, Image.BICUBIC)

        # We now have an image with at least w = 440 OR h = 264 (unless one of
        # the size is smaller). But the image can still be larger than
        # requested size, so we have to crop the image in the middle.
        crop_box = None
        if w > size[0]:
            left = (w - size[0]) / 2
            right = left + size[0]
            crop_box = (left, 0, right, h)
        elif h > size[1]:
            upper = (h - size[1]) / 2
            lower = upper + size[1]
            crop_box = (0, upper, w, lower)
        if crop_box is not None:
            image = image.crop(crop_box)

        # And we save the final image.
        image.save(output_path)


def get_orientation_exif_key():
    for (key, tag) in ExifTags.TAGS.items():
        if tag == "Orientation":
            return key


def main():
    pictures_folder = os.path.join(os.curdir, PICTURES_DIR_NAME)
    galleries = list(list_galleries_in(pictures_folder))
    if len(galleries) == 0:
        return

    galleries.sort(key=itemgetter("name"))
    output_path = generate_output_dir()
    generate_style(output_path)
    generate_index(output_path, galleries)
    for gallery in galleries:
        generate_gallery(output_path, gallery)


if __name__ == "__main__":
    main()