From 7223f4b87630a0c700e61ae35cf35ba209047550 Mon Sep 17 00:00:00 2001 From: DingZQ Date: Sat, 19 Apr 2025 12:31:27 +0800 Subject: [PATCH] Refine --- api/s41_pipe_risk_probability.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/api/s41_pipe_risk_probability.py b/api/s41_pipe_risk_probability.py index e3ecd51..3257731 100644 --- a/api/s41_pipe_risk_probability.py +++ b/api/s41_pipe_risk_probability.py @@ -58,16 +58,19 @@ def get_pipe_risk_probability_geometries(name: str) -> dict[str, Any]: pipe_risk_probability_geometries = {} key_pipeId = '编码' - key_startnode = '上游节点' - key_endnode = '下游节点' + # key_startnode = '上游节点' + # key_endnode = '下游节点' key_geometry = 'geometry' with conn[name].cursor(row_factory=dict_row) as cur: - cur.execute(f"select {key_pipeId}, {key_startnode}, {key_endnode}, ST_AsGeoJSON(geometry) AS {key_geometry} from gis_pipe") + cur.execute(f"select {key_pipeId}, ST_AsGeoJSON(geometry) AS {key_geometry} from gis_pipe") for record in cur: - print(record[key_pipeId]) - print(record[key_startnode]) - print(record[key_endnode]) - print(record[key_geometry]) + id = record[key_pipeId] + geom = record[key_geometry] + + pipe_risk_probability_geometries[id] = { + 'start': geom['coordinates'][0], + 'end': geom['coordinates'][1] + } return pipe_risk_probability_geometries \ No newline at end of file