def add_category(id=0):
"""
add category
"""
category = Category.query.filter(Category.id == id).first()
category_form = CategoryForm(obj=category)
if request.method == 'POST' and category_form.validate():
if category:
category_form.populate_obj(category)
db_session.flush()
db_session.commit()
else:
category = Category(category_form.name.data)
category.content = category_form.content.data
db_session.add(category)
db_session.flush()
db_session.commit()
return redirect("/c/%s" % category.id)
else:
return render_template('/topic/add_category.html',
category_form=category_form)
def add_category(id=0):
"""
add category
"""
category = Category.query.filter(Category.id == id).first()
category_form = CategoryForm(obj=category)
if request.method == 'POST' and category_form.validate():
if category:
category_form.populate_obj(category)
db_session.flush()
db_session.commit()
else:
category = Category(category_form.name.data)
category.content = category_form.content.data
db_session.add(category)
db_session.flush()
db_session.commit()
return redirect("/c/%s" % category.id)
else:
return render_template('/topic/add_category.html',
category_form=category_form)
def add(id = 0):
topic = Topic.query.filter(Topic.id == id).first()
topic_form = TopicForm(obj = topic)
if request.method == 'POST' and topic_form.validate():
if topic:
if topic.category_id != topic_form.category_id.data:
new_category = Category.get(topic_form.category_id.data)
new_category.num += 1
old_category = Category.get(topic.category_id)
old_category.num -= 1
tag_list = topic_form.tag.data.split(" ")
for tag_name in tag_list:
if tag_name.strip():
tag = TopicTag.query.filter(TopicTag.name == tag_name).first()
if tag:
topic_to_tag = TopicToTag.query.filter(TopicToTag.topic_id == id).filter(TopicToTag.tag_id == tag.id).first()
if topic_to_tag is None:
tag.num += 1
topic_to_tag = TopicToTag(id, tag.id)
db_session.add(topic_to_tag)
else:
tag = TopicTag(tag_name)
db_session.add(tag)
db_session.flush()
topic_to_tag = TopicToTag(topic.id, tag.id)
db_session.add(topic_to_tag)
topic_form.populate_obj(topic)
db_session.commit()
else:
topic = Topic(topic_form.title.data)
topic.content = topic_form.content.data
topic.category_id = topic_form.category_id.data
topic.tag = topic_form.tag.data
db_session.add(topic)
db_session.flush()
category = Category.get(topic_form.category_id.data)
category.num += 1
tag_list = topic_form.tag.data.split(" ")
for tag_name in tag_list:
if tag_name.strip():
tag = TopicTag.query.filter(TopicTag.name == tag_name).first()
if tag:
tag.num += 1
else:
tag = TopicTag(tag_name)
db_session.add(tag)
db_session.flush()
topic_to_tag = TopicToTag(topic.id, tag.id)
db_session.add(topic_to_tag)
db_session.commit()
return redirect('/admin/topic/list')
else:
return render_template('/admin/addtopic.html',
topic_form = topic_form,
)
def topic_add(id=0):
"""
add topic
"""
topic = Topic.query.filter(Topic.id == id).first()
topic_form = TopicForm(obj=topic)
topic_form.category_id.choices = [
(c.id, c.name)
for c in Category.query.order_by(Category.id.asc()).all()
]
if request.method == 'POST' and topic_form.validate():
if topic:
if topic.category_id != topic_form.category_id.data:
new_category = Category.get(topic_form.category_id.data)
new_category.num += 1
old_category = Category.get(topic.category_id)
old_category.num -= 1
tag_list = topic_form.tag.data.split(" ")
for tag_name in tag_list:
if tag_name.strip():
tag = TopicTag.query.filter(
TopicTag.name == tag_name).first()
if tag:
topic_to_tag = TopicToTag.query.filter(
TopicToTag.topic_id == id).filter(
TopicToTag.tag_id == tag.id).first()
if topic_to_tag is None:
tag.num += 1
topic_to_tag = TopicToTag(id, tag.id)
db_session.add(topic_to_tag)
else:
tag = TopicTag(tag_name)
db_session.add(tag)
db_session.flush()
topic_to_tag = TopicToTag(topic.id, tag.id)
db_session.add(topic_to_tag)
topic_form.populate_obj(topic)
db_session.commit()
else:
topic = Topic(topic_form.title.data)
topic.content = topic_form.content.data
topic.category_id = topic_form.category_id.data
topic.tag = topic_form.tag.data
db_session.add(topic)
db_session.flush()
category = Category.get(topic_form.category_id.data)
category.num += 1
tag_list = topic_form.tag.data.split(" ")
for tag_name in tag_list:
if tag_name.strip():
tag = TopicTag.query.filter(
TopicTag.name == tag_name).first()
if tag:
tag.num += 1
else:
tag = TopicTag(tag_name)
db_session.add(tag)
db_session.flush()
topic_to_tag = TopicToTag(topic.id, tag.id)
db_session.add(topic_to_tag)
db_session.commit()
return redirect('/%s' % topic.id)
else:
return render_template('/topic/add.html', topic_form=topic_form)