o2o-site-AEO/solution/backend/crud/site_crud.py

from abc import ABC, abstractmethod
from typing import Tuple

from sqlalchemy import and_, func, 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, places, site_publish_logs, site_versions, sites, users
from common.enums import BuildStatus, ErrorType, MediaStatus, SiteStatus
from common.logger import LOG
from common.utils.gtime import GTime


def _primary_photo_subquery():
    """place_photos 에서 대표 사진 한 장의 url 만 고르는 상관 서브쿼리(사업장당 1행)."""
    return (
        select(place_photos.url)
        .where(
            place_photos.place_id == places.place_id,
            place_photos.deleted == False,  # noqa: E712
            place_photos.status == MediaStatus.APPROVED.value,
            place_photos.unit_id.is_(None),
        )
        .order_by(place_photos.sort_order.asc(), place_photos.created_at.asc())
        .limit(1)
        .correlate(places)
        .scalar_subquery()
    )


# 사이트/버전/발행로그 CRUD.
class ISiteCRUD(ABC):
    @abstractmethod
    async def get_site_by_place(self, cdb: AsyncSession, place_id) -> Tuple[ErrorType, sites]:
        pass

    @abstractmethod
    async def get_site_by_domain(self, cdb: AsyncSession, domain: str) -> Tuple[ErrorType, sites]:
        pass

    @abstractmethod
    async def list_owner_sites(self, cdb: AsyncSession, owner_user_id, skip, limit) -> Tuple[ErrorType, list, int]:
        """(ErrorType, [(place, site, built_at, primary_photo_url)], 총건수)."""
        pass

    @abstractmethod
    async def list_all_sites(self, cdb: AsyncSession, skip, limit) -> Tuple[ErrorType, list, int]:
        """전 계정 사이트 목록(회사 스코프 없음) — 내부 운영(DEVELOPER) 전용."""
        pass

    @abstractmethod
    async def taken_domains(self, cdb: AsyncSession, domains: list) -> Tuple[ErrorType, set]:
        pass

    @abstractmethod
    async def add_site(self, cdb: AsyncSession, site: sites) -> ErrorType:
        pass

    @abstractmethod
    async def next_version_no(self, cdb: AsyncSession, site_id) -> Tuple[ErrorType, int]:
        pass

    @abstractmethod
    async def add_version(self, cdb: AsyncSession, version: site_versions) -> ErrorType:
        pass

    @abstractmethod
    async def get_version(self, cdb: AsyncSession, site_id, site_version_id) -> Tuple[ErrorType, site_versions]:
        pass

    @abstractmethod
    async def get_version_by_number(self, cdb: AsyncSession, site_id, version: int) -> Tuple[ErrorType, site_versions]:
        pass

    @abstractmethod
    async def finish_version(self, cdb: AsyncSession, site_version_id, data: dict) -> Tuple[ErrorType, int]:
        pass

    @abstractmethod
    async def update_site(self, cdb: AsyncSession, site_id, data: dict) -> Tuple[ErrorType, int]:
        pass

    @abstractmethod
    async def add_log(self, cdb: AsyncSession, log: site_publish_logs) -> ErrorType:
        pass

    @abstractmethod
    async def list_logs(self, cdb: AsyncSession, site_id, limit: int) -> Tuple[ErrorType, list]:
        pass

    @abstractmethod
    async def list_published(self, cdb: AsyncSession, limit: int) -> Tuple[ErrorType, list]:
        """(ErrorType, [(site, place, primary_photo_url)])."""
        pass


