82 lines
2.7 KiB
Python
82 lines
2.7 KiB
Python
from flask import render_template, current_app, url_for, redirect, request
|
|
from flask_wtf.csrf import CSRFError
|
|
import werkzeug.exceptions as werkzeug_exceptions
|
|
|
|
import atst.domain.exceptions as exceptions
|
|
from atst.domain.invitations import (
|
|
InvitationError,
|
|
ExpiredError as InvitationExpiredError,
|
|
WrongUserError as InvitationWrongUserError,
|
|
)
|
|
from atst.domain.authnid.crl import CRLInvalidException
|
|
from atst.domain.portfolios import PortfolioError
|
|
from atst.utils.flash import formatted_flash as flash
|
|
|
|
|
|
def log_error(e):
|
|
error_message = e.message if hasattr(e, "message") else str(e)
|
|
current_app.logger.exception(error_message)
|
|
|
|
|
|
def handle_error(e, message="Not Found", code=404):
|
|
log_error(e)
|
|
current_app.notification_sender.send(message)
|
|
return render_template("error.html", message=message), code
|
|
|
|
|
|
def make_error_pages(app):
|
|
@app.errorhandler(werkzeug_exceptions.NotFound)
|
|
@app.errorhandler(exceptions.NotFoundError)
|
|
@app.errorhandler(exceptions.UnauthorizedError)
|
|
@app.errorhandler(PortfolioError)
|
|
@app.errorhandler(exceptions.NoAccessError)
|
|
# pylint: disable=unused-variable
|
|
def not_found(e):
|
|
return handle_error(e)
|
|
|
|
@app.errorhandler(CRLInvalidException)
|
|
# pylint: disable=unused-variable
|
|
def missing_crl(e):
|
|
return handle_error(e, message="Error Code 008", code=401)
|
|
|
|
@app.errorhandler(exceptions.UnauthenticatedError)
|
|
# pylint: disable=unused-variable
|
|
def unauthorized(e):
|
|
return handle_error(e, message="Log in Failed", code=401)
|
|
|
|
@app.errorhandler(CSRFError)
|
|
# pylint: disable=unused-variable
|
|
def session_expired(e):
|
|
log_error(e)
|
|
url_args = {"next": request.path}
|
|
flash("session_expired")
|
|
if request.method == "POST":
|
|
url_args[app.form_cache.PARAM_NAME] = app.form_cache.write(request.form)
|
|
return redirect(url_for("atst.root", **url_args))
|
|
|
|
@app.errorhandler(Exception)
|
|
# pylint: disable=unused-variable
|
|
def exception(e):
|
|
log_error(e)
|
|
if current_app.debug:
|
|
raise e
|
|
return (
|
|
render_template("error.html", message="An Unexpected Error Occurred"),
|
|
500,
|
|
)
|
|
|
|
@app.errorhandler(InvitationError)
|
|
@app.errorhandler(InvitationWrongUserError)
|
|
# pylint: disable=unused-variable
|
|
def invalid_invitation(e):
|
|
return handle_error(e, message="The link you followed is invalid.", code=404)
|
|
|
|
@app.errorhandler(InvitationExpiredError)
|
|
# pylint: disable=unused-variable
|
|
def invalid_invitation(e):
|
|
return handle_error(
|
|
e, message="The invitation you followed has expired.", code=404
|
|
)
|
|
|
|
return app
|