jiang
5966d039de
Reorganize algorithm packages by business responsibility, move orchestration into services, and keep database access behind pooled repositories. Harden analysis API validation, remove unsafe legacy simulation endpoints, and add regression and architecture boundary coverage. BREAKING CHANGE: legacy algorithm module paths and obsolete simulation endpoints are removed.
287 lines
8.8 KiB
Python
287 lines
8.8 KiB
Python
import logging
|
|
from typing import Any
|
|
from urllib.parse import quote
|
|
from uuid import UUID
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, Path, Query, status
|
|
from fastapi.responses import StreamingResponse
|
|
from starlette.concurrency import run_in_threadpool
|
|
|
|
from app.auth.metadata_dependencies import get_current_metadata_user
|
|
from app.auth.project_dependencies import (
|
|
ProjectContext,
|
|
get_project_context,
|
|
use_project_business_routing,
|
|
)
|
|
from app.infra.db.project_routing import ActiveProjectRouting
|
|
from app.domain.schemas.sensor_placement import (
|
|
SensorPointResponse,
|
|
SensorPlacementExportRequest,
|
|
SensorPlacementOptimizeRequest,
|
|
SensorPlacementSchemeResponse,
|
|
SensorPlacementUpdateRequest,
|
|
)
|
|
from app.services.sensor_placement import (
|
|
SensorPlacementConflictError,
|
|
SensorPlacementNotFoundError,
|
|
SensorPlacementValidationError,
|
|
build_sensor_placement_workbook,
|
|
can_edit_sensor_placement,
|
|
get_sensor_placement_candidate,
|
|
get_sensor_placement_run,
|
|
list_sensor_placement_runs,
|
|
optimize_sensor_placement_by_kmeans,
|
|
optimize_sensor_placement_by_sensitivity,
|
|
update_sensor_placement_run,
|
|
)
|
|
|
|
router = APIRouter()
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _project_network(network: str, project_context: ProjectContext) -> str:
|
|
if network != project_context.project_code:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail="请求的管网不属于当前项目",
|
|
)
|
|
return project_context.project_code
|
|
|
|
|
|
def _can_modify_project(project_context: ProjectContext) -> bool:
|
|
return project_context.project_role == "member"
|
|
|
|
|
|
def _require_project_write(
|
|
project_context: ProjectContext,
|
|
) -> None:
|
|
if not _can_modify_project(project_context):
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail="当前项目角色为只读，不能修改监测点方案",
|
|
)
|
|
|
|
|
|
def _service_http_error(exc: Exception) -> HTTPException:
|
|
if isinstance(exc, SensorPlacementNotFoundError):
|
|
return HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
detail=str(exc),
|
|
)
|
|
if isinstance(exc, SensorPlacementConflictError):
|
|
return HTTPException(
|
|
status_code=status.HTTP_409_CONFLICT,
|
|
detail=str(exc),
|
|
)
|
|
return HTTPException(
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
detail=str(exc),
|
|
)
|
|
|
|
|
|
def _get_run_response(
|
|
network: str,
|
|
run_id: UUID,
|
|
current_user: Any,
|
|
project_context: ProjectContext,
|
|
) -> dict[str, Any]:
|
|
try:
|
|
run = get_sensor_placement_run(network, run_id)
|
|
return {
|
|
**run,
|
|
"can_edit": (
|
|
_can_modify_project(project_context)
|
|
and can_edit_sensor_placement(current_user, run)
|
|
),
|
|
}
|
|
except (
|
|
SensorPlacementNotFoundError,
|
|
SensorPlacementValidationError,
|
|
) as exc:
|
|
raise _service_http_error(exc) from exc
|
|
|
|
|
|
@router.get(
|
|
"/sensor-placement-candidates/{node_id}",
|
|
response_model=SensorPointResponse,
|
|
summary="获取监测点候选节点详情",
|
|
)
|
|
async def get_sensor_placement_candidate_detail(
|
|
node_id: str = Path(..., min_length=1, max_length=32),
|
|
project_context: ProjectContext = Depends(get_project_context),
|
|
_routing: ActiveProjectRouting = Depends(use_project_business_routing),
|
|
) -> dict[str, Any]:
|
|
try:
|
|
return await run_in_threadpool(
|
|
get_sensor_placement_candidate,
|
|
project_context.project_code,
|
|
node_id,
|
|
)
|
|
except SensorPlacementValidationError as exc:
|
|
raise _service_http_error(exc) from exc
|
|
|
|
|
|
@router.post(
|
|
"/sensor-placement-runs",
|
|
response_model=SensorPlacementSchemeResponse,
|
|
summary="创建并返回监测点优化方案",
|
|
)
|
|
async def optimize_sensor_placement_scheme(
|
|
payload: SensorPlacementOptimizeRequest,
|
|
project_context: ProjectContext = Depends(get_project_context),
|
|
current_user=Depends(get_current_metadata_user),
|
|
) -> dict[str, Any]:
|
|
network = _project_network(payload.network, project_context)
|
|
_require_project_write(project_context)
|
|
optimizer = (
|
|
optimize_sensor_placement_by_sensitivity
|
|
if payload.method == "sensitivity"
|
|
else optimize_sensor_placement_by_kmeans
|
|
)
|
|
try:
|
|
created = await run_in_threadpool(
|
|
optimizer,
|
|
project_code=network,
|
|
run_name=payload.run_name,
|
|
sensor_count=payload.sensor_count,
|
|
min_diameter=payload.min_diameter,
|
|
created_by=current_user.username,
|
|
)
|
|
run = get_sensor_placement_run(network, created["run_id"])
|
|
return {**run, "can_edit": True}
|
|
except (
|
|
SensorPlacementConflictError,
|
|
SensorPlacementValidationError,
|
|
ValueError,
|
|
) as exc:
|
|
raise _service_http_error(exc) from exc
|
|
except Exception as exc:
|
|
logger.exception("Sensor placement optimization failed")
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail="监测点优化失败，请稍后重试",
|
|
) from exc
|
|
|
|
|
|
@router.get(
|
|
"/sensor-placement-runs",
|
|
response_model=list[SensorPlacementSchemeResponse],
|
|
summary="获取监测点优化运行",
|
|
)
|
|
async def get_sensor_placement_runs(
|
|
project_context: ProjectContext = Depends(get_project_context),
|
|
_routing: ActiveProjectRouting = Depends(use_project_business_routing),
|
|
) -> list[dict[str, Any]]:
|
|
return await run_in_threadpool(
|
|
list_sensor_placement_runs,
|
|
project_context.project_code,
|
|
)
|
|
|
|
|
|
@router.get(
|
|
"/sensor-placement-runs/{run_id}",
|
|
response_model=SensorPlacementSchemeResponse,
|
|
summary="获取监测点方案详情",
|
|
)
|
|
def get_sensor_placement_run_detail(
|
|
run_id: UUID,
|
|
network: str = Query(..., min_length=1),
|
|
project_context: ProjectContext = Depends(get_project_context),
|
|
current_user=Depends(get_current_metadata_user),
|
|
) -> dict[str, Any]:
|
|
return _get_run_response(
|
|
_project_network(network, project_context),
|
|
run_id,
|
|
current_user,
|
|
project_context,
|
|
)
|
|
|
|
|
|
@router.put(
|
|
"/sensor-placement-runs/{run_id}",
|
|
response_model=SensorPlacementSchemeResponse,
|
|
summary="覆盖保存监测点方案",
|
|
)
|
|
def overwrite_sensor_placement_run(
|
|
run_id: UUID,
|
|
payload: SensorPlacementUpdateRequest,
|
|
network: str = Query(..., min_length=1),
|
|
project_context: ProjectContext = Depends(get_project_context),
|
|
current_user=Depends(get_current_metadata_user),
|
|
) -> dict[str, Any]:
|
|
network = _project_network(network, project_context)
|
|
_require_project_write(project_context)
|
|
run = _get_run_response(
|
|
network,
|
|
run_id,
|
|
current_user,
|
|
project_context,
|
|
)
|
|
if not run["can_edit"]:
|
|
raise HTTPException(status_code=403, detail="无权修改该监测点优化运行")
|
|
|
|
try:
|
|
updated = update_sensor_placement_run(
|
|
network,
|
|
run_id,
|
|
expected_sensor_locations=payload.expected_sensor_locations,
|
|
sensor_locations=payload.sensor_locations,
|
|
)
|
|
return {**updated, "can_edit": True}
|
|
except (
|
|
SensorPlacementConflictError,
|
|
SensorPlacementNotFoundError,
|
|
SensorPlacementValidationError,
|
|
) as exc:
|
|
raise _service_http_error(exc) from exc
|
|
|
|
|
|
@router.post(
|
|
"/sensor-placement-runs/{run_id}/exports/excel",
|
|
summary="导出监测点工程清单",
|
|
)
|
|
async def export_sensor_placement_excel(
|
|
run_id: UUID,
|
|
payload: SensorPlacementExportRequest,
|
|
network: str = Query(..., min_length=1),
|
|
project_context: ProjectContext = Depends(get_project_context),
|
|
current_user=Depends(get_current_metadata_user),
|
|
) -> StreamingResponse:
|
|
network = _project_network(network, project_context)
|
|
run = _get_run_response(
|
|
network,
|
|
run_id,
|
|
current_user,
|
|
project_context,
|
|
)
|
|
if (
|
|
payload.sensor_locations != run["sensor_locations"]
|
|
and not run["can_edit"]
|
|
):
|
|
raise HTTPException(status_code=403, detail="无权导出该方案的未保存草稿")
|
|
|
|
try:
|
|
workbook = await run_in_threadpool(
|
|
build_sensor_placement_workbook,
|
|
network=network,
|
|
scheme=run,
|
|
sensor_location=payload.sensor_locations,
|
|
adjustment_status=payload.adjustment_status,
|
|
)
|
|
except SensorPlacementValidationError as exc:
|
|
raise _service_http_error(exc) from exc
|
|
|
|
filename = f"{run['name']}_监测点清单.xlsx"
|
|
encoded_filename = quote(filename)
|
|
return StreamingResponse(
|
|
workbook,
|
|
media_type=(
|
|
"application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
|
|
),
|
|
headers={
|
|
"Content-Disposition": (
|
|
f"attachment; filename*=UTF-8''{encoded_filename}"
|
|
)
|
|
},
|
|
)
|