return error message if node not found

CityofToronto · Jan 2, 2025 · 046a296 · 046a296
1 parent 9c18eac
commit 046a296
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/backend/app/get_node.py b/backend/app/get_node.py
@@ -23,6 +23,8 @@ def get_node(node_id, conflate_with_centreline=False):
     with getConnection() as connection:
         with connection.cursor() as cursor:
             cursor.execute(SQL, {"node_id": node_id})
+            if cursor.rowcount != 1:
+                return None
             geojson, street_names = cursor.fetchone()
             node = {
                 'node_id': node_id,

diff --git a/backend/app/routes.py b/backend/app/routes.py
@@ -71,8 +71,8 @@ def node(node_id):
     doConflation = False
     if request.args.get('doConflation') is not None:
         doConflation = True
-
-    return jsonify(get_node(node_id, doConflation))
+    node = get_node(node_id, doConflation)
+    return jsonify(node if node else {'error': 'node not found'})
 
 # test URL /link-nodes/here/30421154/30421153
 #shell function - outputs json for use on frontend