mirror of
https://github.com/graphql-python/graphene.git
synced 2024-11-26 03:23:55 +03:00
Use graphql-django-view
to handle GraphQLView
This commit is contained in:
parent
5c4db65cc0
commit
5a1e014b9a
|
@ -1,67 +1,10 @@
|
||||||
import json
|
from graphql_django_view import GraphQLView as BaseGraphQLView
|
||||||
|
|
||||||
from django.conf import settings
|
|
||||||
from django.http import HttpResponse
|
|
||||||
from django.views.generic import View
|
|
||||||
|
|
||||||
from graphql.core.error import GraphQLError, format_error
|
|
||||||
|
|
||||||
|
|
||||||
def form_error(error):
|
class GraphQLView(BaseGraphQLView):
|
||||||
if isinstance(error, GraphQLError):
|
def __init__(self, schema, **kwargs):
|
||||||
return format_error(error)
|
super(GraphQLView, self).__init__(
|
||||||
return error
|
schema=schema.schema,
|
||||||
|
executor=schema.executor,
|
||||||
|
**kwargs
|
||||||
class GraphQLView(View):
|
)
|
||||||
schema = None
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def format_result(result):
|
|
||||||
data = {'data': result.data}
|
|
||||||
if result.errors:
|
|
||||||
data['errors'] = list(map(form_error, result.errors))
|
|
||||||
|
|
||||||
return data
|
|
||||||
|
|
||||||
def response_errors(self, *errors):
|
|
||||||
errors = [{
|
|
||||||
"message": str(e)
|
|
||||||
} for e in errors]
|
|
||||||
return HttpResponse(json.dumps({'errors': errors}), content_type='application/json')
|
|
||||||
|
|
||||||
def execute_query(self, request, query, *args, **kwargs):
|
|
||||||
if not query:
|
|
||||||
return self.response_errors(Exception("Must provide query string."))
|
|
||||||
else:
|
|
||||||
try:
|
|
||||||
result = self.schema.execute(query, *args, **kwargs)
|
|
||||||
data = self.format_result(result)
|
|
||||||
except Exception as e:
|
|
||||||
if settings.DEBUG:
|
|
||||||
raise e
|
|
||||||
return self.response_errors(e)
|
|
||||||
return HttpResponse(json.dumps(data), content_type='application/json')
|
|
||||||
|
|
||||||
def get(self, request, *args, **kwargs):
|
|
||||||
query = request.GET.get('query')
|
|
||||||
return self.execute_query(request, query or '')
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def get_content_type(request):
|
|
||||||
meta = request.META
|
|
||||||
return meta.get('CONTENT_TYPE', meta.get('HTTP_CONTENT_TYPE', ''))
|
|
||||||
|
|
||||||
def post(self, request, *args, **kwargs):
|
|
||||||
content_type = self.get_content_type(request)
|
|
||||||
if content_type == 'application/json':
|
|
||||||
try:
|
|
||||||
received_json_data = json.loads(request.body.decode())
|
|
||||||
query = received_json_data.get('query')
|
|
||||||
except ValueError:
|
|
||||||
return self.response_errors(ValueError("Malformed json body in the post data"))
|
|
||||||
elif content_type == 'application/graphql':
|
|
||||||
query = request.body.decode()
|
|
||||||
else:
|
|
||||||
query = request.POST.get('query') or request.GET.get('query')
|
|
||||||
return self.execute_query(request, query or '')
|
|
||||||
|
|
3
setup.py
3
setup.py
|
@ -56,7 +56,8 @@ setup(
|
||||||
install_requires=[
|
install_requires=[
|
||||||
'six>=1.10.0',
|
'six>=1.10.0',
|
||||||
'blinker',
|
'blinker',
|
||||||
'graphql-core==0.4.7b0',
|
'graphql-core==0.4.7b2',
|
||||||
|
'graphql-django-view>=1.0.0',
|
||||||
'graphql-relay==0.3.3'
|
'graphql-relay==0.3.3'
|
||||||
],
|
],
|
||||||
tests_require=[
|
tests_require=[
|
||||||
|
|
|
@ -23,7 +23,8 @@ class Human(DjangoNode):
|
||||||
class Meta:
|
class Meta:
|
||||||
model = Article
|
model = Article
|
||||||
|
|
||||||
def resolve_raises(self, *args):
|
@staticmethod
|
||||||
|
def resolve_raises(*args):
|
||||||
raise Exception("This field should raise exception")
|
raise Exception("This field should raise exception")
|
||||||
|
|
||||||
def get_node(self, id):
|
def get_node(self, id):
|
||||||
|
|
|
@ -26,7 +26,7 @@ def test_client_post_malformed_json(settings, client):
|
||||||
response = client.post('/graphql', 'MALFORMED', 'application/json')
|
response = client.post('/graphql', 'MALFORMED', 'application/json')
|
||||||
json_response = format_response(response)
|
json_response = format_response(response)
|
||||||
assert json_response == {'errors': [
|
assert json_response == {'errors': [
|
||||||
{'message': 'Malformed json body in the post data'}]}
|
{'message': 'POST body sent invalid JSON.'}]}
|
||||||
|
|
||||||
|
|
||||||
def test_client_post_empty_query_json(settings, client):
|
def test_client_post_empty_query_json(settings, client):
|
||||||
|
|
Loading…
Reference in New Issue
Block a user