"""v1 asset endpoints — admin asset catalog CRUD + tag apply/remove (REQ-010, REQ-005).""" from __future__ import annotations import logging from fastapi import APIRouter, Depends, HTTPException, Query, status from app.core.auth import AuthContext, require_admin from app.repositories.factory import get_portfolio_repository_from_settings from app.schemas.v1 import ( AssetCatalogCreateRequest, AssetCatalogListResponse, AssetCatalogResponse, AssetCatalogUpdateRequest, TagApplyRequest, TagApplyResponse, TagRefItem, ) logger = logging.getLogger(__name__) router = APIRouter(prefix="/v1", tags=["v1"]) def _get_repo(settings=None): from app.core.config import get_settings return get_portfolio_repository_from_settings(settings or get_settings()) def _build_tag_refs(tag_ids: list[str], tag_active_map: dict[str, bool]) -> list[TagRefItem]: """Convert raw tag_id list into TagRefItem array with archived flag synthesized. tag_ids not present in tag_active_map are surfaced as archived=True so the array is never silently pruned (REQ-002 Intent: "조용히 제거하지 않고 archived 상태로 반환"). """ refs: list[TagRefItem] = [] for tid in tag_ids: active = tag_active_map.get(tid, False) refs.append(TagRefItem(tag_id=tid, archived=not active)) return refs def _asset_response(asset, tag_active_map: dict[str, bool] | None = None) -> AssetCatalogResponse: tag_ids = getattr(asset, "tag_ids", []) or [] if tag_active_map is None and tag_ids: # 단일 응답 경로: 필요한 tag_ids만 1회 조회 repo = _get_repo() try: tags = repo.get_tags_by_ids(tag_ids) finally: if callable(getattr(type(repo), "close", None)): repo.close() tag_active_map = {tid: t.is_active for tid, t in tags.items()} elif tag_active_map is None: tag_active_map = {} return AssetCatalogResponse( asset_id=asset.asset_id, symbol=asset.symbol, markets=asset.markets, display_name=asset.display_name, status=asset.status, tag_ids=_build_tag_refs(tag_ids, tag_active_map), created_at=asset.created_at, updated_at=asset.updated_at, ) _CATALOG_STATUS_FILTERS = {"active", "inactive", "all"} # --------------------------------------------------------------------------- # GET /v1/assets — admin asset catalog list (REQ-010 CHG-20260521-02) # --------------------------------------------------------------------------- @router.get("/assets", response_model=AssetCatalogListResponse) def list_catalog_assets( status_filter: str = Query(default="active", alias="status"), market: str | None = Query(default=None), auth: AuthContext = Depends(require_admin), ) -> AssetCatalogListResponse: if status_filter not in _CATALOG_STATUS_FILTERS: raise HTTPException( status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail=f"status must be one of {sorted(_CATALOG_STATUS_FILTERS)}", ) repo = _get_repo() try: assets = repo.list_assets() if status_filter != "all": assets = [a for a in assets if a.status == status_filter] if market is not None: assets = [a for a in assets if market in (a.markets or [])] all_tag_ids: list[str] = [] seen: set[str] = set() for a in assets: for tid in (getattr(a, "tag_ids", []) or []): if tid not in seen: seen.add(tid) all_tag_ids.append(tid) tag_active_map: dict[str, bool] = {} if all_tag_ids: tags = repo.get_tags_by_ids(all_tag_ids) tag_active_map = {tid: t.is_active for tid, t in tags.items()} finally: if callable(getattr(type(repo), "close", None)): repo.close() items = [_asset_response(a, tag_active_map) for a in assets] return AssetCatalogListResponse(items=items, total=len(items)) # --------------------------------------------------------------------------- # Admin asset catalog CRUD (REQ-010) # --------------------------------------------------------------------------- @router.post("/assets", response_model=AssetCatalogResponse, status_code=201) def create_catalog_asset( body: AssetCatalogCreateRequest, auth: AuthContext = Depends(require_admin), ) -> AssetCatalogResponse: repo = _get_repo() try: asset = repo.insert_asset( symbol=body.symbol, markets=body.markets, display_name=body.display_name, status=body.status, ) finally: if callable(getattr(type(repo), "close", None)): repo.close() return _asset_response(asset) @router.patch("/assets/{asset_id}/deactivate", response_model=AssetCatalogResponse) def deactivate_catalog_asset( asset_id: str, auth: AuthContext = Depends(require_admin), ) -> AssetCatalogResponse: repo = _get_repo() try: asset = repo.update_asset(asset_id, status="inactive") finally: if callable(getattr(type(repo), "close", None)): repo.close() return _asset_response(asset) @router.post("/assets/{asset_id}/tags", response_model=TagApplyResponse) def apply_asset_tag( asset_id: str, body: TagApplyRequest, auth: AuthContext = Depends(require_admin), ) -> TagApplyResponse: if auth.token_source != "db": raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="DB token required") repo = _get_repo() try: try: repo.get(asset_id) except ValueError: raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"asset을 찾을 수 없습니다: {asset_id}") tag = repo.get_tag_by_id(body.tag_id) if tag is None: raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"태그를 찾을 수 없습니다: {body.tag_id}") if not tag.is_admin_tag: raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="admin 생성 태그만 asset에 적용할 수 있습니다") if not tag.is_active: raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="비활성화된 태그는 적용할 수 없습니다") repo.append_asset_tag_id(asset_id, body.tag_id) except HTTPException: raise except ValueError as exc: raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=str(exc)) from exc finally: if callable(getattr(type(repo), "close", None)): repo.close() return TagApplyResponse(tag_id=body.tag_id, status="applied") @router.delete("/assets/{asset_id}/tags/{tag_id}", response_model=TagApplyResponse) def remove_asset_tag( asset_id: str, tag_id: str, auth: AuthContext = Depends(require_admin), ) -> TagApplyResponse: if auth.token_source != "db": raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail="DB token required") repo = _get_repo() try: try: repo.get(asset_id) except ValueError: raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"asset을 찾을 수 없습니다: {asset_id}") repo.remove_asset_tag_id(asset_id, tag_id) except HTTPException: raise except ValueError as exc: raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=str(exc)) from exc finally: if callable(getattr(type(repo), "close", None)): repo.close() return TagApplyResponse(tag_id=tag_id, status="removed") @router.patch("/assets/{asset_id}", response_model=AssetCatalogResponse) def update_catalog_asset( asset_id: str, body: AssetCatalogUpdateRequest, auth: AuthContext = Depends(require_admin), ) -> AssetCatalogResponse: repo = _get_repo() try: asset = repo.update_asset( asset_id, markets=body.markets, display_name=body.display_name, status=body.status, ) finally: if callable(getattr(type(repo), "close", None)): repo.close() return _asset_response(asset)