class SiteCRUD(ISiteCRUD):
    async def get_site_by_place(self, cdb: AsyncSession, place_id) -> Tuple[ErrorType, sites]:
        try:
            query = select(sites).where(sites.place_id == place_id, sites.deleted == False).limit(1)  # noqa: E712
            err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
            if err_type != ErrorType.SUCCESS:
                return err_type, None
            return ErrorType.SUCCESS, (rows[0] if rows else None)
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, None

    async def get_site_by_domain(self, cdb: AsyncSession, domain: str) -> Tuple[ErrorType, sites]:
        """주소(도메인 라벨)의 주인을 찾는다."""
        try:
            query = select(sites).where(sites.domain == domain, sites.deleted == False).limit(1)  # noqa: E712
            err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
            if err_type != ErrorType.SUCCESS:
                return err_type, None
            return ErrorType.SUCCESS, (rows[0] if rows else None)
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, None

    async def list_owner_sites(self, cdb: AsyncSession, owner_user_id, skip: int, limit: int) -> Tuple[ErrorType, list, int]:
        """사장님의 사업장 + 사이트 + 마지막 빌드 시각 + 빌더 대표 사진."""
        try:
            where = and_(places.deleted == False, places.owner_user_id == owner_user_id)  # noqa: E712

            cnt_err, cnt_rows = await DB_SESSION_MNG.execute(cdb, select(func.count()).select_from(places).where(where))
            if cnt_err != ErrorType.SUCCESS:
                return cnt_err, [], 0
            total = int(cnt_rows[0] or 0) if cnt_rows else 0

            query = (
                select(places, sites, site_versions.built_at, _primary_photo_subquery())
                .outerjoin(sites, and_(sites.place_id == places.place_id, sites.deleted == False))  # noqa: E712
                .outerjoin(site_versions, site_versions.site_version_id == sites.current_version_id)
                .where(where)
                .order_by(places.created_at.desc())
                .offset(skip)
                .limit(limit)
            )
            list_err, rows = await DB_SESSION_MNG.execute(cdb, query)
            if list_err != ErrorType.SUCCESS:
                return list_err, [], 0
            return ErrorType.SUCCESS, list(rows), total
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, [], 0

    async def list_all_sites(self, cdb: AsyncSession, skip: int, limit: int) -> Tuple[ErrorType, list, int]:
        """list_owner_sites 와 같은 조인이되 owner_user_id 필터가 없다 — 소유자 계정 정보를 같이 얹는다."""
        try:
            where = places.deleted == False  # noqa: E712

            cnt_err, cnt_rows = await DB_SESSION_MNG.execute(cdb, select(func.count()).select_from(places).where(where))
            if cnt_err != ErrorType.SUCCESS:
                return cnt_err, [], 0
            total = int(cnt_rows[0] or 0) if cnt_rows else 0

            query = (
                select(places, sites, site_versions.built_at, _primary_photo_subquery(), users.id, users.email, users.name)
                .join(users, users.user_id == places.owner_user_id)
                .outerjoin(sites, and_(sites.place_id == places.place_id, sites.deleted == False))  # noqa: E712
                .outerjoin(site_versions, site_versions.site_version_id == sites.current_version_id)
                .where(where)
                .order_by(places.created_at.desc())
                .offset(skip)
                .limit(limit)
            )
            list_err, rows = await DB_SESSION_MNG.execute(cdb, query)
            if list_err != ErrorType.SUCCESS:
                return list_err, [], 0
            return ErrorType.SUCCESS, list(rows), total
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, [], 0

    async def taken_domains(self, cdb: AsyncSession, domains: list) -> Tuple[ErrorType, set]:
        """후보 주소들 중 이미 쓰이는 것만 추린다."""
        try:
            if not domains:
                return ErrorType.SUCCESS, set()
            query = select(sites.domain).where(
                sites.domain.in_(domains), sites.deleted == False  # noqa: E712
            )
            err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
            if err_type != ErrorType.SUCCESS:
                return err_type, set()
            return ErrorType.SUCCESS, {r for r in rows if r}
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, set()

    async def add_site(self, cdb: AsyncSession, site: sites) -> ErrorType:
        try:
            return await DB_SESSION_MNG.insert(cdb, site)
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED

    async def next_version_no(self, cdb: AsyncSession, site_id) -> Tuple[ErrorType, int]:
        """다음 버전 번호."""
        try:
            query = select(func.max(site_versions.version)).where(
                site_versions.site_id == site_id, site_versions.deleted == False  # noqa: E712
            )
            err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
            if err_type != ErrorType.SUCCESS:
                return err_type, 0
            current = rows[0] if rows else None
            return ErrorType.SUCCESS, int(current or 0) + 1
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, 0

    async def add_version(self, cdb: AsyncSession, version: site_versions) -> ErrorType:
        try:
            return await DB_SESSION_MNG.insert(cdb, version)
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED

    async def get_version(self, cdb: AsyncSession, site_id, site_version_id) -> Tuple[ErrorType, site_versions]:
        try:
            query = (
                select(site_versions)
                .where(
                    site_versions.site_version_id == site_version_id,
                    site_versions.site_id == site_id,
                    site_versions.deleted == False,  # noqa: E712
                )
                .limit(1)
            )
            err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
            if err_type != ErrorType.SUCCESS:
                return err_type, None
            if len(rows) != 1:
                return ErrorType.DB_INVALID_KEY, None
            return ErrorType.SUCCESS, rows[0]
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, None

    async def get_version_by_number(self, cdb: AsyncSession, site_id, version: int) -> Tuple[ErrorType, site_versions]:
        """롤백 대상 조회 — site_versions.version(사람이 보는 번호) 로 찾는다."""
        try:
            query = (
                select(site_versions)
                .where(
                    site_versions.site_id == site_id,
                    site_versions.version == version,
                    site_versions.deleted == False,  # noqa: E712
                )
                .limit(1)
            )
            err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
            if err_type != ErrorType.SUCCESS:
                return err_type, None
            if len(rows) != 1:
                return ErrorType.DB_INVALID_KEY, None
            return ErrorType.SUCCESS, rows[0]
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, None

    async def finish_version(self, cdb: AsyncSession, site_version_id, data: dict) -> Tuple[ErrorType, int]:
        """빌드 결과를 버전에 기록한다(BUILT 또는 FAILED)."""
        try:
            query = (
                update(site_versions)
                .where(site_versions.site_version_id == site_version_id, site_versions.deleted == False)  # noqa: E712
                .values(**data, updated_at=GTime.UTC())
            )
            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 update_site(self, cdb: AsyncSession, site_id, data: dict) -> Tuple[ErrorType, int]:
        try:
            query = (
                update(sites)
                .where(sites.site_id == site_id, sites.deleted == False)  # noqa: E712
                .values(**data, updated_at=GTime.UTC())
            )
            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 add_log(self, cdb: AsyncSession, log: site_publish_logs) -> ErrorType:
        try:
            return await DB_SESSION_MNG.insert(cdb, log)
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED

    async def list_logs(self, cdb: AsyncSession, site_id, limit: int = 50) -> Tuple[ErrorType, list]:
        try:
            query = (
                select(site_publish_logs)
                .where(site_publish_logs.site_id == site_id, site_publish_logs.deleted == False)  # noqa: E712
                .order_by(site_publish_logs.created_at.desc())
                .limit(limit)
            )
            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 list_published(self, cdb: AsyncSession, limit: int = 12) -> Tuple[ErrorType, list]:
        """발행된 사이트 + 그 사업장 + 빌더 대표 사진을 최신순으로."""
        try:
            query = (
                select(sites, places, _primary_photo_subquery())
                .join(places, places.place_id == sites.place_id)
                .where(
                    sites.status == SiteStatus.PUBLISHED.value,
                    sites.deleted == False,  # noqa: E712
                    places.deleted == False,  # noqa: E712
                )
                .order_by(sites.published_at.desc().nulls_last(), sites.created_at.desc())
                .limit(limit)
            )
            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, []