generate_landing_page_images: Rewrite in pyvips.

This commit is contained in:
Alex Vandiver 2024-06-12 19:58:17 +00:00 committed by Alex Vandiver
parent 475d4800f9
commit 215c22ec3d
1 changed files with 11 additions and 26 deletions

View File

@ -5,21 +5,20 @@ import glob
import os
import sys
from pathlib import Path
from typing import Tuple
from PIL import Image
ZULIP_PATH = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
if ZULIP_PATH not in sys.path:
sys.path.append(ZULIP_PATH)
import pyvips
LANDING_IMAGES_DIR = os.path.join(ZULIP_PATH, "static", "images", "landing-page", "hello")
ORIGINAL_IMAGES_DIR = os.path.join(LANDING_IMAGES_DIR, "original")
GENERATED_IMAGES_DIR = os.path.join(LANDING_IMAGES_DIR, "generated")
def get_x_size(size: Tuple[float, float], x: int) -> Tuple[int, int]:
return int(x / 3 * size[0]), int(x / 3 * size[1])
def get_x_size(size: int, x: int) -> int:
return int(x / 3.0 * size)
def generate_landing_page_images() -> None:
@ -32,27 +31,13 @@ def generate_landing_page_images() -> None:
for image_file_path in glob.glob(f"{ORIGINAL_IMAGES_DIR}/*"):
file_name = Path(image_file_path).stem
with Image.open(image_file_path) as image:
size_2x = get_x_size(image.size, 2)
size_1x = get_x_size(image.size, 1)
## Generate WEBP images.
image_2x = image.resize(size_2x)
image_2x.save(f"{GENERATED_IMAGES_DIR}/{file_name}-2x.webp", quality=50)
image_1x = image.resize(size_1x)
image_1x.save(f"{GENERATED_IMAGES_DIR}/{file_name}-1x.webp", quality=70)
## Generate JPG images.
# Convert from RGBA to RGB since jpg doesn't support transparency.
rgb_image = image.convert("RGB")
rgb_image_2x = rgb_image.resize(size_2x)
rgb_image_2x.save(
f"{GENERATED_IMAGES_DIR}/{file_name}-2x.jpg", quality=50, optimize=True
)
rgb_image_1x = rgb_image.resize(size_1x)
rgb_image_1x.save(
f"{GENERATED_IMAGES_DIR}/{file_name}-1x.jpg", quality=70, optimize=True
)
image = pyvips.Image.new_from_file(image_file_path)
for size in (1, 2):
scaled_width = get_x_size(image.width, size)
scaled_height = get_x_size(image.height, size)
scaled = image.thumbnail_image(scaled_width, height=scaled_height)
for format in ("webp[Q=50]", "jpg[Q=70,optimize-coding=true]"):
scaled.write_to_file(f"{GENERATED_IMAGES_DIR}/{file_name}-{size}x.{format}")
if __name__ == "__main__":