def delete_post(post_id):
cache.delete('view/post/{}'.format(post_id))
post = Post.query.get_or_404(post_id)
db.session.delete(post)
db.session.commit()
flash('文章已删除.', 'success')
return redirect_back()
def change_theme(theme_name):
if theme_name not in current_app.config['BDWMS_BLOG_THEMES'].keys():
abort(404)
response = make_response(redirect_back())
response.set_cookie('theme', theme_name, max_age=30 * 24 * 60 * 60) # cookie过期时间为30天
return response
def set_comment(post_id):
cache.delete('view/post/{}'.format(post_id))
post = Post.query.get_or_404(post_id)
if post.can_comment:
post.can_comment = False
flash('评论已关闭', 'success')
else:
post.can_comment = True
flash('评论已打开', 'success')
db.session.commit()
return redirect_back()
def login():
if current_user.is_authenticated:
return redirect(url_for('blog.index'))
form = LoginForm() # 表单类继承的FlaskForm基类默认会从request.form获取表单数据， 所以不需要手动传入
if form.validate_on_submit():
username = form.username.data
password = form.password.data
remember = form.remember.data
admin = Admin.query.first()
if admin:
if username == admin.username and admin.validate_password(
password):
login_user(admin, remember) # 登入用户
flash('欢迎回来', 'info')
return redirect_back()
flash('无效的用户名或者密码', 'warning')
else:
flash('账户不存在', 'warning')
return render_template('auth/login.html', form=form)
def approve_comment(comment_id):
comment = Comment.query.get_or_404(comment_id)
comment.reviewed = True
db.session.commit()
flash('评论成功', 'success')
return redirect_back()
def delete_comment(comment_id):
comment = Comment.query.get_or_404(comment_id)
db.session.delete(comment)
db.session.commit()
flash('评论已删除', 'success')
return redirect_back()
def logout():
logout_user()
flash('成功退出', 'info')
return redirect_back()