add Flask error handlers

This commit is contained in:
dandds
2018-08-08 11:26:54 -04:00
parent 2cfc142417
commit 7b8934e0cb
7 changed files with 34 additions and 18 deletions

View File

@@ -1,4 +1,4 @@
from flask import Blueprint, render_template, g, redirect, session, url_for, request
from flask import Blueprint, abort, render_template, g, redirect, session, url_for, request
from flask import current_app as app
import pendulum
@@ -39,15 +39,7 @@ def login_redirect():
return redirect(url_for("atst.home"))
else:
return redirect(url_for("atst.unauthorized"))
@bp.route("/unauthorized")
def unauthorized():
template = render_template('unauthorized.html')
response = app.make_response(template)
response.status_code = 401
return response
return abort(401)
def _is_valid_certificate(request):

13
atst/routes/errors.py Normal file
View File

@@ -0,0 +1,13 @@
from flask import render_template
def make_error_pages(app):
@app.errorhandler(404)
def not_found(e):
return render_template("not_found.html"), 404
@app.errorhandler(401)
def unauthorized(e):
return render_template('unauthorized.html'), 401

View File

@@ -1,4 +1,4 @@
from flask import g, redirect, render_template, url_for, request as http_request
from flask import abort, g, redirect, render_template, url_for, request as http_request
from . import requests_bp
from atst.domain.requests import Requests
@@ -27,7 +27,7 @@ def requests_form_new(screen):
@requests_bp.route("/requests/new/<int:screen>/<string:request_id>", methods=["GET"])
def requests_form_update(screen=1, request_id=None):
if request_id and not _can_view_request(request_id):
return redirect(url_for("atst.unauthorized"))
abort(404)
request = Requests.get(request_id) if request_id is not None else None
jedi_flow = JEDIRequestFlow(screen, request, request_id=request_id)