def post():
form = PostForm()
if form.validate_on_submit():
import pdb; pdb.set_trace()
image = request.files.get('image')
filename = None
try:
filename = uploaded_images.save(image)
except:
flash("The Image was not uploaded")
if form.new_category.data:
new_category = Category(form.new_category.data)
db.session.add(new_category)
db.session.flush()
category = new_category
else:
category = form.category.data
blog = Blog.query.first()
author = User.query.filter_by(username=session['username']).first()
title = form.title.data
body = form.body.data
slug = slugify(title)
post = Post(blog, author, title, body, category, slug, filename)
db.session.add(post)
db.session.commit()
return redirect(url_for('article', slug=slug))
return render_template('blog/post.html', form=form, action='new')
def edit(post_id):
post = Post.query.filter_by(id=post_id).first_or_404()
form = PostForm(obj=post)
if form.validate_on_submit():
original_image = post.image
form.populate_obj(post)
if form.image.has_file():
image = request.files.get('image')
try:
filename = uploaded_images.save(image)
except:
flash('The image was not uploaded!')
if filename:
post.image = filename
else:
post.image = original_image
if form.new_category.data:
new_category = Category(form.new_category.data)
db.session.add(new_category)
db.session.flush()
post.category = new_category
db.session.commit()
return redirect(url_for('article', slug=post.slug))
return render_template('blog/post.html', form=form, post=post, action='edit')
