diff --git a/api/s41_pipe_risk_probability.py b/api/s41_pipe_risk_probability.py index f1857d0..2806bf0 100644 --- a/api/s41_pipe_risk_probability.py +++ b/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 \ No newline at end of file