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