def write(): if not current_user.is_administration(): abort(403) forms=PostForm() #查询到所有标签 categories=Category.query.order_by(Category.count.desc()) if forms.validate_on_submit(): #如果能查询到标签的话 tag = Category.query.filter_by(tag=forms.category.data).first() if tag is not None: post=Post(title=forms.title.data,body=forms.body.data, summary=forms.summary.data,category=tag) tag.count+=1 db.session.add(post) db.session.commit() else: category=Category(tag=forms.category.data,count=1) post=Post(title=forms.title.data,body=forms.body.data, summary=forms.summary.data,category=category) db.session.add(post) db.session.commit() flash('You have written an article') return redirect(url_for('main.index')) return render_template('write.html',form=forms,categories=categories)
def edit(id): if not current_user.is_administration(): abort(403) form=EditForm() post=Post.query.get_or_404(id) if form.validate_on_submit(): post.title=form.title.data post.body= form.body.data post.summary=form.summary.data db.session.add(post) db.session.commit() flash("The article has been updated!") return redirect(url_for('main.post',id=id)) form.title.data=post.title form.body.data=post.body form.summary.data=post.summary return render_template('edit.html',form=form)