def test_deletePictures(self, activeClient, testPicture):
name = testPicture.name
id = Picture.query.filter_by(name=name).first().id
testform = form(None, deletePictures([id]))
assert testPicture.is_file()
Picture.deletePictures(testform)
assert not testPicture.is_file()
def reviewEdit(id):
review = Review.query.get(id)
if request.method == 'GET':
session['referrer'] = request.referrer
form = ReviewEditForm(id=review.id,
name=review.provider.name,
category_id=review.category_id,
rating=review.rating,
cost=review.cost,
price_paid=review.price_paid,
description=review.description,
service_date=review.service_date,
comments=review.comments)
form.populate_choices(review)
return render_template("review_edit.html",
title="Review - Edit",
form=form,
review=review)
else:
form = ReviewEditForm()
form.populate_choices(review)
if form.validate_on_submit():
if form.deletePictures.data is not None:
try:
Picture.deletePictures(form)
except FileNotFoundError as e:
flash(str(e))
if form.picture.data:
new_pictures = Picture.savePictures(form, review.pictures)
try:
review.update(category_id=form.category.data,
rating=form.rating.data,
cost=form.cost.data,
price_paid=form.price_paid.data,
description=form.description.data,
service_date=form.service_date.data,
comments=form.comments.data,
pictures=new_pictures)
flash("Review updated.")
except SQLAlchemyError as e:
flash(str(e))
return render_template("review_edit.html",
title="Review - Edit",
form=form), 422
try:
url = session['referrer']
session.pop('referrer')
except KeyError:
url = url_for('main.user', username=current_user.username)
finally:
return redirect(url)
flash("Please correct form errors.")
return render_template("review_edit.html",
title="Review - Edit",
form=form), 422
