Add error handler

Author: alexisluque

01-Login/server.py

@@ -9,6 +9,7 @@
 from flask import request
 from flask import session
 from flask import redirect
+from flask import jsonify
 from flask_oauthlib.client import OAuth
 from functools import wraps
 from six.moves.urllib.parse import urlencode
@@ -30,6 +31,21 @@
 APP = Flask(__name__, static_url_path='/public', static_folder='./public')
 APP.secret_key = constants.SECRET_KEY
 APP.debug = True
+
+
+# Format error response and append status code.
+class AuthError(Exception):
+    def __init__(self, error, status_code):
+        self.error = error
+        self.status_code = status_code
+
+
+@APP.errorhandler(Exception)
+def handle_auth_error(ex):
+    response = jsonify(ex.error)
+    response.status_code = ex.status_code
+    return response
+
 oauth = OAuth(APP)
 
 
@@ -67,10 +83,7 @@ def home():
 def callback_handling():
     resp = auth0.authorized_response()
     if resp is None:
-        raise Exception('Access denied: reason=%s error=%s' % (
-            request.args['error_reason'],
-            request.args['error_description']
-        ))
+        raise AuthError({'code': request.args['error'], 'description': request.args['error_description']}, 401)
 
     # Obtain JWT and the keys to validate the signature
     idToken = resp['id_token']