Пример #1
0
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)
Пример #2
0
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)
Пример #3
0
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,
			)
Пример #4
0
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)