Refine

api/s41_pipe_risk_probability.py

@@ -42,13 +42,13 @@ def get_network_pipe_risk_probability_now(name: str) -> list[dict[str, Any]]:
 def get_pipes_risk_probability(name: str, pipe_ids: list[str]) -> list[dict[str, Any]]:
     pipe_risk_probability_list = []
     with conn[name].cursor(row_factory=dict_row) as cur:
-        cur.execute(f"select * from pipe_risk_probability where pipeid in ({','.join(pipe_ids)})")
+        cur.execute(f"select * from pipe_risk_probability")
         for record in cur:
-            #pipe_risk_probability_list.append(record)
-            t = {}
-            t['pipeid'] = record['pipeid']
-            t['x'] = record['y']
-            t['y'] = record['y']
-            pipe_risk_probability_list.append(t)
+            if record['pipeid'] in pipe_ids:
+                t = {}
+                t['pipeid'] = record['pipeid']
+                t['x'] = record['x']
+                t['y'] = record['y']
+                pipe_risk_probability_list.append(t)
 
     return pipe_risk_probability_list