Fixed syntax using black

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

View File

@ -118,19 +118,21 @@ class Field(MountedType):
get_resolver = None
def wrap_resolve(self, parent_resolver):
'''
"""
Wraps a function resolver, using the ObjectType resolve_{FIELD_NAME}
(parent_resolver) if the Field definition has no resolver.
'''
"""
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.resolver or parent_resolver
def wrap_subscribe(self, parent_subscribe):
'''
"""
Wraps a function subscribe, using the ObjectType subscribe_{FIELD_NAME}
(parent_subscribe) if the Field definition has no subscribe.
'''
"""
return parent_subscribe

View File

@ -317,10 +317,7 @@ class TypeMap(dict):
)
subscribe = field.wrap_subscribe(
self.get_function_for_type(
graphene_type,
f"subscribe_{name}",
name,
field.default_value,
graphene_type, f"subscribe_{name}", name, field.default_value,
)
)
@ -333,14 +330,17 @@ class TypeMap(dict):
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:
field_default_resolver = None
resolve = field.wrap_resolve(
self.get_function_for_type(
graphene_type, f"resolve_{name}", name, field.default_value
) or field_default_resolver
)
or field_default_resolver
)
_field = GraphQLField(
@ -356,7 +356,7 @@ class TypeMap(dict):
return fields
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):
return
resolver = getattr(graphene_type, func_name, None)

View File

@ -51,4 +51,6 @@ async def test_subscription_fails_when_query_is_not_valid():
result = await schema.subscribe(subscription)
assert not result.data
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]
)