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])