def send_message(recipient): user = User.query.filter_by(username=recipient).first_or_404() form = MessageForm() if form.validate_on_submit(): msg = Message(author=current_user, recipient=user, body=form.message.data) db.session.add(msg) user.add_notification('unread_message_count', user.new_messages()) db.session.commit() flash(_('Your message has been sent.')) return redirect(url_for('main.user', username=recipient)) return render_template('send_message.html', title=_('Send Message'), form=form, recipient=recipient)
def message(): form = MessageForm() if form.validate_on_submit(): title = form.title.data body = form.body.data message = Message(title=title, body=body, user_id=current_user.id) db.session.add(message) db.session.commit() flash('已成功发送', 'info') return redirect(url_for('message')) page = request.args.get('page', 1, type=int) per_page = app.config['MESSAGE_PER_PAGE'] pagination = Message.query.order_by(Message.timestamp.desc()).paginate( page, per_page=per_page) messages = pagination.items return render_template('message.html', pagination=pagination, messages=messages, bs=bs, form=form)
def test_message_form(self): form_data = {"message": "Test"} form = MessageForm(data=form_data) self.assertTrue(form.is_valid())