def new_post():
form = PostForm()
if form.validate_on_submit():
form.save(set_author=True)
flash('Added post.', 'success')
return form.redirect(url_for('admin.index'))
return render_template('admin/post_form.html', title='New post', form=form)
def edit_post(id, slug=None):
post = Post.query.get_or_404(id)
if slug is None:
return redirect(url_for('admin.edit_post', id=id, slug=post.slug))
form = PostForm(obj=post)
if form.validate_on_submit():
form.save()
flash('Edited post.', 'success')
return form.redirect(url_for('admin.index'))
return render_template('admin/post_form.html', title='Edit post', form=form)
