graphql-python/graphene-django
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene-django.git synced 2025-07-13 17:52:19 +03:00
Code Issues Packages Projects Releases Wiki Activity

fixed custom directives

Browse Source
This commit is contained in:
MardanovTimur 2019-03-22 22:51:19 +03:00
parent 85d45414f9
commit 87a98fc6f8
8 changed files with 127 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
examples/neomodel/__init__.py Normal file
View File

0
examples/neomodel/models.py Normal file
View File

0
examples/neomodel/nodes Normal file
View File

0
examples/neomodel/queries.py Normal file
View File

28
examples/neomodel/test.py Normal file
View File

@ -0,0 +1,28 @@
def find_brakes(self, recursion):
pass
REGEXES = [
r'((?P<recursive_points>\*+)(?P<model_name>\s?\w+\s*)(?P<body>(\{|.*|\n*|\})*))',
r'(\*node\s)*(\{(((?>[^{}]+)|(?R))*)\})',
]
query = """
query recursion_point {
module {
id
name
*module {
edges {
node {
id
}
}
}
}
}
"""
if __name__ == "__main__":

24
graphene_django/directives.py Normal file
View File

@ -0,0 +1,24 @@
from .middleware import BaseCustomDirective
from graphql.type.definition import GraphQLArgument, GraphQLNonNull
from graphql.type.scalars import GraphQLString
class DefaultDirective(BaseCustomDirective):
"""
Default to given value if None
"""
@staticmethod
def get_args():
return {
'to': GraphQLArgument(
type=GraphQLNonNull(GraphQLString),
description='Value to default to',
),
}
@staticmethod
def process(value, directive, root, info, **kwargs):
if value is None:
to_argument = [arg for arg in directive.arguments if arg.name.value == 'to'][0]
return to_argument.value.value
return value

68
graphene_django/middleware.py Normal file
View File

@ -0,0 +1,68 @@
import six
from graphql import DirectiveLocation, GraphQLDirective
from promise import Promise
class DirectivesMiddleware(object):
def resolve(self, next, root, info, **kwargs):
result = next(root, info, **kwargs)
return result.then(
lambda resolved: self.__process_value(resolved, root, info, **kwargs),
lambda error: Promise.rejected(error)
)
def __process_value(self, value, root, info, **kwargs):
field = info.field_asts[0]
# if field has not directive - return model-value
if not field.directives:
return value
new_value = value
# for each by field.directives
for directive in field.directives:
directive_class = CustomDirectiveMeta.REGISTRY.get(directive.name.value)
if directive_class:
# if directive class found
new_value = directive_class.process(new_value, directive, root, info, **kwargs)
return new_value
class CustomDirectiveMeta(type):
REGISTRY = {}
def __new__(mcs, name, bases, attrs):
newclass = super(CustomDirectiveMeta, mcs).__new__(mcs, name, bases, attrs)
if name != 'BaseCustomDirective':
mcs.register(newclass)
return newclass
@classmethod
def register(mcs, target):
mcs.REGISTRY[target.get_name()] = target
@classmethod
def get_all_directives(cls):
return [d() for d in cls.REGISTRY.values()]
class BaseCustomDirective(six.with_metaclass(CustomDirectiveMeta, GraphQLDirective)):
__metaclass__ = CustomDirectiveMeta
def __init__(self):
super(BaseCustomDirective, self).__init__(
name=self.get_name(),
description=self.__doc__,
args=self.get_args(),
locations=[
DirectiveLocation.FIELD
]
)
@classmethod
def get_name(cls):
return cls.__name__.replace('Directive', '').lower()
@staticmethod
def get_args():
return {}
from .directives import * # define the directives

7
graphene_django/views.py
View File

@ -17,6 +17,7 @@ from graphql.execution import ExecutionResult
from graphql.type.schema import GraphQLSchema
from .settings import graphene_settings
from .middleware import DirectivesMiddleware
class HttpError(Exception):
@ -84,6 +85,7 @@ class GraphQLView(View):
middleware = graphene_settings.MIDDLEWARE
self.schema = self.schema or schema
middleware = self.get_directive_middleware()
if middleware is not None:
self.middleware = list(instantiate_middleware(middleware))
self.executor = executor
@ -98,6 +100,11 @@ class GraphQLView(View):
), "A Schema is required to be provided to GraphQLView."
assert not all((graphiql, batch)), "Use either graphiql or batch processing"
def get_directive_middleware(self, custom=True):
return [DirectivesMiddleware, ]
# noinspection PyUnusedLocal
def get_root_value(self, request):
return self.root_value