add Flask error handlers
This commit is contained in:
@@ -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
13
atst/routes/errors.py
Normal 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
|
||||
|
@@ -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)
|
||||
|
Reference in New Issue
Block a user