TEMP UPDATE: for hide stacktrace. visit graphql-python/flask-graphql#49

graphql-python/flask-graphql#49
rscarrera27 · Jul 21, 2018 · 377a5b2 · 377a5b2
1 parent 266fd5a
commit 377a5b2
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/flask_graphql_auth/decorators.py b/flask_graphql_auth/decorators.py
@@ -88,7 +88,11 @@ def jwt_required(fn):
     @wraps(fn)
     def wrapper(*args, **kwargs):
         token = kwargs.pop(current_app.config['JWT_TOKEN_ARGUMENT_NAME'])
-        verify_jwt_in_argument(token)
+        try:
+            verify_jwt_in_argument(token)
+        except Exception as e:
+            return GraphQLError(e)
+
         return fn(*args, **kwargs)
     return wrapper
 
@@ -103,6 +107,10 @@ def jwt_refresh_token_required(fn):
     @wraps(fn)
     def wrapper(*args, **kwargs):
         token = kwargs.pop(current_app.config['JWT_TOKEN_ARGUMENT_NAME'])
-        verify_refresh_jwt_in_argument(token)
+        try:
+            verify_refresh_jwt_in_argument(token)
+        except Exception as e:
+            return GraphQLError(e)
+
         return fn(*args, **kwargs)
     return wrapper