views.py 文件源码

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