Add extensions support to GraphQLError

Author: odero

graphql/error/base.py

@@ -6,7 +6,7 @@
     from ..language.source import Source
     from ..language.location import SourceLocation
     from types import TracebackType
-    from typing import Optional, List, Any, Union
+    from typing import Dict, Optional, List, Any, Union
 
 
 class GraphQLError(Exception):
@@ -19,6 +19,7 @@ class GraphQLError(Exception):
         "_positions",
         "_locations",
         "path",
+        "extensions",
     )
 
     def __init__(
@@ -30,6 +31,7 @@ def __init__(
         positions=None,  # type: Optional[Any]
         locations=None,  # type: Optional[Any]
         path=None,  # type: Union[List[Union[int, str]], List[str], None]
+        extensions=None,  # type: Optional[Dict[str, Any]]
     ):
         # type: (...) -> None
         super(GraphQLError, self).__init__(message)
@@ -40,6 +42,7 @@ def __init__(
         self._positions = positions
         self._locations = locations
         self.path = path
+        self.extensions = extensions
         return None
 
     @property

graphql/error/format_error.py

@@ -18,4 +18,7 @@ def format_error(error):
         if error.path is not None:
             formatted_error["path"] = error.path
 
+        if error.extensions is not None:
+            formatted_error["extensions"] = error.extensions
+
     return formatted_error

graphql/error/located_error.py

@@ -30,7 +30,10 @@ def __init__(
         if not stack:
             stack = sys.exc_info()[2]
 
+        extensions = (
+            getattr(original_error, "extensions", None) if original_error else None
+        )
         super(GraphQLLocatedError, self).__init__(
-            message=message, nodes=nodes, stack=stack, path=path
+            message=message, nodes=nodes, stack=stack, path=path, extensions=extensions
         )
         self.original_error = original_error