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 places, publish_logs, site_versions, sites from common.enums import BuildStatus, ErrorType, SiteStatus from common.logger import LOG from common.utils.gtime import GTime # 사이트/버전/발행로그 CRUD. 항상 place_id 또는 site_id 로 스코프한다. 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 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 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: 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]: 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]: """주소(도메인 라벨)의 주인을 찾는다. 없으면 (SUCCESS, None). uq_sites_domain(deleted=false AND domain IS NOT NULL)과 같은 조건으로 본다 — 인덱스가 막는 것과 조회가 막는 것이 다르면 "확인은 통과, 저장은 실패"가 난다.""" 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 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]: """다음 버전 번호. 1부터 시작한다.""" 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 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: 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(publish_logs) .where(publish_logs.site_id == site_id, publish_logs.deleted == False) # noqa: E712 .order_by(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]: """발행된 사이트 + 그 사업장을 최신순으로. 랜딩 쇼케이스가 읽는 목록이다. ★ 회사 스코프가 없는 **유일한** 사이트 조회다(비로그인 API 가 쓴다). 그래서 행을 통째로 돌려주고, 무엇이 밖으로 나갈지는 services/showcase_service 한 곳에서만 고른다 — 여기서 열을 골라 두면 나중에 필드를 늘릴 때 공개 여부를 판단할 자리가 사라진다.""" try: query = ( select(sites, places) .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, []