From c877a3e7be5740c69bd5e43b925f7ca7a15a6fed Mon Sep 17 00:00:00 2001 From: Syrus Date: Sun, 26 Jul 2020 22:07:24 -0700 Subject: [PATCH] Fixed syntax using black --- graphene/types/field.py | 12 +++++++----- graphene/types/schema.py | 14 +++++++------- graphene/types/tests/test_subscribe_async.py | 4 +++- 3 files changed, 17 insertions(+), 13 deletions(-) diff --git a/graphene/types/field.py b/graphene/types/field.py index 423fe821..dafb04b5 100644 --- a/graphene/types/field.py +++ b/graphene/types/field.py @@ -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 diff --git a/graphene/types/schema.py b/graphene/types/schema.py index e844d041..c767007e 100644 --- a/graphene/types/schema.py +++ b/graphene/types/schema.py @@ -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) diff --git a/graphene/types/tests/test_subscribe_async.py b/graphene/types/tests/test_subscribe_async.py index 40f8081b..6f7ce4c6 100644 --- a/graphene/types/tests/test_subscribe_async.py +++ b/graphene/types/tests/test_subscribe_async.py @@ -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] + )