65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
# endpoints/list_members.py
|
|
from flask import Blueprint, request, jsonify
|
|
from neo4j import GraphDatabase
|
|
from app import get_driver, neo4j_logger
|
|
|
|
bp = Blueprint('list_members', __name__)
|
|
|
|
@bp.route('/list_members', methods=['GET'])
|
|
def list_members():
|
|
member_name = request.form.get('member_name')
|
|
|
|
if not member_name:
|
|
return jsonify({"error": "No member name provided"}), 400
|
|
|
|
driver = get_driver()
|
|
|
|
try:
|
|
with driver.session() as session:
|
|
result = session.run("""
|
|
MATCH (m:Person {name: $memberName})-[*]-(n)
|
|
RETURN m, n, collect(DISTINCT (m)-[r]->(n))
|
|
""", memberName=member_name)
|
|
|
|
data = {
|
|
"nodes": [],
|
|
"links": []
|
|
}
|
|
|
|
for record in result:
|
|
node1 = record["m"]
|
|
nodes = [record["n"]]
|
|
|
|
# Add the first node
|
|
if {"id": node1.id, "label": node1.labels[0], "name": node1.get("name")} not in data["nodes"]:
|
|
data["nodes"].append({"id": node1.id, "label": node1.labels[0], "name": node1.get("name")})
|
|
|
|
# Add other nodes and links
|
|
for node2 in nodes:
|
|
if {"id": node2.id, "label": node2.labels[0], "name": node2.get("name")} not in data["nodes"]:
|
|
data["nodes"].append({"id": node2.id, "label": node2.labels[0], "name": node2.get("name")})
|
|
|
|
for rel in record[2]:
|
|
if {"source": rel.start_node.id, "target": rel.end_node.id, "type": rel.type} not in data["links"]:
|
|
data["links"].append({"source": rel.start_node.id, "target": rel.end_node.id, "type": rel.type})
|
|
|
|
return jsonify(data)
|
|
|
|
except Exception as e:
|
|
neo4j_logger.error(f"Error querying Neo4j: {e}")
|
|
return jsonify({"error": str(e)}), 500
|
|
|
|
@bp.route('/get_member_names', methods=['GET'])
|
|
def get_member_names():
|
|
driver = get_driver()
|
|
|
|
try:
|
|
with driver.session() as session:
|
|
result = session.run("MATCH (p:Person) RETURN p.name AS name")
|
|
members = [record['name'] for record in result]
|
|
return jsonify(members)
|
|
|
|
except Exception as e:
|
|
neo4j_logger.error(f"Error fetching member names: {e}")
|
|
return jsonify({"error": str(e)}), 500
|