This commit is contained in:
DingZQ
2025-04-17 23:09:15 +08:00
parent f675ec00de
commit ab99d9a81c
5 changed files with 37 additions and 15 deletions

View File

@@ -23,4 +23,32 @@ def get_pipe_risk_probability(name: str, pipe_id: str) -> dict[str, Any]:
d['x'] = t['x']
d['y'] = t['y']
return d
return d
def get_network_pipe_risk_probability_now(name: 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")
for record in cur:
#pipe_risk_probability_list.append(record)
t = {}
t['pipeid'] = record['pipeid']
t['pipeage'] = record['pipeage']
t['risk_probability_now'] = record['risk_probability_now']
pipe_risk_probability_list.append(t)
return pipe_risk_probability_list
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)})")
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)
return pipe_risk_probability_list