77 lines
2.2 KiB
Python
77 lines
2.2 KiB
Python
from uuid import UUID
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, Query
|
|
from psycopg import AsyncConnection
|
|
|
|
from app.infra.db.postgresql.analysis import AnalysisRepository
|
|
from app.auth.project_dependencies import get_project_pg_connection
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
async def get_database_connection(
|
|
conn: AsyncConnection = Depends(get_project_pg_connection),
|
|
):
|
|
"""获取数据库连接"""
|
|
yield conn
|
|
|
|
|
|
@router.get("/analysis/runs", summary="获取分析运行列表")
|
|
async def get_analysis_runs(
|
|
conn: AsyncConnection = Depends(get_database_connection),
|
|
):
|
|
"""
|
|
获取所有方案信息
|
|
|
|
返回项目中所有方案的详细信息
|
|
"""
|
|
try:
|
|
runs = await AnalysisRepository.list_runs(conn)
|
|
return {"success": True, "data": runs, "count": len(runs)}
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=f"查询分析运行时发生错误: {str(e)}")
|
|
|
|
|
|
@router.get("/analysis/runs/{run_id}", summary="获取分析运行")
|
|
async def get_analysis_run(
|
|
run_id: UUID,
|
|
conn: AsyncConnection = Depends(get_database_connection),
|
|
):
|
|
"""
|
|
获取所有爆管定位结果
|
|
|
|
返回项目中所有的爆管定位分析结果
|
|
"""
|
|
try:
|
|
run = await AnalysisRepository.get_run(conn, run_id)
|
|
if run is None:
|
|
raise HTTPException(status_code=404, detail="分析运行不存在")
|
|
return run
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
raise HTTPException(
|
|
status_code=500, detail=f"查询分析运行时发生错误: {str(e)}"
|
|
)
|
|
|
|
|
|
@router.get("/analysis/runs/{run_id}/results", summary="获取分析结果")
|
|
async def get_analysis_results(
|
|
run_id: UUID,
|
|
result_type: str | None = Query(default=None, description="结果类型"),
|
|
conn: AsyncConnection = Depends(get_database_connection),
|
|
):
|
|
"""
|
|
根据爆管事件ID查询爆管定位结果
|
|
|
|
参数:
|
|
burst_incident: 爆管事件的唯一标识符
|
|
"""
|
|
try:
|
|
return await AnalysisRepository.list_results(conn, run_id, result_type)
|
|
except Exception as e:
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail=f"查询分析结果时发生错误: {str(e)}",
|
|
)
|