o2o-site-AEO/solution/backend/services/media_service.py

import uuid

from fastapi import Depends

from common.database.db_session_manager import DB_SESSION_MNG
from common.database.model.models import place_photos, places
from common.enums import DBWRType, ErrorType, MediaStatus, SourceType
from common.utils.gtime import GTime
from common.models.gmodel import UserInfo
from crud.media_crud import IMediaCRUD, MediaCRUD
from crud.place_crud import PlaceCRUD
from router.v1.media.protocol import MediaData, Res_MediaList


def _is_publishable(row) -> bool:
    """이 사진이 지금 사이트에 실릴 수 있는가.

    ★ 판단 기준을 services/snapshot.py 와 한 글자도 다르지 않게 맞춘다 —
      관리 화면이 '나간다'고 표시한 사진이 발행에서 빠지면 그게 제일 설명하기 어려운 버그다.
      승인(APPROVED)만으로는 부족하다. alt 가 빈 사진은 빌더가 렌더 자체를 하지 않는다."""
    return row.status == MediaStatus.APPROVED.value and bool((row.alt_text or "").strip()) and bool((row.url or "").strip())


class MediaService:
    """사진 조회 · 노출 여부 · 순서.

    ★ 이 서비스가 지키는 규칙은 둘이다.
      1. 회사 스코프 — 사업장을 먼저 회사 스코프로 로드해서 남의 회사 사진에 닿지 못하게 한다.
         (fact/site 와 같은 _load_place 패턴. 없는 것과 남의 것은 똑같이 PLACE_NOT_FOUND 로 답한다)
      2. 출처 보존 — source_type / origin_url 을 절대 응답에서 빼지 않는다.
         크롤링 이미지 재게시 권리가 미결이고(docs/DECISIONS.md 1-2), 결론이 '불가'면
         발행에서 source_type = CRAWL 을 통째로 제외해야 한다. 그 필터를 화면이 미리
         보여주려면 출처가 목록에 실려 있어야 한다.
    """

    def __init__(self, crud: IMediaCRUD = Depends(MediaCRUD), place_crud: PlaceCRUD = Depends(PlaceCRUD)):
        self.crud = crud
        self.place_crud = place_crud

    # ---- 사업장 로드(회사 스코프) ----
    async def _load_place(self, user_info: UserInfo, place_id: str):
        err_type, place = await DB_SESSION_MNG.execute_lambda(
            places.DBType(),
            DBWRType.DB_READ.value,
            lambda s: self.place_crud.get_place(s, uuid.UUID(user_info.user_id), uuid.UUID(place_id)),
        )
        if err_type != ErrorType.SUCCESS:
            return ErrorType.PLACE_NOT_FOUND, None
        return ErrorType.SUCCESS, place

    # ---- 조회 ----
    async def list_media(self, user_info: UserInfo, place_id: str, unit_id=None, publishable_only: bool = False) -> Res_MediaList:
        """사진 목록. 관리자 빌더 캔버스와 사장님 확인 화면이 같은 엔드포인트를 쓴다.

        publishable_only=True 는 '발행하면 실제로 실릴 것'만 — 승인 + alt 있음.
        alt 조건을 여기서 같이 거는 게 중요하다. 승인만 보고 목록을 그리면 캔버스에는
        사진이 보이는데 발행된 사이트엔 없는 상태가 되고, 원인을 찾는 데 반나절이 든다.

        사진이 0장인 것은 오류가 아니다 — 수집 전이거나 Vision 이 아직 안 돌았을 뿐이라
        빈 배열을 그대로 돌려준다(호출자가 '수집을 돌리세요'를 띄울 수 있게)."""
        res = Res_MediaList()
        err_type, _place = await self._load_place(user_info, place_id)
        if err_type != ErrorType.SUCCESS:
            res.result.SetResult(err_type)
            return res

        # publishable_only 의 승인 조건만 CRUD 에 넘기고, alt 조건은 목록 계산과 함께 아래에서 건다.
        status = MediaStatus.APPROVED.value if publishable_only else None
        list_err, rows = await DB_SESSION_MNG.execute_lambda(
            place_photos.DBType(),
            DBWRType.DB_READ.value,
            lambda s: self.crud.list_media(
                s, uuid.UUID(place_id), status, False, unit_id, publishable_only
            ),
        )
        if list_err != ErrorType.SUCCESS:
            res.result.SetResult(list_err)
            return res

        items = []
        for r in rows:
            data = MediaData.model_validate(r)
            data.publishable = _is_publishable(r)
            items.append(data)
        res.media = items
        res.publishable = sum(1 for x in items if x.publishable)
        # 사람 확인 큐에 남은 수 — 관리 화면의 '검토할 것' 배지.
        res.pending_review = sum(1 for r in rows if r.status == MediaStatus.PENDING_REVIEW.value)
        # ★ 재게시 권리(1-2)가 '불가'로 결론나면 통째로 빠질 사진 수. 미리 보여줘야 사장님이
        #   직접 올릴 사진을 몇 장 준비해야 하는지 안다.
        res.crawled = sum(1 for r in rows if r.source_type == SourceType.CRAWL.value)
        return res

    # ---- 노출 여부 · 순서 ----
    #
    # ★ 업로드·교체는 여기 없다. 이미지 재게시 권리가 미결이라(docs/DECISIONS.md 1-2) 저장
    #   경로를 일부러 안 만들어 둔 것이고(5-3), 아래 둘은 **이미 우리 DB 에 있는 사진의
    #   노출과 순서**만 바꾼다 — 그 결론이 어느 쪽으로 나든 영향받지 않는다.

    async def _owned_photo(self, user_info: UserInfo, place_id: str, media_id: str):
        """사업장 스코프 + 그 사업장의 사진인지. ★ 없는 것과 남의 것을 똑같이 답한다."""
        err_type, _place = await self._load_place(user_info, place_id)
        if err_type != ErrorType.SUCCESS:
            return err_type, None
        _err, rows = await DB_SESSION_MNG.execute_lambda(
            place_photos.DBType(),
            DBWRType.DB_READ.value,
            lambda s: self.crud.list_media(s, uuid.UUID(place_id), None, False, None, False),
        )
        for row in rows or []:
            if str(row.media_id) == str(media_id):
                return ErrorType.SUCCESS, row
        return ErrorType.MEDIA_NOT_FOUND, None

    async def hide_media(self, user_info: UserInfo, place_id: str, media_id: str) -> Res_MediaList:
        """사진을 내린다 — REJECTED 로 내려 발행본에서 빠진다.

        ★ 지우지 않는다. `origin_url`·`source_type` 이 남아 있어야 재게시 권리(1-2) 결론이
          나왔을 때 무엇을 실었는지 되짚을 수 있고, 잘못 내렸을 때 되돌릴 수도 있다."""
        err_type, row = await self._owned_photo(user_info, place_id, media_id)
        if err_type != ErrorType.SUCCESS:
            res = Res_MediaList()
            res.result.SetResult(err_type)
            return res
        await DB_SESSION_MNG.execute_lambda_write(
            place_photos.DBType(),
            lambda s: self.crud.set_status(s, uuid.UUID(place_id), row.media_id, MediaStatus.REJECTED.value, GTime.UTC()),
        )
        return await self.list_media(user_info, place_id)

    async def set_primary(self, user_info: UserInfo, place_id: str, media_id: str) -> Res_MediaList:
        """대표 사진으로 올린다.

        ★ 대표는 목록의 **첫 장**이다(site_payload.primary_media) — 별도 칸을 두지 않는다.
          그래서 이 사진의 sort_order 를 지금 가장 작은 값보다 하나 더 작게 내리면 된다.
        ★ 객실·메뉴 전용 사진(unit_id 가 있는 것)은 대표가 될 수 없다 — primary_media 가
          그런 사진을 건너뛰므로, 지정해도 화면만 거짓말하게 된다."""
        err_type, row = await self._owned_photo(user_info, place_id, media_id)
        res = Res_MediaList()
        if err_type != ErrorType.SUCCESS:
            res.result.SetResult(err_type)
            return res
        if row.unit_id is not None:
            res.result.SetResult(ErrorType.INVALID_REQUEST_DATA)
            return res

        _err, rows = await DB_SESSION_MNG.execute_lambda(
            place_photos.DBType(),
            DBWRType.DB_READ.value,
            lambda s: self.crud.list_media(s, uuid.UUID(place_id), None, False, None, False),
        )
        lowest = min((r.sort_order or 0) for r in (rows or [row]))
        await DB_SESSION_MNG.execute_lambda_write(
            place_photos.DBType(),
            lambda s: self.crud.set_sort_order(s, uuid.UUID(place_id), row.media_id, lowest - 1, GTime.UTC()),
        )
        return await self.list_media(user_info, place_id)