def edit_post():
form = CreatePost()
if request.method == 'POST':
post = Posts.query.filter_by(id=request.args['id']).first()
""" UPDATE DATA """
post.title = form.title.data
post.text = form.text.data
post.pub_date = form.date.data
""" Get image info """
if (form.image.data):
image = form.image.data
""" Create new file name """
old_filename, extension = os.path.splitext(image.filename)
filename = str(int(calendar.timegm(time.gmtime()))) + extension
""" Check directory """
directory = os.path.join(app.config['UPLOAD_FOLDER'], str(g.user.get_id()))
if not os.path.exists(directory):
os.makedirs(directory)
""" Save image """
image.save(os.path.join(directory, filename))
""" add new imagename """
post.img = filename
post.public = form.public.data
""" UPDATE DATA IN DB """
db.session.commit()
""" Succes message """
flash('Done')
return (redirect(url_for("index")))
elif request.method == 'GET':
""" Get info for post """
post = Posts.query.filter_by(id=request.args['id']).first()
if post is not None and post.user_id == g.user.get_id():
return render_template('edit.html',
post=post,
title='Update post',
form=form)
flash('Something is wrong')
return redirect(url_for("index"))
评论列表
文章目录