move graphql error formatting to a function that can be overridden

encode · Aug 17, 2020 · 08e533b · 08e533b
1 parent c566fc6
commit 08e533b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/starlette/graphql.py b/starlette/graphql.py
@@ -102,9 +102,7 @@ async def handle_graphql(self, request: Request) -> Response:
             query, variables=variables, context=context, operation_name=operation_name
         )
         error_data = (
-            [format_graphql_error(err) for err in result.errors]
-            if result.errors
-            else None
+            [self.format_error(err) for err in result.errors] if result.errors else None
         )
         response_data = {"data": result.data}
         if error_data:
@@ -117,6 +115,9 @@ async def handle_graphql(self, request: Request) -> Response:
             response_data, status_code=status_code, background=background
         )
 
+    def format_error(self, error: Exception) -> dict:
+        return format_graphql_error(error)
+
     async def execute(  # type: ignore
         self, query, variables=None, context=None, operation_name=None
     ):