graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2025-03-05 04:15:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

Move NonNull check

Browse Source
This commit is contained in:
Jonathan Kim 2018-02-17 21:50:40 +00:00
parent 38baf7ab52
commit c25bcb3345
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
graphene/relay/connection.py
View File

@ -115,9 +115,6 @@ class IterableConnectionField(Field):
@classmethod @classmethod
def resolve_connection(cls, connection_type, args, resolved): def resolve_connection(cls, connection_type, args, resolved):
if isinstance(connection_type, NonNull):
connection_type = connection_type.of_type
if isinstance(resolved, connection_type): if isinstance(resolved, connection_type):
return resolved return resolved
@ -139,6 +136,9 @@ class IterableConnectionField(Field):
def connection_resolver(cls, resolver, connection_type, root, info, **args): def connection_resolver(cls, resolver, connection_type, root, info, **args):
resolved = resolver(root, info, **args) resolved = resolver(root, info, **args)
if isinstance(connection_type, NonNull):
connection_type = connection_type.of_type
on_resolve = partial(cls.resolve_connection, connection_type, args) on_resolve = partial(cls.resolve_connection, connection_type, args)
if is_thenable(resolved): if is_thenable(resolved):
return Promise.resolve(resolved).then(on_resolve) return Promise.resolve(resolved).then(on_resolve)