o2o-site-AEO/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 media, places
from common.enums import DBWRType, ErrorType, MediaStatus, SourceType
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.company_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(
            media.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