删除env.local;新增漏损区域识别功能
This commit is contained in:
@@ -0,0 +1,30 @@
|
||||
from typing import Any
|
||||
|
||||
from fastapi import APIRouter, HTTPException
|
||||
from pydantic import BaseModel
|
||||
|
||||
from app.services.leakage_identifier import run_leakage_identification
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
class LeakageIdentifyRequest(BaseModel):
|
||||
network: str
|
||||
observed_pressure_data: str | dict[str, list[Any]] | list[dict[str, Any]]
|
||||
start_time: float = 0
|
||||
duration: float = 24
|
||||
timestep: float = 5
|
||||
q_sum: float = 0.2
|
||||
q_sum_unit: str = "m3/s"
|
||||
output_dir: str = "Results"
|
||||
pop_size: int = 50
|
||||
max_gen: int = 100
|
||||
output_flow_unit: str = "m3/s"
|
||||
|
||||
|
||||
@router.post("/identify/")
|
||||
async def identify_leakage(data: LeakageIdentifyRequest) -> dict[str, Any]:
|
||||
try:
|
||||
return run_leakage_identification(**data.dict())
|
||||
except Exception as exc:
|
||||
raise HTTPException(status_code=400, detail=str(exc))
|
||||
Reference in New Issue
Block a user