Fix unnecessary pillow import by using image size from CDN endpoint

geotribu_cli/content/header_check.py

@@ -1,19 +1,23 @@
 import argparse
 import logging
 import os
+from functools import lru_cache
 from pathlib import Path
 
 import frontmatter
+import requests
+from requests import Response
 
 from geotribu_cli.constants import GeotribuDefaults
 from geotribu_cli.json.json_client import JsonFeedClient
-from geotribu_cli.utils.check_image_size import get_image_dimensions_by_url
 from geotribu_cli.utils.check_path import check_path
 from geotribu_cli.utils.slugger import sluggy
 
 logger = logging.getLogger(__name__)
 defaults_settings = GeotribuDefaults()
 
+URL_CDN = "https://cdn.geotribu.fr"
+
 MANDATORY_KEYS = [
     "title",
     "authors",
@@ -110,10 +114,21 @@ def check_author_md(author: str, folder: Path) -> bool:
     return os.path.exists(p)
 
 
+@lru_cache(maxsize=512)
+def download_image_sizes() -> dict:
+    # download images sizes and indexes
+    image_req: Response = requests.get(f"{URL_CDN}/img/search-index.json")
+    image_req.raise_for_status()
+    return image_req.json()["images"]
+
+
 def check_image_size(
-    image_url: str, minw: int, maxw: int, minh: int, maxh: int
+    image_url: str, images: dict, minw: int, maxw: int, minh: int, maxh: int
 ) -> bool:
-    width, height = get_image_dimensions_by_url(image_url)
+    key = image_url.replace(f"{URL_CDN}/img/", "")
+    if key not in images:
+        return False
+    width, height = images[key]
     return minw <= width <= maxw and minh <= height <= maxh
 
 
@@ -179,6 +194,7 @@ def run(args: argparse.Namespace) -> None:
                     logger.error("Pas d'URL pour l'image")
                 elif not check_image_size(
                     yaml_meta["image"],
+                    download_image_sizes(),
                     args.min_image_width,
                     args.max_image_width,
                     args.min_image_height,

tests/test_utils_images_size.py

@@ -126,8 +126,16 @@ def test_check_image_dimensions(self):
 
     def test_image_url_dimensions(self):
         for url, width, height in [
-            ("https://cdn.geotribu.fr/img/coup_de_gueule.jpg", 74, 64),
-            ("https://cdn.geotribu.fr/img/pytroll.png", 100, 100),
+            (
+                "https://cdn.geotribu.fr/img/articles-blog-rdp/articles/2024/lidarhd_pdal/lidar_zoom.png",
+                1380,
+                763,
+            ),
+            (
+                "https://cdn.geotribu.fr/img/articles-blog-rdp/articles/2024/lidarhd_pdal/sol.png",
+                1000,
+                557,
+            ),
         ]:
             w, h = get_image_dimensions_by_url(url)
             self.assertEqual(w, width)