70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
import os
|
|
from typing import Any
|
|
from datetime import datetime
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from pydantic import BaseModel
|
|
|
|
from app.auth.keycloak_dependencies import get_current_keycloak_username
|
|
from app.services.leakage_identifier import (
|
|
get_leakage_identify_scheme_detail,
|
|
list_leakage_identify_schemes,
|
|
run_leakage_identification,
|
|
)
|
|
|
|
router = APIRouter()
|
|
DEFAULT_N_WORKERS = max(1, min((os.cpu_count() or 1) - 1, 4))
|
|
|
|
|
|
class LeakageIdentifyRequest(BaseModel):
|
|
network: str
|
|
observed_pressure_data: str | dict[str, list[Any]] | list[dict[str, Any]] | None = (
|
|
None
|
|
)
|
|
start_time: float = 0
|
|
duration: float = 24
|
|
timestep: float = 5
|
|
q_sum: float = 0.2
|
|
q_sum_unit: str = "m3/s"
|
|
output_dir: str = "db_inp"
|
|
pop_size: int = 50
|
|
max_gen: int = 100
|
|
n_workers: int = DEFAULT_N_WORKERS
|
|
output_flow_unit: str = "m3/s"
|
|
dma_count: int | None = None
|
|
scada_start: datetime | None = None
|
|
scada_end: datetime | None = None
|
|
sensor_nodes: list[str] | None = None
|
|
scheme_name: str | None = None
|
|
|
|
|
|
@router.post("/identify/")
|
|
async def identify_leakage(
|
|
data: LeakageIdentifyRequest,
|
|
username: str = Depends(get_current_keycloak_username),
|
|
) -> dict[str, Any]:
|
|
try:
|
|
return run_leakage_identification(**data.model_dump(), username=username)
|
|
except Exception as exc:
|
|
raise HTTPException(status_code=400, detail=str(exc))
|
|
|
|
|
|
@router.get("/schemes/")
|
|
async def query_leakage_schemes(
|
|
network: str, query_date: datetime | None = None
|
|
) -> list[dict[str, Any]]:
|
|
try:
|
|
return list_leakage_identify_schemes(network=network, query_date=query_date)
|
|
except Exception as exc:
|
|
raise HTTPException(status_code=400, detail=str(exc))
|
|
|
|
|
|
@router.get("/schemes/{scheme_name}")
|
|
async def query_leakage_scheme_detail(network: str, scheme_name: str) -> dict[str, Any]:
|
|
try:
|
|
return get_leakage_identify_scheme_detail(
|
|
network=network, scheme_name=scheme_name
|
|
)
|
|
except Exception as exc:
|
|
raise HTTPException(status_code=400, detail=str(exc))
|