def post(self, user_id):
"""
Accept the friend request
Add the user into friend list
Remove the request from friend request
"""
args = friendsParser.parse_args()
profile_id = args['profile_id']
if profile_id is None:
abort(400)
success = Request.objects(user=user_id).only(
'requests_list').update_one(pull__requests_list=profile_id)
if success is 0:
abort(400)
success = Friend.objects(user=user_id).only(
'friends_list').update_one(add_to_set__friends_list=profile_id)
if success is 0:
friends = Friend(user=user_id, friends_list=[profile_id])
friends.save()
return {'status': 'success', 'message':
'The user has been added to your friend list'}
def post(self, user_id):
"""
Add a specific user to friend list, and
send a friend request to that user
"""
args = friendsParser.parse_args()
profile_id = args['profile_id']
if profile_id is None:
abort(400)
friend_profile = Profile.objects(id=profile_id).only('user').first()
if friend_profile is None:
abort(400)
# add the user to friend list
success = Friend.objects(user=user_id).only(
'friends_list').update_one(add_to_set__friends_list=profile_id)
if success is 0:
friends = Friend(user=user_id, friends_list=[profile_id])
friends.save()
# put the friend request to the user's request list
user_profile = Profile.objects(user=user_id).only('id').first()
success = Request.objects(user=friend_profile.user).update_one(
add_to_set__requests_list=user_profile)
if success is 0:
friend_request = Request(
user=friend_profile.user,
type='friends', requests_list=[user_profile])
friend_request.save()
return {'status': 'success', 'message':
'The user has been added to your friend list'}
def get(self, user_id):
"""
Get the user's friend list
"""
friends = Friend.objects(user=user_id).only('friends_list').first()
if friends is None:
return abort(400)
return friends_list_serialize(friends.friends_list)
def delete(self, user_id):
"""
Delete a specific user from the friend list
"""
args = friendsParser.parse_args()
profile_id = args['profile_id']
if profile_id is None:
abort(400)
success = Friend.objects(user=user_id).only(
'friends_list').update_one(pull__friends_list=profile_id)
if success is 0:
abort(400)
return {'status': 'success', 'message':
'The user has been delete from your friend list'}
