def send_message(room_id):
"""
Send message function
POST json format:
{"message": message}
Response 200 if success
if the room is not exist, return 400
if the is not online now, return 406
if the user is not in this room, return 406
"""
sender = g.user
print "sender is %s" % sender
if not online_users.is_online(sender):
return login_first()
room = ChatRoom.query.get(room_id)
if room is None:
return room_not_exist_by_id(room_id)
if not room_manager.is_in_room(room_id, sender.id):
return not_in_room_by_id(room_id)
info = get_json_post_data()
msg = info.get('message', None)
if msg is None:
return Response(status=406)
room_manager.send_message(sender.id, room_id, msg)
return Response("successful send message to room %s" % room.title, 200)
def update_message(room_id):
"""
Update message function, a long polling interface,
the user must ensure 'Keep-Alive' in 'Connection' field of
HTTP header
POST:
no data
Response json format:
{'messages':
[{'msg_id': msg_id,
'sender_id': sender_id,
'sender_name': sender_name,
'msg': msg,
'send_time': YYYY/MM/DD,HH:MM:SS}, ...]
}, the lastest 40 messages(if have enough messages)
if the room is not exist, return 400
if the 'Keep-Alive' is not in 'Connection' field, return 406
if the user is not in this room, return 406
"""
connection = request.headers.get('Connection', None)
if connection is None or connection.lower() != 'keep-alive':
return add_keep_alive()
room = ChatRoom.query.get(room_id)
if room is None:
return room_not_exist_by_id(room_id)
update_user = g.user
if not room_manager.is_in_room(room_id, update_user.id):
return not_in_room_by_id(room_id)
room_manager.update_message(room_id)
# construct the message json
msgs = room.records.order_by(desc(ChatRoomRecord.send_time)).all()
if len(msgs) > 40:
msgs = msgs[:40]
data = []
for m in msgs:
sender = User.query.get(m.sender_id)
if sender is not None:
data.append(
{'msg_id': m.id,
'sender_id': m.sender_id,
'sender_name': sender.username,
'msg': m.record,
'send_time': m.send_time.strftime("%Y/%m/%d,%H:%M:%S")}
)
return jsonify(messages=data)
def exit_room(room_id):
"""
Exit room function
POST:
no data
Response 200 if success
if the room is not exist, return 400
if the user is not online now, return 406
if the user is not in this room, return 406
"""
exit_user = g.user
if not online_users.is_online(exit_user):
return login_first()
room = ChatRoom.query.get(room_id)
if room is None:
return room_not_exist_by_id(room_id)
if not room_manager.is_in_room(room_id, exit_user.id):
return not_in_room_by_id(room_id)
room_manager.exit_room(room_id, exit_user.id)
# broadcast a message to all users in this room
room_manager.send_message(exit_user.id, room_id, "%s exit room!" % exit_user.username)
return Response("successful exit room %s" % room.title, 200)
