Mina Choi
41e49c693b
# Conflicts: # docs/DEVLOG.md # solution/backend/crud/site_crud.py # solution/backend/router/router.py # solution/backend/router/v1/site/protocol.py
261 lines
11 KiB
Python
261 lines
11 KiB
Python
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 list_company_sites(self, cdb: AsyncSession, company_id, skip, limit) -> Tuple[ErrorType, list, int]:
|
|
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 list_company_sites(self, cdb: AsyncSession, company_id, skip: int, limit: int) -> Tuple[ErrorType, list, int]:
|
|
"""회사의 사업장 + 사이트 + 마지막 빌드 시각. (ErrorType, [(place, site, built_at)], 총건수).
|
|
|
|
따로 읽으면 줄마다 사이트를 다시 물어 N+1 이다. LEFT JOIN 이라 사이트가 없는 사업장
|
|
(위저드만 걸어온 것)도 내려간다 — 빠지면 만들다 만 것을 찾을 길이 없다."""
|
|
try:
|
|
where = and_(places.deleted == False, places.company_id == company_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)
|
|
.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]:
|
|
"""다음 버전 번호. 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, []
|