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

Fixed syntax using black

Browse Source
This commit is contained in:
Syrus 2020-07-26 22:07:24 -07:00
parent de81875fe7
commit c877a3e7be
3 changed files with 17 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
graphene/types/field.py
View File

@ -118,19 +118,21 @@ class Field(MountedType):
get_resolver = None get_resolver = None
def wrap_resolve(self, parent_resolver): def wrap_resolve(self, parent_resolver):
''' """
Wraps a function resolver, using the ObjectType resolve_{FIELD_NAME} Wraps a function resolver, using the ObjectType resolve_{FIELD_NAME}
(parent_resolver) if the Field definition has no resolver. (parent_resolver) if the Field definition has no resolver.
''' """
if self.get_resolver is not None: if self.get_resolver is not None:
warn_deprecation("The get_resolver method is being deprecated, please rename it to wrap_resolve.") warn_deprecation(
"The get_resolver method is being deprecated, please rename it to wrap_resolve."
)
return self.get_resolver(parent_resolver) return self.get_resolver(parent_resolver)
return self.resolver or parent_resolver return self.resolver or parent_resolver
def wrap_subscribe(self, parent_subscribe): def wrap_subscribe(self, parent_subscribe):
''' """
Wraps a function subscribe, using the ObjectType subscribe_{FIELD_NAME} Wraps a function subscribe, using the ObjectType subscribe_{FIELD_NAME}
(parent_subscribe) if the Field definition has no subscribe. (parent_subscribe) if the Field definition has no subscribe.
''' """
return parent_subscribe return parent_subscribe

14
graphene/types/schema.py
View File

@ -317,10 +317,7 @@ class TypeMap(dict):
) )
subscribe = field.wrap_subscribe( subscribe = field.wrap_subscribe(
self.get_function_for_type( self.get_function_for_type(
graphene_type, graphene_type, f"subscribe_{name}", name, field.default_value,
f"subscribe_{name}",
name,
field.default_value,
) )
) )
@ -333,14 +330,17 @@ class TypeMap(dict):
default_resolver = ( default_resolver = (
graphene_type._meta.default_resolver or get_default_resolver() graphene_type._meta.default_resolver or get_default_resolver()
) )
field_default_resolver = partial(default_resolver, name, field.default_value) field_default_resolver = partial(
default_resolver, name, field.default_value
)
else: else:
field_default_resolver = None field_default_resolver = None
resolve = field.wrap_resolve( resolve = field.wrap_resolve(
self.get_function_for_type( self.get_function_for_type(
graphene_type, f"resolve_{name}", name, field.default_value graphene_type, f"resolve_{name}", name, field.default_value
) or field_default_resolver )
or field_default_resolver
) )
_field = GraphQLField( _field = GraphQLField(
@ -356,7 +356,7 @@ class TypeMap(dict):
return fields return fields
def get_function_for_type(self, graphene_type, func_name, name, default_value): def get_function_for_type(self, graphene_type, func_name, name, default_value):
'''Gets a resolve or subscribe function for a given ObjectType''' """Gets a resolve or subscribe function for a given ObjectType"""
if not issubclass(graphene_type, ObjectType): if not issubclass(graphene_type, ObjectType):
return return
resolver = getattr(graphene_type, func_name, None) resolver = getattr(graphene_type, func_name, None)

4
graphene/types/tests/test_subscribe_async.py
View File

@ -51,4 +51,6 @@ async def test_subscription_fails_when_query_is_not_valid():
result = await schema.subscribe(subscription) result = await schema.subscribe(subscription)
assert not result.data assert not result.data
assert result.errors assert result.errors
assert "Anonymous Subscription must select only one top level field." in str(result.errors[0]) assert "Anonymous Subscription must select only one top level field." in str(
result.errors[0]
)