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)