def get(self, id=None):
if id:
try:
userFriend = UserFriend.select().where(UserFriend.user==id)
except UserFriend.DoesNotExist:
return Response(status=404)
else:
user_friendList = list()
for friend in userFriend:
user_friendList.append(friend.serialize)
return Response(json.dumps(user_friendList, indent=4), mimetype='application/json')
else:
return Response(status=404)
def put(self, id=None):
#block or accept friend
if id:
try:
resp = request.get_json()
fr = resp.get('friend')
userFriend = UserFriend.get(user=id,friend=fr)
except UserFriend.DoesNotExist:
return Response(status=404)
else:
resp = request.get_json()
accepted = resp.get('accepted') or ''
blocked = resp.get('blocked') or ''
userFriend.accepted = resp.get('accepted', userFriend.accepted)
userFriend.blocked = resp.get('blocked', userFriend.blocked)
userFriend.save()
return Response(json.dumps(userFriend.serialize), mimetype='application/json')
else:
return Response(status=404)
def post(self, id=None):
if id:
try:
User.get(id=id)
#hier prüfen ob der user den freund schon besitzt oder via app mit /<id>/friends/ ?
except User.DoesNotExist:
return Response(status=404)
else:
#add new user friend
resp = request.get_json()
friend = resp.get('friend')
accepted = resp.get('accepted') or False
blocked = resp.get('blocked') or False
friend = UserFriend.create(
user=id,
friend=friend,
accepted=accepted,
blocked=blocked,
)
return Response(json.dumps(friend.serialize), mimetype='application/json')
else:
return Response(status=404)
