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)