def create_category(self):
"""Helper function creates category for tests"""
category = Category(user_id=1,
cat_name=self.test_category_name,
description=self.test_category_description)
category.add()
def do_add_category(user_id):
category = Category(
user_id=int(user_id),
cat_name=valid_data['category_name'],
description=valid_data['category_description'],
)
category.add()
return jsonify({
'id': category.cat_id,
'category_name': category.cat_name,
'category_description': category.cat_description,
'category_date': category.cat_date,
'message': 'category created'
}), 201
def category_add():
"""Register new category"""
if request.method == "POST":
# Define variables
name = request.form.get('name')
slug = request.form.get('slug')
if slug == "":
slug = slugify(name)
description = request.form.get('description')
category_exists = Category.get_item_by_slug(slug) is not None
if not category_exists:
# Add new item to database
item = Category.add(name, slug, description)
# Redirect to add category page
return redirect(url_for("category_add"))
else:
flash('Category already exists.')
# Render category-add.html and serve page
return render_template("category/category-add.html",
categories=Category.get_items())
