graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2025-02-08 23:50:38 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fixed format_error

Browse Source
This commit is contained in:
Syrus Akbary 2017-04-10 16:10:35 -07:00
parent b369ccfa08
commit 917dc16ea6
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
graphene/test/__init__.py
View File

@ -1,9 +1,17 @@
from graphene.types.schema import Schema from graphene.types.schema import Schema
from graphql.error import format_error from graphql.error import format_error as format_graphql_error
from graphql.error import GraphQLError
def format_execution_result(execution_result): def default_format_error(error):
if isinstance(error, GraphQLError):
return format_graphql_error(error)
return {'message': six.text_type(error)}
def format_execution_result(execution_result, format_error):
if execution_result: if execution_result:
response = {} response = {}
@ -17,12 +25,14 @@ def format_execution_result(execution_result):
class Client(object): class Client(object):
def __init__(self, schema, **execute_options): def __init__(self, schema, format_error=None, **execute_options):
assert isinstance(schema, Schema) assert isinstance(schema, Schema)
self.schema = schema self.schema = schema
self.execute_options = execute_options self.execute_options = execute_options
self.format_error = format_error or default_format_error
def execute(self, *args, **kwargs): def execute(self, *args, **kwargs):
return format_execution_result( return format_execution_result(
self.schema.execute(*args, **dict(self.execute_options, **kwargs)) self.schema.execute(*args, **dict(self.execute_options, **kwargs)),
self.format_error
) )