def add_item_to_drop_list(droplist_id):
"""Add an item to the drop list"""
droplist = DropList.query.get_or_404(droplist_id)
form = ItemForm()
form.set_choices(db, Location)
if form.validate_on_submit():
item = Item(row_letter=form.row_letter.data,
column_number=form.column_number.data,
location_id=form.location_id.data,
description=form.description.data,
droplist_id=droplist.id)
db.session.add(item)
db.session.commit()
return redirect(f"/droplists/{droplist.id}")
return render_template("/droplist_items_new.html", form=form)
def edit_droplist_item(droplist_id, item_id):
"""edit a droplist item"""
droplist = DropList.query.get_or_404(droplist_id)
item = Item.query.get_or_404(item_id)
if droplist.check_item(item) == False:
return render_template("404.html"), 404
form = ItemForm(obj=item)
form.set_choices(db, Location)
if form.validate_on_submit():
item.description = form.description.data
item.row_letter = form.row_letter.data
item.column_number = form.column_number.data
item.location_id = form.location_id.data
db.session.commit()
flash("Item successfully updated", "success")
return redirect(f"/droplists/{droplist_id}")
return render_template("/droplist_items_edit.html", form=form)
