ddp/books
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

handle DB errors

Browse Source
This commit is contained in:
Damien De Paoli
2020-12-31 11:51:06 +11:00
parent 0ae4019a1a
commit 7e080f65c5
1 changed files with 33 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
loan.py
View File

@@ -4,6 +4,8 @@ from flask import request, render_template, redirect
from wtforms.fields.html5 import DateField from wtforms.fields.html5 import DateField
from __main__ import db, app, ma from __main__ import db, app, ma
from datetime import date from datetime import date
from sqlalchemy import Sequence
from sqlalchemy.exc import SQLAlchemyError
from status import st, Status from status import st, Status
################################################################################ ################################################################################
@@ -56,14 +58,21 @@ def loans():
@app.route("/loan", methods=["GET", "POST"]) @app.route("/loan", methods=["GET", "POST"])
def new_loan(): def new_loan():
form = LoanForm(request.form) form = LoanForm(request.form)
page_title='Create new Loan'
if 'firstnames' not in request.form: if 'firstnames' not in request.form:
return render_template("loan.html", form=form, page_title='Create new Loan' ) return render_template("loan.html", form=form, page_title=page_title )
else: else:
loan = Loan( firstnames=request.form["firstnames"], surname=request.form["surname"], date_lent=request.form["date_lent"], contact_details=request.form["contact_details"] ) loan = Loan( firstnames=request.form["firstnames"], surname=request.form["surname"], date_lent=request.form["date_lent"], contact_details=request.form["contact_details"] )
try:
db.session.add(loan) db.session.add(loan)
db.session.commit() db.session.commit()
st.SetMessage( "Created new Loan (id={})".format(loan.id) ) st.SetMessage( "Created new Loan (id={})".format(loan.id) )
return redirect( '/loans' ) return redirect( '/loans' )
except SQLAlchemyError as e:
st.SetAlert( "danger" )
st.SetMessage( "<b>Failed to add Loan:</b>&nbsp;{}".format( e.orig) )
return render_template("edit_id_name.html", form=form, page_title=page_title, alert=st.GetAlert(), message=st.GetMessage() )
################################################################################ ################################################################################
# /loan/<id> -> GET/POST(save or delete) -> shows/edits/delets a single loan # /loan/<id> -> GET/POST(save or delete) -> shows/edits/delets a single loan
@@ -72,8 +81,10 @@ def new_loan():
def loan(id): def loan(id):
### DDP: should this be request.form or request.values? ### DDP: should this be request.form or request.values?
form = LoanForm(request.form) form = LoanForm(request.form)
page_title='Edit Loan'
if request.method == 'POST' and form.validate(): if request.method == 'POST' and form.validate():
loan = Loan.query.get(id) loan = Loan.query.get(id)
try:
if 'delete' in request.form: if 'delete' in request.form:
st.SetMessage("Successfully deleted (id={}, who={} {})".format( loan.id, loan.firstnames, loan.surname ) ) st.SetMessage("Successfully deleted (id={}, who={} {})".format( loan.id, loan.firstnames, loan.surname ) )
loan = Loan.query.filter(Loan.id==id).delete() loan = Loan.query.filter(Loan.id==id).delete()
@@ -86,7 +97,11 @@ def loan(id):
loan.date_lent = request.form['date_lent'] loan.date_lent = request.form['date_lent']
db.session.commit() db.session.commit()
return redirect( '/loans' ) return redirect( '/loans' )
except SQLAlchemyError as e:
st.SetAlert( "danger" )
st.SetMessage( "<b>Failed to modify Loan:</b>&nbsp;{}".format(e.orig) )
return render_template("edit_id_name.html", form=form, page_title=page_title, alert=st.GetAlert(), message=st.GetMessage() )
else: else:
loan = Loan.query.get(id) loan = Loan.query.get(id)
form = LoanForm(request.values, obj=loan) form = LoanForm(request.values, obj=loan)
return render_template("loan.html", loan=loan, form=form, page_title='Edit Loan' ) return render_template("loan.html", loan=loan, form=form, page_title=page_title)