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

better try/except handling for files

Browse Source
This commit is contained in:
Damien De Paoli
2021-01-13 13:54:47 +11:00
parent 2a7151dc66
commit c6bf0424ab
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
refimg.py
View File

@@ -55,13 +55,12 @@ def new_refimg():
if request.method == 'GET': if request.method == 'GET':
return render_template("refimg.html", form=form, page_title=page_title ) return render_template("refimg.html", form=form, page_title=page_title )
else: else:
# save the actual uploaded image to reference_images/
f=request.files['refimg_file']
f.save(os.path.join("reference_images/", request.form["fname"]))
# now save into the DB # now save into the DB
refimg = Refimg( fname=request.form["fname"] ) refimg = Refimg( fname=request.form["fname"] )
try: try:
# save the actual uploaded image to reference_images/
f=request.files['refimg_file']
f.save(os.path.join("reference_images/", request.form["fname"]))
db.session.add(refimg) db.session.add(refimg)
db.session.commit() db.session.commit()
st.SetMessage( "Created new Refimg ({})".format(refimg.fname) ) st.SetMessage( "Created new Refimg ({})".format(refimg.fname) )
@@ -69,7 +68,10 @@ def new_refimg():
except SQLAlchemyError as e: except SQLAlchemyError as e:
st.SetAlert( "danger" ) st.SetAlert( "danger" )
st.SetMessage( "<b>Failed to add Refimg:</b>&nbsp;{}".format(e.orig) ) st.SetMessage( "<b>Failed to add Refimg:</b>&nbsp;{}".format(e.orig) )
return render_template("refimg.html", form=form, page_title=page_title, alert=st.GetAlert(), message=st.GetMessage() ) except Exception as e:
st.SetAlert( "danger" )
st.SetMessage( "<b>Failed to modify Refimg:</b>&nbsp;{}".format(e) )
return render_template("refimg.html", form=form, page_title=page_title, alert=st.GetAlert(), message=st.GetMessage() )
################################################################################ ################################################################################
# /refimg/<id> -> GET/POST(save or delete) -> shows/edits/delets a single # /refimg/<id> -> GET/POST(save or delete) -> shows/edits/delets a single
@@ -98,7 +100,10 @@ def refimg(id):
except SQLAlchemyError as e: except SQLAlchemyError as e:
st.SetAlert( "danger" ) st.SetAlert( "danger" )
st.SetMessage( "<b>Failed to modify Refimg:</b>&nbsp;{}".format(e.orig) ) st.SetMessage( "<b>Failed to modify Refimg:</b>&nbsp;{}".format(e.orig) )
return render_template("refimg.html", form=form, page_title=page_title, alert=st.GetAlert(), message=st.GetMessage() ) except Exception as e:
st.SetAlert( "danger" )
st.SetMessage( "<b>Failed to modify Refimg:</b>&nbsp;{}".format(e) )
return render_template("refimg.html", form=form, page_title=page_title, alert=st.GetAlert(), message=st.GetMessage() )
else: else:
refimg = Refimg.query.get(id) refimg = Refimg.query.get(id)
form = RefimgForm(request.values, obj=refimg) form = RefimgForm(request.values, obj=refimg)