def add_item():
"""
Route to add an item, you need to be logged in to
add an item.
Returns:
display page to create item for GET,
else create item and display home page
if POST
"""
form = CreateItemForm()
if form.validate_on_submit():
# user_info = user_info()
valid_category_ids = [c._id for c in Catalog.query.all()]
if int(form.category_id.data) not in valid_category_ids:
flash("Please enter a valid category id")
return redirect(url_for('catalog.add_item'))
item = Item(name=form.name.data,
description=form.description.data,
category_id=form.category_id.data,
owner=user_info()['email'])
db.session.add(item)
db.session.commit()
flash('Item added successfully')
return redirect(url_for('catalog.display_catalog'))
return render_template('create_item.html', form=form, user=user_info())
def add_category():
"""
Route to add a category, you need to be logged in to
add a cateory.
Returns:
display page to create category for GET,
else create category and display home page
if POST
"""
form = CreateCategoryForm()
if form.validate_on_submit():
# user_info = user_info()
category = Catalog(name=form.name.data, owner=user_info()['email'])
db.session.add(category)
db.session.commit()
flash('category added successfully')
return redirect(url_for('catalog.display_catalog'))
return render_template('create_category.html', form=form, user=user_info())
def display_item(item_id):
"""
Function to display the item details page
Args:
item_id: id of the item to display
Returns:
item jinja template with options to
edit / delete the item
"""
item = Item.query.get(item_id)
return render_template('item.html', item=item, user=user_info())
def display_category(category_id):
"""
Display the category details page
Args:
category_id: id of the category to display
Returns:
category jinja template with options to
edit / delete the category
"""
category = Catalog.query.get(category_id)
return render_template('category.html',
category=category,
user=user_info())
def display_catalog():
"""
Function to display the home page with categories
and items.
Returns:
home page jinja template that renders the
categories and items
"""
catalogs = Catalog.query.all()
items = Item.query.all()
return render_template('home.html',
catalogs=catalogs,
items=items,
user=user_info())
def delete_item(item_id):
"""
Route to delete item
Args:
item_id: id of the item to delete
Returns:
item delete page jinja template if GET
else delete item and show home page if POST
"""
item = Item.query.get(item_id)
if request.method == 'POST':
db.session.delete(item)
db.session.commit()
flash('item deletion successful')
return redirect(url_for('catalog.display_catalog'))
return render_template('delete_item.html',
item=item,
item_id=item_id,
user=user_info())
def delete_category(category_id):
"""
Delete category page
Args:
category_id: id of the category to delete
Returns:
category delete page template if GET
else delete category and display home
page if POST
"""
category = Catalog.query.get(category_id)
if request.method == 'POST':
db.session.delete(category)
db.session.commit()
flash('category deletion successful')
return redirect(url_for('catalog.display_catalog'))
return render_template('delete_category.html',
category=category,
category_id=category_id,
user=user_info())
def edit_category(category_id):
"""
Edit category page
Args:
category_id: id of the category to edit
Returns:
category edit page jinja template if GET
else edit category and show home page if POST
"""
category = Catalog.query.get(category_id)
form = EditCategoryForm()
if form.validate_on_submit():
category.name = form.name.data
# category.description = form.description.data
# category.category_id = form.category_id.data
db.session.add(category)
db.session.commit()
flash('Category edit successful')
return redirect(url_for('catalog.display_catalog'))
return render_template('edit_category.html', form=form, user=user_info())
def edit_item(item_id):
"""
Route to edit item
Args:
item_id: id of the item to edit
Returns:
item edit page jinja template if GET
else edit item and show home page if POST
"""
item = Item.query.get(item_id)
form = EditItemForm(obj=item)
if form.validate_on_submit():
item.name = form.name.data
item.description = form.description.data
item.category_id = form.category_id.data
db.session.add(item)
db.session.commit()
flash('Item edit successful')
return redirect(url_for('catalog.display_catalog'))
return render_template('edit_item.html', form=form, user=user_info())