added error handling for DB exceptions

This commit is contained in:
2020-12-30 22:21:06 +11:00
parent f9e2060741
commit db18edeef3

View File

@@ -3,7 +3,7 @@ from flask import request, render_template, redirect
from flask_wtf import FlaskForm from flask_wtf import FlaskForm
from __main__ import db, app, ma from __main__ import db, app, ma
from sqlalchemy import func, Sequence from sqlalchemy import func, Sequence
from sqlalchemy.exc import SQLAlchemyError, IntegrityError from sqlalchemy.exc import SQLAlchemyError
from status import st, Status from status import st, Status
################################################################################ ################################################################################
@@ -54,7 +54,6 @@ def new_rating():
try: try:
db.session.add(rating) db.session.add(rating)
db.session.commit() db.session.commit()
st.SetAlert( "success" )
st.SetMessage( "Created new Rating (id={})".format(rating.id) ) st.SetMessage( "Created new Rating (id={})".format(rating.id) )
return redirect( '/ratings' ) return redirect( '/ratings' )
except SQLAlchemyError as e: except SQLAlchemyError as e:
@@ -78,7 +77,6 @@ def rating(id):
if 'submit' in request.form: if 'submit' in request.form:
st.SetMessage("Successfully Updated Rating (id={})".format(id) ) st.SetMessage("Successfully Updated Rating (id={})".format(id) )
rating.name = request.form['name'] rating.name = request.form['name']
db.session.flush()
db.session.commit() db.session.commit()
return redirect( '/ratings' ) return redirect( '/ratings' )
except SQLAlchemyError as e: except SQLAlchemyError as e: