o2o-site-AEO/solution/backend/crud/media_crud.py
Mina Choi 1dc4f5fbf5 merge: origin/main(에이전트 여러 동작 · 사진 도구 · 카톡 승인 알림) 을 feature/template-catalog 에 병합
충돌은 이 브랜치의 주석 정리 커밋(11d30bb)과 main 의 기능 변경이 겹친 백엔드 9개 파일 —
main 쪽을 그대로 썼다(기능이 main 에만 있다). AGENTS.md 는 양쪽 링크를 모두 두고,
DEVLOG 는 요약본 위에 main 의 09-28 에이전트 항목 셋을 붙였다.

site eslint 통과 · vitest 123 passed · frontend tsc·eslint 통과 · 백엔드 py_compile 통과
(백엔드 pytest 는 이 기계에 테스트 DB 계정이 없어 실행하지 못함)

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
2026-09-30 13:51:53 +09:00

103 lines
4.8 KiB
Python

from abc import ABC, abstractmethod
from typing import Tuple
from sqlalchemy import and_, func, or_, select, update
from sqlalchemy.ext.asyncio import AsyncSession
from common.database.db_session_manager import DB_SESSION_MNG
from common.database.model.models import place_photos
from common.enums import ErrorType, MediaStatus
from common.logger import LOG
from common.utils.gtime import GTime
# 사진 CRUD.
class IMediaCRUD(ABC):
@abstractmethod
async def list_media(
self, cdb: AsyncSession, place_id, status=None, unlabeled_only=False, unit_id=None, alt_required=False
) -> Tuple[ErrorType, list]:
pass
@abstractmethod
async def apply_vision(self, cdb: AsyncSession, media_id, label, alt_text, confidence, status, ts) -> Tuple[ErrorType, int]:
pass
@abstractmethod
async def set_sort_order(self, cdb: AsyncSession, place_id, media_id, sort_order: int, ts) -> Tuple[ErrorType, int]:
pass
async def set_status(self, cdb: AsyncSession, place_id, media_id, status: int, ts) -> Tuple[ErrorType, int]:
pass
class MediaCRUD(IMediaCRUD):
async def list_media(
self, cdb: AsyncSession, place_id, status=None, unlabeled_only: bool = False, unit_id=None, alt_required: bool = False
) -> Tuple[ErrorType, list]:
"""사진 목록."""
try:
conditions = [place_photos.place_id == place_id, place_photos.deleted == False] # noqa: E712
if status is not None:
conditions.append(place_photos.status == status)
if unlabeled_only:
conditions.append(
or_(place_photos.alt_text.is_(None), func.btrim(place_photos.alt_text) == "")
)
if unit_id is not None:
conditions.append(place_photos.unit_id == unit_id)
if alt_required:
# 공백만 있는 alt 도 빌더에선 '없음'이다 — 같은 기준으로 거른다.
conditions.append(place_photos.alt_text.is_not(None))
conditions.append(func.btrim(place_photos.alt_text) != "")
query = select(place_photos).where(and_(*conditions)).order_by(place_photos.sort_order.asc(), place_photos.created_at.asc())
err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
return (err_type, list(rows) if err_type == ErrorType.SUCCESS else [])
except Exception as ex:
LOG.e_no_callstack(ex)
return ErrorType.DB_RUN_FAILED, []
async def apply_vision(self, cdb: AsyncSession, media_id, label, alt_text, confidence, status: int, ts) -> Tuple[ErrorType, int]:
"""Vision 분석 결과를 반영한다."""
try:
query = (
update(place_photos)
.where(place_photos.media_id == media_id, place_photos.deleted == False) # noqa: E712
.values(label=label, alt_text=alt_text, vision_confidence=confidence, status=status, updated_at=ts)
)
return await DB_SESSION_MNG.add_with_rowcount(cdb, query)
except Exception as ex:
LOG.e_no_callstack(ex)
return ErrorType.DB_RUN_FAILED, 0
async def set_sort_order(self, cdb: AsyncSession, place_id, media_id, sort_order: int, ts) -> Tuple[ErrorType, int]:
"""사진 순서를 바꾼다.
★ 목록이 `ORDER BY sort_order, created_at` 이고 대표 사진은 그 **첫 장**이다
(site_payload.primary_media). 그래서 '대표로 지정' 은 이 값을 가장 작게 만드는 일이다 —
별도의 is_primary 칸을 두지 않는 이유는, 두면 두 규칙(칸 · 순서)이 생겨
검색 결과에 뜨는 그림과 화면 첫 장이 갈릴 수 있기 때문이다."""
try:
query = (
update(place_photos)
.where(place_photos.media_id == media_id, place_photos.place_id == place_id, place_photos.deleted == False) # noqa: E712
.values(sort_order=sort_order, updated_at=ts)
)
return await DB_SESSION_MNG.add_with_rowcount(cdb, query)
except Exception as ex:
LOG.e_no_callstack(ex)
return ErrorType.DB_RUN_FAILED, 0
async def set_status(self, cdb: AsyncSession, place_id, media_id, status: int, ts) -> Tuple[ErrorType, int]:
"""사람이 사진을 승인/반려한다."""
try:
query = (
update(place_photos)
.where(place_photos.media_id == media_id, place_photos.place_id == place_id, place_photos.deleted == False) # noqa: E712
.values(status=status, updated_at=ts)
)
return await DB_SESSION_MNG.add_with_rowcount(cdb, query)
except Exception as ex:
LOG.e_no_callstack(ex)
return ErrorType.DB_RUN_FAILED, 0