Preview: simplify pyvips timing to single processing ms + req total.

2026-04-25 00:06:35 +00:00
parent 48112dec26
commit bbb73be07e
2 changed files with 112 additions and 16 deletions
@@ -0,0 +1,102 @@
+import argparse
+import mimetypes
+from pathlib import Path
+
+from cista.preview import process_image_with_timing
+
+
+def parse_args() -> argparse.Namespace:
+    parser = argparse.ArgumentParser(
+        description="Generate image previews for all files in a folder, one at a time.",
+    )
+    parser.add_argument("folder", type=Path, help="Folder to scan recursively")
+    parser.add_argument(
+        "--px",
+        type=int,
+        default=1024,
+        help="Maximum preview dimension in pixels (default: 1024)",
+    )
+    parser.add_argument(
+        "--quality",
+        type=int,
+        default=60,
+        help="AVIF quality passed to preview generation (default: 60)",
+    )
+    return parser.parse_args()
+
+
+def is_image_file(path: Path) -> bool:
+    mime_type, _ = mimetypes.guess_type(path.name)
+    return bool(mime_type and mime_type.startswith("image/"))
+
+
+def main() -> int:
+    args = parse_args()
+    folder = args.folder.resolve()
+    if not folder.is_dir():
+        raise SystemExit(f"Not a directory: {folder}")
+
+    files = sorted(path for path in folder.rglob("*") if path.is_file() and is_image_file(path))
+    if not files:
+        print(f"No image files found under {folder}")
+        return 0
+
+    total_files = 0
+    total_bytes = 0
+    total_load_ms: float = 0.0
+    total_process_ms: float = 0.0
+    total_save_ms: float = 0.0
+    total_preview_ms: float = 0.0
+    failures = 0
+
+    print(f"Scanning {folder}")
+    print(f"Generating previews for {len(files)} image files")
+
+    for path in files:
+        total_files += 1
+        rel = path.relative_to(folder)
+        try:
+            preview, timing = process_image_with_timing(
+                path,
+                maxsize=args.px,
+                quality=args.quality,
+            )
+        except Exception as exc:
+            failures += 1
+            print(f"FAIL  {rel}  error={exc}")
+            continue
+
+        total_bytes += len(preview)
+        total_load_ms += timing.load_ms or 0.0
+        total_process_ms += timing.process_ms or 0.0
+        total_save_ms += timing.save_ms or 0.0
+        total_preview_ms += timing.total_ms or 0.0
+
+        if timing.load_ms is not None:
+            detail = (
+                f"load={timing.load_ms:.1f}ms  process={timing.process_ms:.1f}ms  "
+                f"save={timing.save_ms:.1f}ms  total={timing.total_ms:.1f}ms"
+            )
+        else:
+            detail = f"total={timing.total_ms:.1f}ms"
+        print(f"OK    {rel}  backend={timing.backend}  bytes={len(preview)}  {detail}")
+
+    completed = total_files - failures
+    print()
+    print("Summary")
+    print(f"  files={total_files}")
+    print(f"  completed={completed}")
+    print(f"  failed={failures}")
+    print(f"  preview_bytes={total_bytes}")
+    if completed:
+        if total_load_ms or total_process_ms or total_save_ms:
+            print(f"  load_total_ms={total_load_ms:.1f}")
+            print(f"  process_total_ms={total_process_ms:.1f}")
+            print(f"  save_total_ms={total_save_ms:.1f}")
+        print(f"  preview_total_ms={total_preview_ms:.1f}")
+        print(f"  preview_avg_ms={total_preview_ms / completed:.1f}")
+    return 0 if failures == 0 else 1
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())