Handle AuthorisationError at the top-level

Add an error handler to gracefully handle the custom
AuthorisationError at the application's top-level to avoid having to
manually handle it everywhere that the error (and its sub-classes)
might be raised.

1 files changed, 20 insertions, 0 deletions

diff --git a/gn_auth/errors.py b/gn_auth/errors.py
new file mode 100644
index 0000000..bd2c5eb
--- /dev/null
+++ b/gn_auth/errors.py
@@ -0,0 +1,20 @@
+"""Handle application level errors."""
+from flask import Flask, jsonify, current_app
+
+from gn_auth.auth.authorisation.errors import AuthorisationError
+
+def handle_authorisation_error(exc: AuthorisationError):
+    """Handle AuthorisationError if not handled anywhere else."""
+    current_app.logger.error(exc)
+    return jsonify({
+        "error": type(exc).__name__,
+        "error_description": " :: ".join(exc.args)
+    }), exc.error_code
+
+__error_handlers__ = {
+    AuthorisationError: handle_authorisation_error
+}
+def register_error_handlers(app: Flask):
+    """Register ALL defined error handlers"""
+    for klass, error_handler in __error_handlers__.items():
+        app.register_error_handler(klass, error_handler)