Improved query execution using pluggable backend

Author: syrusakbary

graphql_server/__init__.py

@@ -2,11 +2,9 @@
 from collections import namedtuple, MutableMapping
 
 import six
-from graphql import Source, execute, parse, validate
-from graphql.error import format_error as format_graphql_error
-from graphql.error import GraphQLError
+from graphql import Source, execute, parse, validate, get_default_backend
+from graphql.error import format_error as default_format_error
 from graphql.execution import ExecutionResult
-from graphql.utils.get_operation_ast import get_operation_ast
 
 from .error import HttpQueryError
 
@@ -19,13 +17,6 @@ class SkipException(Exception):
 GraphQLResponse = namedtuple('GraphQLResponse', 'result,status_code')
 
 
-def default_format_error(error):
-    if isinstance(error, GraphQLError):
-        return format_graphql_error(error)
-
-    return {'message': six.text_type(error)}
-
-
 def run_http_query(schema, request_method, data, query_data=None, batch_enabled=False, catch=False, **execute_options):
     if request_method not in ('get', 'post'):
         raise HttpQueryError(
@@ -118,7 +109,7 @@ def load_json_variables(variables):
 def get_graphql_params(data, query_data):
     query = data.get('query') or query_data.get('query')
     variables = data.get('variables') or query_data.get('variables')
-    # id = data.get('id')
+    # document_id = data.get('documentId')
     operation_name = data.get('operationName') or query_data.get('operationName')
 
     return GraphQLParams(query, load_json_variables(variables), operation_name)
@@ -159,42 +150,34 @@ def format_execution_result(execution_result, format_error):
     return GraphQLResponse(response, status_code)
 
 
-def execute_graphql_request(schema, params, allow_only_query=False, **kwargs):
+def execute_graphql_request(schema, params, allow_only_query=False, backend=None, **kwargs):
     if not params.query:
         raise HttpQueryError(400, 'Must provide query string.')
 
     try:
-        source = Source(params.query, name='GraphQL request')
-        ast = parse(source)
-        validation_errors = validate(schema, ast)
-        if validation_errors:
-            return ExecutionResult(
-                errors=validation_errors,
-                invalid=True,
-            )
+        if not backend:
+            backend = get_default_backend()
+        document = backend.document_from_string(schema, params.query)
     except Exception as e:
         return ExecutionResult(errors=[e], invalid=True)
 
     if allow_only_query:
-        operation_ast = get_operation_ast(ast, params.operation_name)
-        if operation_ast and operation_ast.operation != 'query':
+        operation_type = document.get_operation_type(params.operation_name)
+        if operation_type and operation_type != 'query':
             raise HttpQueryError(
                 405,
-                'Can only perform a {} operation from a POST request.'.format(operation_ast.operation),
+                'Can only perform a {} operation from a POST request.'.format(operation_type),
                 headers={
                     'Allow': 'POST',
                 }
             )
 
     try:
-        return execute(
-            schema,
-            ast,
+        return document.execute(
             operation_name=params.operation_name,
-            variable_values=params.variables,
+            variables=params.variables,
             **kwargs
         )
-
     except Exception as e:
         return ExecutionResult(errors=[e], invalid=True)
 

setup.py

@@ -1,10 +1,13 @@
 from setuptools import setup, find_packages
 
-required_packages = ['graphql-core', 'promise']
+required_packages = [
+    'graphql-core>=2.1rc1',
+    'promise'
+]
 
 setup(
     name='graphql-server-core',
-    version='1.0.dev20170322001',
+    version='1.1rc0',
     description='GraphQL Server tools for powering your server',
     long_description=open('README.rst').read(),
     url='https://github.com/graphql-python/graphql-server-core',

tests/schema.py

@@ -12,15 +12,15 @@ def resolve_raises(*_):
     fields={
         'thrower': GraphQLField(GraphQLNonNull(GraphQLString), resolver=resolve_raises),
         'request': GraphQLField(GraphQLNonNull(GraphQLString),
-                                resolver=lambda obj, args, context, info: context.args.get('q')),
+                                resolver=lambda obj, info: context.args.get('q')),
         'context': GraphQLField(GraphQLNonNull(GraphQLString),
-                                resolver=lambda obj, args, context, info: context),
+                                resolver=lambda obj, info: context),
         'test': GraphQLField(
             type=GraphQLString,
             args={
                 'who': GraphQLArgument(GraphQLString)
             },
-            resolver=lambda obj, args, context, info: 'Hello %s' % (args.get('who') or 'World')
+            resolver=lambda obj, info, who='World': 'Hello %s' % who
         )
     }
 )

tox.ini

@@ -7,7 +7,7 @@ setenv =
     PYTHONPATH = {toxinidir}
 deps =
     pytest>=2.7.2
-    graphql-core>=1.0
+    graphql-core>=2.1rc1
     pytest-cov
 commands =
     py{py,27,33,34,35}: py.test tests {posargs}
@@ -22,6 +22,6 @@ commands =
 basepython=python3.5
 deps =
     isort
-    graphql-core>=1.0
+    graphql-core>=2.1rc1
 commands =
     isort --check-only graphql_server/ -rc