def delete_wall() -> str:
wall_id = request.args.get("wall_id")
form = WarrantyForm()
if form.validate_on_submit():
if form.no.data:
flash("Wall has not been deleted.")
next_page = request.args.get("next_page")
if not next_page:
next_page = url_for("masonry_works.walls")
return redirect(next_page)
elif form.yes.data:
Wall.delete_wall(wall_id)
flash("Wall has been deleted.")
return redirect(url_for("masonry_works.walls"))
return render_template(
"warranty_form.html",
title="Delete Wall",
form=form,
)
def test_delete_wall_when_no_wall(app_and_db):
assert not Wall.query.filter_by(id=1).first()
Wall.delete_wall(1)
def test_delete_wall(add_wall):
Wall.add_hole(wall_id=1, width=1, height=2, amonunt=1)
Wall.add_processing(wall_id=1, year=2020, month="December", done=0.5)
assert Wall.query.filter_by(id=1).first()
Wall.delete_wall(1)
assert not Wall.query.filter_by(id=1).first()
