def get_user_info(user_id): """ Get user info function GET: no data Response json format: {'username': username, 'email': email} If the user is not exist, return error code 400 """ user = User.query.get(user_id) if user is None: return user_not_exist_by_id(user_id) return jsonify(username=user.username, email=user.email)
def add_friend(user_id): """ Add friend function POST: no data Response 200 if success If the user is not exist, return error code 406 """ from_user = g.user if from_user is None: pass to_user = User.query.get(user_id) if to_user is None: return user_not_exist_by_id(user_id) # add friend from_user.add_friend(to_user) return Response("add friend %s" % to_user.username, status=200)
def launch_talk(talk_to_user_id): """ Launch a new talk function, for current logined user want to talk to user for _user_id POST: no data Response 200 if success If the user is not exist return error code 400 If the user is not both online, return 406 """ user = g.user if user is None: pass talk_to_user = User.query.get(talk_to_user_id) if talk_to_user is None: return user_not_exist_by_id(talk_to_user_id) if user.id not in online_users.get_online_users(): return Response("User %s is not online now" % user.username, 406) if talk_to_user_id not in online_users.get_online_users(): return Response("User %s is not online now" % talk_to_user.username, 406) talk_manager.new_talk(user.id, talk_to_user_id) return Response("launch talk to %s" % talk_to_user.username, 200)