Skip to content
Snippets Groups Projects
Commit 690ffe96 authored by Noah Jefferson Baumann's avatar Noah Jefferson Baumann
Browse files

adding all option for type filter

parent 09181843
No related branches found
No related tags found
No related merge requests found
......@@ -28,13 +28,13 @@ Session = scoped_session(session_factory)
def get_nodes():
try:
year = request.args.get('year')
node_type = request.args.get('type') # Get the type filter from the request
type_filter = request.args.get('type')
with Session() as session:
query = session.query(Node)
if year:
query = query.filter(Node.year == year)
if node_type:
query = query.filter(Node.type == node_type)
if type_filter and type_filter != 'All':
query = query.filter(Node.station_type == type_filter)
nodes = query.all()
node_list = [node.to_dict() for node in nodes]
return jsonify(node_list)
......@@ -45,13 +45,15 @@ def get_nodes():
def get_edges():
try:
year = request.args.get('year')
edge_type = request.args.get('type') # Get the type filter from the request
type_filter = request.args.get('type')
with Session() as session:
query = session.query(Edge)
if year:
query = query.filter(Edge.year == year)
if edge_type:
query = query.filter(Edge.edge_type == edge_type)
if type_filter and type_filter != 'All':
# Assuming Edge has a type field that needs filtering.
# Adjust according to your actual schema.
query = query.filter(Edge.edge_type == type_filter)
edges = query.all()
edge_list = [edge.to_dict() for edge in edges]
return jsonify(edge_list)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment