neo4j_recommendations_fof.py

neo4j_recommendations_fof.py

def get_recommendations_fof(user_id, relationship = 'IS_FRIEND_WITH'):
    query = f"""
    MATCH (u:User)-[:{relationship}]->(u1:User)-[:{relationship}]->(u2:User)
    WHERE u.id = '{user_id}'
    RETURN collect(DISTINCT u2.id) AS recommendations
    """
    with driver.session() as session:
        result = session.run(query)

    return result.values()[0][0]
  
recommendations = get_recommendations_fof('user_id', 'FOLLOWS')