first commit
This commit is contained in:
54
backend/app/api/v1/analysis.py
Normal file
54
backend/app/api/v1/analysis.py
Normal file
@@ -0,0 +1,54 @@
|
||||
from uuid import UUID
|
||||
|
||||
from fastapi import APIRouter, Depends, HTTPException
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.core.database import get_db
|
||||
from app.repositories.case_repo import CaseRepository
|
||||
from app.schemas.analysis import AnalysisStatusOut, AnalysisTriggerOut
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
@router.post("/cases/{case_id}/analyze", response_model=AnalysisTriggerOut)
|
||||
async def trigger_analysis(case_id: UUID, db: AsyncSession = Depends(get_db)):
|
||||
repo = CaseRepository(db)
|
||||
case = await repo.get(case_id)
|
||||
if not case:
|
||||
raise HTTPException(404, "案件不存在")
|
||||
|
||||
from app.workers.analysis_tasks import run_full_analysis
|
||||
try:
|
||||
task = run_full_analysis.delay(str(case_id))
|
||||
task_id = task.id
|
||||
except Exception:
|
||||
task_id = "sync-fallback"
|
||||
from app.services.analysis_pipeline import run_analysis_sync
|
||||
await run_analysis_sync(case_id, db)
|
||||
|
||||
return AnalysisTriggerOut(task_id=task_id, message="分析任务已提交")
|
||||
|
||||
|
||||
@router.get("/cases/{case_id}/analyze/status", response_model=AnalysisStatusOut)
|
||||
async def analysis_status(case_id: UUID, db: AsyncSession = Depends(get_db)):
|
||||
repo = CaseRepository(db)
|
||||
case = await repo.get(case_id)
|
||||
if not case:
|
||||
raise HTTPException(404, "案件不存在")
|
||||
|
||||
step_map = {
|
||||
"pending": ("等待上传", 0),
|
||||
"uploading": ("上传中", 15),
|
||||
"analyzing": ("分析中", 50),
|
||||
"reviewing": ("待复核", 85),
|
||||
"completed": ("已完成", 100),
|
||||
}
|
||||
step_label, progress = step_map.get(case.status.value, ("未知", 0))
|
||||
|
||||
return AnalysisStatusOut(
|
||||
case_id=str(case_id),
|
||||
status=case.status.value,
|
||||
progress=progress,
|
||||
current_step=step_label,
|
||||
message=f"当前状态: {step_label}",
|
||||
)
|
||||
Reference in New Issue
Block a user