def editor():
msg_id = request.args.get('msg_id')
if msg_id is not None:
result = Message.find_by_id(msg_id, session.get('user_id'))
print(result)
if len(result) == 1:
return render_template('editor.html',
msg_type=type_map,
msg=result[0][0],
msg_id=msg_id)
else:
abort(404)
else:
# edit new message
return render_template('editor.html',
msg_type=type_map,
msg=None,
msg_id=0)
def toggle_message():
if session.get('isLogin') != 'true':
return 'permission-denied'
msg_id = request.form.get('msg_id')
hide = bool(int(request.form.get('toggle')))
result = Message.find_by_id(msg_id, session.get('user_id'))
if len(result) != 1:
return 'invalid'
try:
if hide:
Message.hide_message(msg_id, session.get('user_id'))
else:
Message.show_message(msg_id, session.get('user_id'))
return 'success'
except IOError as e:
print(e)
return 'fail'
def get_msg(msg_id):
"""get message with message id message_id"""
try:
users = User.find_new(4)
result = Message.find_by_id(msg_id, session.get('user_id'))
if len(result) != 1:
# if length == 0, then message doesn't exist
# if length > 1, then the Database integrity is corrupted
abort(404)
except IOError as e:
print(e)
abort(500)
# increase message's read count by 1
Message.increase_read_count(msg_id)
result = result[0]
return render_template('message-detail.html',
msg=result[0],
users=users,
type_map=type_map,
nickname=result[1],
user_id=result[2])
