added error handling for DB exceptions
This commit is contained in:
11
condition.py
11
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"] )
|
||||
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( "<b>Failed to add condition:</b> {}".format( e.orig) )
|
||||
return render_template("edit_id_name.html", form=form, page_title='Create new Condition', alert=st.GetAlert(), message=st.GetMessage() )
|
||||
|
||||
################################################################################
|
||||
# /condition/<id> -> GET/POST(save or delete) -> shows/edits/delets a single condition
|
||||
@@ -63,6 +69,7 @@ def condition(id):
|
||||
### DDP: should this be request.form or request.values?
|
||||
form = ConditionForm(request.form)
|
||||
if request.method == 'POST' and form.validate():
|
||||
try:
|
||||
condition = Condition.query.get(id)
|
||||
if 'delete' in request.form:
|
||||
st.SetMessage("Successfully deleted (id={}, name={})".format( condition.id, condition.name ) )
|
||||
@@ -72,6 +79,10 @@ def condition(id):
|
||||
condition.name = request.form['name']
|
||||
db.session.commit()
|
||||
return redirect( '/conditions' )
|
||||
except SQLAlchemyError as e:
|
||||
st.SetAlert( "danger" )
|
||||
st.SetMessage( "<b>Failed to modity condition:</b> {}".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)
|
||||
|
||||
Reference in New Issue
Block a user