feat(investigations): refine report authoring and replay flushing

2026-11-14 23:00:15 +08:00
parent 2e4f6cb93b
commit b45c5c6f46
27 changed files with 1125 additions and 94 deletions
@@ -189,11 +189,12 @@ def saved_scenario(network: str, username: str, args: BaseModel) -> Any:
            modify_valve_opening={valve: 0.0 for valve in parameters["valves"]})
    if args.scenario == "flushing":
        from app.native.wndb.model.options import get_option
+        from app.mcp.scenario_service import flushing_valve_controls
        if get_option(network).get("UNITS") not in {"LPS", "CMH"}:
            raise ValueError("Flushing supports LPS or CMH models only")
        return scenarios.flushing_analysis(**common, username=username,
            drainage_node_ID=parameters["drainage_node_id"], flushing_flow=parameters["flushing_flow_m3h"],
-            modify_valve_opening={valve: 0.0 for valve in parameters["closed_valves"]})
+            valve_control=flushing_valve_controls(parameters))
    mappings = query_corresponding_element_id_and_query_id(network)
    controls = parameters["pump_control"]
    if set(controls) - (set(mappings.fixed_pumps) | set(mappings.variable_pumps)):