o2o-site-AEO/solution/backend/router/v1/fact/fact.py

from typing import Optional
from uuid import UUID

from fastapi import APIRouter, Depends, Query

from common.models.gmodel import UserInfo
from router.v1.validator.dependencies import IsValidAccessToken, RemoveNoneResponse
from services.fact_service import FactService
from .protocol import (
    Req_ExtractFacts, Req_TransitionFact, Req_UpsertFact,
    Res_CategorySchema, Res_ExtractFacts, Res_Fact, Res_FactList,
)

# fact 라우터. 사업장(place_id) 하위 리소스이며, 회사 스코프는 service 가 사업장 조회로 강제한다.
router = APIRouter(prefix="/v1/place/{place_id}/fact", tags=["Fact"], responses={404: {"description": "Not found"}})


@router.get(
    path="/schema",
    response_model=Res_CategorySchema,
    summary="업종 fact 스키마",
    description="이 사업장의 업종에 존재하는 fact 필드 정의. 관리 화면이 입력 폼을 그리는 데 쓴다.",
)
async def get_schema(place_id: UUID, service: FactService = Depends(), user_info: UserInfo = Depends(IsValidAccessToken)):
    return RemoveNoneResponse(await service.get_category_schema(user_info, str(place_id)))


@router.get(
    path="/list",
    response_model=Res_FactList,
    summary="fact 목록",
    description="publishable_only=true 면 ★ 사이트에 나갈 수 있는 것(VERIFIED·CORRECTED)만.",
)
async def list_facts(
    place_id: UUID,
    service: FactService = Depends(),
    user_info: UserInfo = Depends(IsValidAccessToken),
    unit_id: Optional[UUID] = Query(None, description="객실·메뉴·프로그램 단위 fact 만"),
    publishable_only: bool = Query(False),
):
    return RemoveNoneResponse(await service.list_facts(user_info, str(place_id), unit_id, publishable_only))


@router.post(
    path="",
    response_model=Res_Fact,
    summary="fact 기록",
    description="같은 key 의 활성 fact 는 EXPIRED 로 밀리고 새 행이 들어간다(값 이력 보존). "
                "★ 자동 수집(api/crawl/llm)은 사장님 수정본(CORRECTED)을 덮어쓸 수 없다.",
)
async def upsert_fact(
    place_id: UUID, req: Req_UpsertFact, service: FactService = Depends(), user_info: UserInfo = Depends(IsValidAccessToken)
):
    return RemoveNoneResponse(await service.upsert_fact(user_info, str(place_id), req))


@router.post(
    path="/{fact_id}/transition",
    response_model=Res_Fact,
    summary="검증 상태 전이",
    description="허용 전이만 통과한다(FACT_STATUS_TRANSITIONS). CORRECTED 로 갈 때는 value 가 필수다.",
)
async def transition_fact(
    place_id: UUID,
    fact_id: UUID,
    req: Req_TransitionFact,
    service: FactService = Depends(),
    user_info: UserInfo = Depends(IsValidAccessToken),
):
    return RemoveNoneResponse(await service.transition(user_info, str(place_id), str(fact_id), req))


@router.post(
    path="/extract",
    response_model=Res_ExtractFacts,
    summary="붙여넣은 텍스트에서 fact 추출",
    description="사장님이 붙여넣은 소개글·안내문에서 업종 스키마 fact 를 뽑는다. "
                "★ 뽑은 값의 **근거 문장이 원문에 실제로 있는지 대조**해 통과한 것만 저장한다. "
                "저장된 값은 전부 후보(UNVERIFIED)라 사장님이 확인해야 사이트에 나간다. "
                "버려진 항목은 사유와 함께 돌려준다.",
)
async def extract_facts_from_text(
    place_id: UUID,
    req: Req_ExtractFacts,
    service: FactService = Depends(),
    user_info: UserInfo = Depends(IsValidAccessToken),
):
    return RemoveNoneResponse(await service.extract_from_text(user_info, str(place_id), req.text))