def add_type():
if request.method == 'GET':
return render_template('back/category_add.html')
if request.method == 'POST':
atype = request.form.get('atype')
if atype:
# 保存分类
art_type = ArticleType()
art_type.t_name = atype
art_type.save()
return redirect(url_for('back.a_type'))
else:
error = '请填写分类信息！'
return render_template('back/category_add.html', error=error)
def articles_category_add():
if request.method == 'GET':
return render_template('back/articles_category_add.html')
elif request.method == 'POST':
art_type = request.form.get('art_type')
if art_type:
articletype = ArticleType.query.filter(
ArticleType.at_name == art_type).first()
if articletype:
error = '该分类已存在'
return render_template('back/articles_category_add.html',
error=error)
else:
# 保存分类信息
articletype = ArticleType()
articletype.at_name = art_type
articletype.us = session['user_id']
articletype.save()
return redirect(url_for('back.articles_category'))
else:
error = '请填写分类信息'
return render_template('back/articles_category_add.html',
error=error)
