diff --git a/solution/backend/scripts/backfill_thumbnails.py b/solution/backend/scripts/backfill_thumbnails.py new file mode 100644 index 0000000..0253be5 --- /dev/null +++ b/solution/backend/scripts/backfill_thumbnails.py @@ -0,0 +1,107 @@ +"""썸네일이 없는 발행 사이트에 썸네일을 채운다. + + python scripts/backfill_thumbnails.py (backend/ 에서 실행) + python scripts/backfill_thumbnails.py --dry-run (올리지 않고 대상만 본다) + python scripts/backfill_thumbnails.py --all (이미 있는 것도 다시 만든다) + +★ 왜 필요한가 — 썸네일은 **발행 잡이 끝날 때** 만들어진다(build_service). 그래서 이 기능이 + 들어오기 전에 발행된 사이트는 thumbnail_url 이 영영 NULL 이고, 쇼케이스에서 글자 카드로만 + 나온다. 재발행을 시키면 채워지지만 사장님 사이트를 우리 사정으로 다시 굽는 건 다른 일이다 + — 썸네일만 따로 만든다. + +★ 발행본 HTML 을 건드리지 않는다. 읽는 건 site_versions.snapshot 의 사진 목록뿐이고, + 쓰는 건 Blob 의 thumbs/ 와 sites.thumbnail_url 한 칸이다. +""" +import argparse, asyncio, os, sys + +sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) +os.environ.setdefault("APP_ENV", "local") + +from sqlalchemy import select # noqa: E402 + +from common.database.db_session_manager import DB_SESSION_MNG # noqa: E402 +from common.database.model.models import places, site_versions, sites # noqa: E402 +from common.enums import DBWRType, ErrorType, SiteStatus # noqa: E402 +from crud.site_crud import SiteCRUD # noqa: E402 +from services import site_payload, site_thumbnail # noqa: E402 + +_crud = SiteCRUD() + + +def _targets(only_missing: bool): + """발행된 사이트 + 그 사이트의 현재 버전 스냅샷. 슬러그 계산에 places 가 필요하다.""" + stmt = ( + select(sites, places, site_versions) + .join(places, places.place_id == sites.place_id) + .join(site_versions, site_versions.site_version_id == sites.current_version_id) + .where( + sites.deleted == False, # noqa: E712 + sites.status == SiteStatus.PUBLISHED.value, + ) + ) + if only_missing: + stmt = stmt.where(sites.thumbnail_url.is_(None)) + + async def run_query(session): + return await DB_SESSION_MNG.execute(session, stmt) + + return run_query + + +async def run(dry_run: bool, only_missing: bool) -> None: + # --dry-run 은 목록만 본다 — Azure 설정 없이도 대상이 맞는지 확인할 수 있어야 한다. + if not dry_run and not site_thumbnail.is_configured(): + raise SystemExit( + "AZURE_STORAGE_CONNECTION_STRING 이 없습니다 — 업로드할 곳이 없어 아무것도 하지 않습니다." + ) + + err, rows = await DB_SESSION_MNG.execute_lambda( + sites.DBType(), DBWRType.DB_READ.value, _targets(only_missing) + ) + if err != ErrorType.SUCCESS: + raise SystemExit(f"대상 조회 실패: {err}") + + print(f"[thumb] 대상 {len(rows)}건 ({'없는 것만' if only_missing else '전부'})") + made = skipped = failed = 0 + + for site, place, version in rows: + slug = site_payload.publish_slug(place, site) + snapshot = version.snapshot or {} + if dry_run: + has_photo = bool(site_payload.primary_media(snapshot)) + print(f" - {slug:<24} {place.name} {'' if has_photo else '(대표 사진 없음 — 건너뜀)'}") + continue + + try: + url = await site_thumbnail.store(slug, snapshot) + except Exception as ex: # noqa: BLE001 — 한 건 실패가 나머지를 막지 않는다 + print(f" ✗ {slug}: {type(ex).__name__}: {ex}") + failed += 1 + continue + + if not url: + # 대표 사진이 없거나 받지 못했다. 정상적인 경우다 — 사진 없는 가게가 있다. + skipped += 1 + continue + + await DB_SESSION_MNG.execute_lambda_claim( + sites.DBType(), lambda s, sid=site.site_id, u=url: _crud.update_site(s, sid, {"thumbnail_url": u}) + ) + print(f" ✓ {slug} → {url}") + made += 1 + + if not dry_run: + print(f"[thumb] 완료 — 만듦 {made} · 사진 없음 {skipped} · 실패 {failed}") + await DB_SESSION_MNG.dispose_all() + + +def main() -> None: + parser = argparse.ArgumentParser() + parser.add_argument("--dry-run", action="store_true", help="업로드 없이 대상만 출력") + parser.add_argument("--all", action="store_true", help="이미 썸네일이 있는 사이트도 다시 만든다") + args = parser.parse_args() + asyncio.run(run(args.dry_run, only_missing=not args.all)) + + +if __name__ == "__main__": + main()