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)
