work

save startup blend for animation tab & whatnot

extensions/user_default/blenderkit/search_price.py

@@ -1,7 +1,40 @@
+"""Search price lookup helper."""
+
+from __future__ import annotations
+
+import time
+import logging
 from typing import Iterable, List, Optional, Tuple
 
 from . import client_lib, paths, utils
 
+bk_logger = logging.getLogger(__name__)
+
+# Simple in-memory cache to avoid repeated price lookups per version UUID.
+# TTL keeps cache fresh-ish while drastically reducing chatter on paged searches.
+_PRICE_CACHE: dict[str, tuple[float, dict]] = {}
+_PRICE_CACHE_TTL_SEC = 300.0  # 5 minutes is plenty for session-level reuse
+
+
+def _cache_get(uuid: str) -> Optional[dict]:
+    entry = _PRICE_CACHE.get(uuid)
+    if not entry:
+        return None
+    ts, payload = entry
+    if time.perf_counter() - ts > _PRICE_CACHE_TTL_SEC:
+        _PRICE_CACHE.pop(uuid, None)
+        return None
+    return payload
+
+
+def _cache_put(uuid: str, payload: dict) -> None:
+    _PRICE_CACHE[uuid] = (time.perf_counter(), payload)
+
+
+def clear_price_cache() -> None:
+    """Clear cached price responses (e.g. on login/logout/user switch)."""
+    _PRICE_CACHE.clear()
+
 
 def _normalize_version_uuid_list(values: Optional[Iterable[str]]) -> List[str]:
     if values is None:
@@ -21,7 +54,7 @@ def query_user_price(
     version_uuids: list[str] = [],
     page_size: int = 15,
     timeout: Tuple[float, float] = (1, 30),
-) -> dict:
+) -> list[dict]:
     """Return results for price lookup of multiple asset versions.
 
     The server endpoint now expects a POST body with `version_uuids`, so we keep
@@ -36,22 +69,44 @@ def query_user_price(
     if page_size > 0:
         version_uuid_list = version_uuid_list[:page_size]
 
-    payload: dict = {"version_uuids": version_uuid_list}
-
-    url = f"{paths.BLENDERKIT_API}/cart/request-price-bulk/"
-
-    if not payload["version_uuids"]:
+    if not version_uuid_list:
         raise ValueError("No version UUIDs provided for price lookup.")
 
-    headers = utils.get_simple_headers()
-    headers.setdefault("Content-Type", "application/json")
+    # Pull cached entries first.
+    fresh_uuids: list[str] = []
+    cached_results: list[dict] = []
+    for vu in version_uuid_list:
+        cached = _cache_get(vu)
+        if cached is None:
+            fresh_uuids.append(vu)
+        else:
+            cached_results.append(cached)
 
-    response = client_lib.blocking_request(
-        url,
-        "POST",
-        headers,
-        json_data=payload,
-        timeout=timeout,
-    )
-    search_results = response.json()
-    return search_results
+    fetched_results: list[dict] = []
+    if fresh_uuids:
+        payload: dict = {"version_uuids": fresh_uuids}
+        url = f"{paths.BLENDERKIT_API}/cart/request-price-bulk/"
+
+        headers = utils.get_simple_headers()
+        headers.setdefault("Content-Type", "application/json")
+        response = client_lib.blocking_request(
+            url,
+            "POST",
+            headers,
+            json_data=payload,
+            timeout=timeout,
+        )
+        fetched_results = response.json() or []
+        bk_logger.debug("Fetched price for %d version UUIDs", len(fetched_results))
+
+        for entry in fetched_results:
+            version_uuid = entry.get("versionUuid")
+            if not version_uuid:
+                continue
+            _cache_put(str(version_uuid), entry)
+
+    # Merge cached + fetched for the caller; order doesn't matter.
+    merged = []
+    merged.extend(cached_results)
+    merged.extend(fetched_results)
+    return merged