nick/photojawn
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

generate HTML paths

Browse source
This commit is contained in:
Nick Pegg 2024-08-03 10:16:17 -07:00
parent ec633fbd80
commit eb828dc48b
1 changed files with 23 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
photoalbum/generate.py
View file

@ -2,7 +2,7 @@ import logging
from pathlib import Path from pathlib import Path
from PIL import Image, UnidentifiedImageError from PIL import Image, UnidentifiedImageError
from rich.progress import track from rich.progress import track, Progress
from photoalbum.config import Config from photoalbum.config import Config
@ -15,7 +15,7 @@ def generate(config: Config, album_path: Path) -> None:
""" """
images = find_images(album_path) images = find_images(album_path)
generate_thumbnails(config, images) generate_thumbnails(config, images)
generate_html(config, album_path) generate_html(config, album_path, images)
def find_images(path: Path) -> list[Path]: def find_images(path: Path) -> list[Path]:
@ -60,16 +60,34 @@ def generate_thumbnails(config: Config, images: list[Path]) -> None:
thumb_img.thumbnail(config.thumbnail_size) thumb_img.thumbnail(config.thumbnail_size)
thumb_filename = image_path.stem + ".thumb" + image_path.suffix thumb_filename = image_path.stem + ".thumb" + image_path.suffix
thumb_img.save(slides_path / thumb_filename) thumb_img.save(slides_path / thumb_filename)
logger.info(f"Generated thumbnail size {image_path} -> {thumb_filename}") logger.info(f"Generated thumbnail size \"{image_path}\" -> \"{thumb_filename}\"")
screen_img = orig_img.copy() screen_img = orig_img.copy()
screen_img.thumbnail(config.view_size) screen_img.thumbnail(config.view_size)
screen_filename = image_path.stem + ".screen" + image_path.suffix screen_filename = image_path.stem + ".screen" + image_path.suffix
screen_img.save(slides_path / screen_filename) screen_img.save(slides_path / screen_filename)
logger.info(f"Generated screen size {image_path} -> {screen_filename}") logger.info(f"Generated screen size \"{image_path}\" -> \"{screen_filename}\"")
def generate_html(config: Config, path: Path) -> None: def generate_html(config: Config, root_path: Path, images: list[Path]) -> None:
""" """
Recursively generate HTML files for this directory and all children Recursively generate HTML files for this directory and all children
""" """
# Find all the album dirs that either have photos, or have child album dirs
# The paths in this set are all relative to the root_path
album_paths = {Path('.')}
for image_path in images:
rel_path = image_path.relative_to(root_path)
for parent in rel_path.parents:
album_paths.add(parent)
# Generate album pages for all album dirs
for album_path in track(album_paths, description="Generating album HTML..."):
html_path = album_path / "index.html"
logger.debug(html_path)
# Generate photo pages for all photos
for image_path in track(images, description="Generating photo HTML..."):
html_path = image_path.parent / "slides" / image_path.with_suffix(".html").name
logger.debug(html_path)