From 2064e02ae8b034a2a3d4e3f15c8eb6067a06401a Mon Sep 17 00:00:00 2001 From: Damien De Paoli Date: Thu, 22 Jul 2021 19:49:09 +1000 Subject: [PATCH] clean up status message for when refimg removed from person --- person.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/person.py b/person.py index 76730ca..b3c438c 100644 --- a/person.py +++ b/person.py @@ -127,17 +127,21 @@ def person(id): db.session.commit() return redirect( f'/persons' ) elif request.form and form.validate(): - st.SetMessage("Successfully Updated Person: (From: {}, {}, {})".format(person.tag, person.firstname, person.surname) ) - person.tag = request.form['tag'] - person.surname = request.form['surname'] - person.firstname = request.form['firstname'] new_refs=[] for ref_img in person.refimg: if "ref-img-id-{}".format(ref_img.id) in request.form: new_refs.append(ref_img) - person.refimg = new_refs + if new_refs != person.refimg: + deld = list(set(person.refimg) - set(new_refs)) + st.SetMessage( f"Successfully Updated Person: removed reference image {deld[0].fname}" ) + person.refimg = new_refs + else: + st.SetMessage("Successfully Updated Person: (From: {}, {}, {})".format(person.tag, person.firstname, person.surname) ) + person.tag = request.form['tag'] + person.surname = request.form['surname'] + person.firstname = request.form['firstname'] + st.AppendMessage(" To: ({}, {}, {})".format(person.tag, person.firstname, person.surname) ) db.session.add(person) - st.AppendMessage(" To: ({}, {}, {})".format(person.tag, person.firstname, person.surname) ) db.session.commit() return redirect( f'/person/{person.id}' ) except SQLAlchemyError as e: