def delete(self, id_or_name):
room = get_room(id_or_name)
if not g.user.is_master(room):
raise PermissionDenied
resp = room_schema(room)
os.remove('chatroom/static/avatars/room/{}.png'.format(room.id))
db.session.delete(room)
db.session.commit()
return make_resp(resp)
def delete(self):
user = g.user
if Room.query.filter_by(master_id=user.id).first() is not None:
return api_abort('400', 'you are the master of the room')
data = user_schema(user)
os.remove('chatroom/static/avatars/user/{}.png'.format(user.id))
db.session.delete(user)
db.session.commit()
return make_resp(data)
def get(self):
room_id = request.args.get('room', None)
if room_id is not None:
room = Room.query.get_or_404(room_id)
if room not in g.user.rooms:
raise PermissionDenied
users = room.users
else:
users = User.query.all()
return make_resp(users_schema(users, room_id))
def search():
q = request.args.get('q', None)
rid = request.args.get('rid', None)
if q is None or rid is None:
return api_abort(400, 'parm missing')
room = Room.query.get_or_404(rid)
if g.user not in room.users:
raise PermissionDenied
messages = Message.query.filter_by(room_id=rid).whooshee_search(q).all()
if messages is None:
return "None"
return make_resp(messages_schema(messages))
def post(self):
data = room_post_reqparse.parse_args()
if Room.query.filter_by(name=data['name']).first() is not None:
return api_abort(400, "room's name already exit")
room = Room.create_room(data['name'], data['introduce'], data['key'], data['topic'])
if 'avatar' in request.files:
f = request.files['avatar']
file = open('chatroom/static/avatars/room/{}.png'.format(room.id), 'wb')
file.close()
f.save('chatroom/static/avatars/room/{}.png'.format(room.id))
resp = make_resp(room_schema(room))
return resp
def put(self):
user = g.user
data = user_put_reqparse.parse_args()
if data['phone'] is not None:
user.phone = data['phone']
if data['email'] is not None:
user.phone = data['email']
if data['country'] is not None:
user.phone = data['country']
if data['area'] is not None:
user.phone = data['area']
if data['username'] is not None:
if user.username != data['username']:
check_name('user', data['username'])
user.username = data['username']
if data['action'] is not None:
action = data['action']
if action == 'join':
key = data['key']
name = data['name']
room = user.join_room(key, name)
if room is None:
raise InvalidAccessKey
elif action == 'leave':
room = Room.query.get(data['room_id'])
if user.is_master(room):
return api_abort(400, "the master can't leave the room")
if room not in user.rooms:
return api_abort(400, "user are not in room")
user.rooms.remove(room)
else:
return api_abort(400, "unknown parm")
db.session.commit()
return make_resp(room_schema(room))
db.session.commit()
return make_resp(user_schema(user))
def get(self):
# raise InvalidTokenError
# from werkzeug.exceptions import NotFound
raise NotFound
abort(404)
# the existed exceptions in werkzeug.exceptions can not be registered to a function
# 部署到服务器时的错误好像就跟这个一样？
if request.args.get('user', None) is not None: # 辅助请求其他用户信息
user = User.query.get_or_404(request.args.get('user'))
data = user_schema(user, False, False, True)
else:
user = g.user
data = user_schema(user)
return make_resp(data)
def post(self, rid_or_name):
room = get_room(rid_or_name)
if g.user not in room.users:
raise PermissionDenied
data = message_post_reqparse.parse_args()
new_message = g.user.send_message(data['type'], data['content'], room)
if data['type'] != 'text':
f = request.files['file']
if not allowed_file(f.filename):
return api_abort(400, 'invalid filename')
filename = secure_filename(f.filename[0:-1])
if new_message.content != filename:
new_message.content = filename
db.session.commit()
f.save('chatroom/static/{}/{}'.format(data['type'], str(new_message.id)+'_'+filename))
return make_resp(message_schema(new_message))
def put(self, id_or_name):
room = get_room(id_or_name)
if not g.user.is_master(room):
raise PermissionDenied
data = room_put_reqparse.parse_args()
if data['name'] is not None:
if data['name'] != room.name:
check_name('room', data['name'])
room.name = data['name']
if data['introduce'] is not None:
room.introduce = data['introduce']
if data['key'] is not None:
room.key = data['key']
if data['topic'] is not None:
room.topic = data['topic']
db.session.commit()
resp = room_schema(room)
return make_resp(resp)
def get(self, id_or_name):
room = get_room(id_or_name)
if g.user not in room.users:
raise PermissionDenied()
return make_resp(room_schema(room))
def get(self, rid_or_name):
room = get_room(rid_or_name)
messages = room.messages
if g.user not in room.users:
raise PermissionDenied
return make_resp(messages_schema(messages))
def get(self, mid):
message = Message.query.get_or_404(mid)
if message.room not in g.user.rooms:
raise PermissionDenied
return make_resp(message_schema(message))
