def make_connections(users, task, connection_type):
if users:
if type(users) == list:
for user_id in users:
user = User.find_one(user_id)
if user.get("message") == "User not found!":
return {"message": f"User {user_id} not found"}
user_task_rel = Relationship(user, connection_type, task)
graph.create(user_task_rel)
return {}
else:
return {"message": "The users attributes must be lists!"}
else:
return {}
def find_all_projects_of_user(user_id):
user = User.find_one(user_id)
if user.get('message') == "User not found!":
return {"message": f"User {user_id} not found"}
projects = []
for rel in graph.match(start_node=user, rel_type="IS_USER"):
project = rel.end_node()
project['id'] = remote(project)._id
project['users'] = Project.find_users(project, "IS_USER")
project['admins'] = Project.find_users(project, "IS_ADMIN")
projects.append(project)
# for rel in graph.match(start_node=user, rel_type="IS_ADMIN"):
# project = rel.end_node()
# project['id'] = remote(project)._id
# if project not in projects:
# projects.append(project)
return projects
def get(current_user_id, user_id):
user = User.find_one(user_id)
return jsonify(user)