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: