def add_post(id): edit = True if id != None else False if edit and Post.query.get(id) == None: abort(404) if edit: post = Post.query.get(id) else: post = None if request.method == 'POST': if not edit: post = Post(request.form['title'], request.form['content'], []) if not (request.form['title'] and request.form['content']): flash('Both title and content need to be specified.') return render_template('add_post.html', post=post, edit=edit) else: if edit: post.title = request.form['title'] post.content = request.form['content'] db_session.commit() flash('Post edited.') else: db_session.add(post) db_session.commit() flash('New post submitted.') return redirect(url_for('display_posts')) return render_template('add_post.html', post=post, edit=edit)
def delete_post(id): post = Post.query.get(id) if request.method == 'GET': return render_template('delete_post.html', post=post) if request.form['confirm'] == 'yes': db_session.delete(post) db_session.commit() flash('Post deleted.') return redirect(url_for('display_posts')) return redirect(url_for('display_posts'))