diff --git a/condition.py b/condition.py index 68c3823..9377688 100644 --- a/condition.py +++ b/condition.py @@ -3,6 +3,7 @@ from flask import request, render_template, redirect from flask_wtf import FlaskForm from __main__ import db, app, ma from sqlalchemy import func, Sequence +from sqlalchemy.exc import SQLAlchemyError from status import st, Status ################################################################################ @@ -50,10 +51,15 @@ def new_condition(): return render_template("edit_id_name.html", form=form, page_title='Create new Condition' ) else: condition = Condition( name=request.form["name"] ) - db.session.add(condition) - db.session.commit() - st.SetMessage( "Created new Condition (id={})".format(condition.id) ) - return redirect( '/conditions' ) + try: + db.session.add(condition) + db.session.commit() + st.SetMessage( "Created new Condition (id={})".format(condition.id) ) + return redirect( '/conditions' ) + except SQLAlchemyError as e: + st.SetAlert( "danger" ) + st.SetMessage( "Failed to add condition: {}".format( e.orig) ) + return render_template("edit_id_name.html", form=form, page_title='Create new Condition', alert=st.GetAlert(), message=st.GetMessage() ) ################################################################################ # /condition/ -> GET/POST(save or delete) -> shows/edits/delets a single condition @@ -63,15 +69,20 @@ def condition(id): ### DDP: should this be request.form or request.values? form = ConditionForm(request.form) if request.method == 'POST' and form.validate(): - condition = Condition.query.get(id) - if 'delete' in request.form: - st.SetMessage("Successfully deleted (id={}, name={})".format( condition.id, condition.name ) ) - condition = Condition.query.filter(Condition.id==id).delete() - if 'submit' in request.form: - st.SetMessage("Successfully Updated Condition (id={})".format(id) ) - condition.name = request.form['name'] - db.session.commit() - return redirect( '/conditions' ) + try: + condition = Condition.query.get(id) + if 'delete' in request.form: + st.SetMessage("Successfully deleted (id={}, name={})".format( condition.id, condition.name ) ) + condition = Condition.query.filter(Condition.id==id).delete() + if 'submit' in request.form: + st.SetMessage("Successfully Updated Condition (id={})".format(id) ) + condition.name = request.form['name'] + db.session.commit() + return redirect( '/conditions' ) + except SQLAlchemyError as e: + st.SetAlert( "danger" ) + st.SetMessage( "Failed to modity condition: {}".format(e.orig) ) + return render_template("edit_id_name.html", form=form, page_title='Edit Condition', alert=st.GetAlert(), message=st.GetMessage() ) else: condition = Condition.query.get(id) form = ConditionForm(request.values, obj=condition)