diff --git a/examples/starwars_relay/schema.py b/examples/starwars_relay/schema.py index 8ad1a643..cdbf86c8 100644 --- a/examples/starwars_relay/schema.py +++ b/examples/starwars_relay/schema.py @@ -17,6 +17,11 @@ class Ship(graphene.ObjectType): return get_ship(id) +class ShipConnection(graphene.Connection): + class Meta: + node = Ship + + class Faction(graphene.ObjectType): '''A faction in the Star Wars saga''' @@ -24,7 +29,7 @@ class Faction(graphene.ObjectType): interfaces = (relay.Node, ) name = graphene.String(description='The name of the faction.') - ships = relay.ConnectionField(Ship, description='The ships used by the faction.') + ships = ShipConnection.Field(description='The ships used by the faction.') @resolve_only_args def resolve_ships(self, **args): diff --git a/graphene/relay/connection.py b/graphene/relay/connection.py index 46dbba98..e6d100a4 100644 --- a/graphene/relay/connection.py +++ b/graphene/relay/connection.py @@ -5,7 +5,7 @@ from functools import partial import six from graphql_relay import connection_from_list -from promise import Promise +from promise import Promise, is_thenable, promisify from ..types import (AbstractType, Boolean, Enum, Int, Interface, List, NonNull, Scalar, String, Union) @@ -90,65 +90,61 @@ class ConnectionMeta(ObjectTypeMeta): class Connection(six.with_metaclass(ConnectionMeta, ObjectType)): - pass + + @classmethod + def Field(cls, *args, **kwargs): # noqa: N802 + return ConnectionField(cls, *args, **kwargs) + + @classmethod + def is_type_of(cls, root, context, info): + return isinstance(root, cls) + + @classmethod + def connection_resolver(cls, resolved, args, context, info): + assert isinstance(resolved, Iterable), ( + 'Resolved value from the connection field have to be iterable or instance of {}. ' + 'Received "{}"' + ).format(cls, resolved) + connection = connection_from_list( + resolved, + args, + connection_type=cls, + edge_type=cls.Edge, + pageinfo_type=PageInfo + ) + connection.iterable = resolved + return connection -class IterableConnectionField(Field): +class ConnectionField(Field): def __init__(self, type, *args, **kwargs): kwargs.setdefault('before', String()) kwargs.setdefault('after', String()) kwargs.setdefault('first', Int()) kwargs.setdefault('last', Int()) - super(IterableConnectionField, self).__init__( + assert issubclass(type, Connection), ( + '{} type have to be a subclass of Connection. Received "{}".' + ).format(str(self), type) + super(ConnectionField, self).__init__( type, *args, **kwargs ) - @property - def type(self): - type = super(IterableConnectionField, self).type - if is_node(type): - connection_type = type.Connection - else: - connection_type = type - assert issubclass(connection_type, Connection), ( - '{} type have to be a subclass of Connection. Received "{}".' - ).format(str(self), connection_type) - return connection_type - - @classmethod - def resolve_connection(cls, connection_type, args, resolved): - if isinstance(resolved, connection_type): - return resolved - - assert isinstance(resolved, Iterable), ( - 'Resolved value from the connection field have to be iterable or instance of {}. ' - 'Received "{}"' - ).format(connection_type, resolved) - connection = connection_from_list( - resolved, - args, - connection_type=connection_type, - edge_type=connection_type.Edge, - pageinfo_type=PageInfo - ) - connection.iterable = resolved - return connection - @classmethod def connection_resolver(cls, resolver, connection_type, root, args, context, info): resolved = resolver(root, args, context, info) - on_resolve = partial(cls.resolve_connection, connection_type, args) - if isinstance(resolved, Promise): - return resolved.then(on_resolve) + if connection_type.is_type_of(resolved, context, info): + return resolved + + on_resolve = partial(connection_type.connection_resolver, args=args, context=context, info=info) + if is_thenable(resolved): + return promisify(resolved).then(on_resolve) return on_resolve(resolved) def get_resolver(self, parent_resolver): - resolver = super(IterableConnectionField, self).get_resolver(parent_resolver) + resolver = super(ConnectionField, self).get_resolver(parent_resolver) return partial(self.connection_resolver, resolver, self.type) - -ConnectionField = IterableConnectionField diff --git a/graphene/relay/node.py b/graphene/relay/node.py index acaee3e4..1657ec0f 100644 --- a/graphene/relay/node.py +++ b/graphene/relay/node.py @@ -21,18 +21,6 @@ def is_node(objecttype): return False -def get_default_connection(cls): - from .connection import Connection - assert issubclass(cls, ObjectType), ( - 'Can only get connection type on implemented Nodes.' - ) - - class Meta: - node = cls - - return type('{}Connection'.format(cls.__name__), (Connection,), {'Meta': Meta}) - - class GlobalID(Field): def __init__(self, node=None, required=True, *args, **kwargs): @@ -101,11 +89,3 @@ class Node(six.with_metaclass(NodeMeta, Interface)): @classmethod def to_global_id(cls, type, id): return to_global_id(type, id) - - @classmethod - def implements(cls, objecttype): - get_connection = getattr(objecttype, 'get_connection', None) - if not get_connection: - get_connection = partial(get_default_connection, objecttype) - - objecttype.Connection = get_connection() diff --git a/graphene/relay/tests/test_connection.py b/graphene/relay/tests/test_connection.py index 87f937ae..19b54ad1 100644 --- a/graphene/relay/tests/test_connection.py +++ b/graphene/relay/tests/test_connection.py @@ -97,7 +97,11 @@ def test_edge_with_bases(): def test_edge_on_node(): - Edge = MyObject.Connection.Edge + class MyObjectConnection(Connection): + class Meta: + node = MyObject + + Edge = MyObjectConnection.Edge assert Edge._meta.name == 'MyObjectEdge' edge_fields = Edge._meta.fields assert list(edge_fields.keys()) == ['node', 'cursor'] diff --git a/graphene/relay/tests/test_connection_query.py b/graphene/relay/tests/test_connection_query.py index 068081d6..72614a64 100644 --- a/graphene/relay/tests/test_connection_query.py +++ b/graphene/relay/tests/test_connection_query.py @@ -4,7 +4,7 @@ from graphql_relay.utils import base64 from promise import Promise from ...types import ObjectType, Schema, String -from ..connection import ConnectionField, PageInfo +from ..connection import ConnectionField, Connection, PageInfo from ..node import Node letter_chars = ['A', 'B', 'C', 'D', 'E'] @@ -18,10 +18,15 @@ class Letter(ObjectType): letter = String() +class LetterConnection(Connection): + class Meta: + node = Letter + + class Query(ObjectType): - letters = ConnectionField(Letter) - connection_letters = ConnectionField(Letter) - promise_letters = ConnectionField(Letter) + letters = LetterConnection.Field() + connection_letters = LetterConnection.Field() + promise_letters = LetterConnection.Field() node = Node.Field() @@ -32,13 +37,13 @@ class Query(ObjectType): return Promise.resolve(list(letters.values())) def resolve_connection_letters(self, args, context, info): - return Letter.Connection( + return LetterConnection( page_info=PageInfo( has_next_page=True, has_previous_page=False ), edges=[ - Letter.Connection.Edge( + LetterConnection.Edge( node=Letter(id=0, letter='A'), cursor='a-cursor' ), diff --git a/graphene/relay/tests/test_mutation.py b/graphene/relay/tests/test_mutation.py index 34fbb936..2b23977f 100644 --- a/graphene/relay/tests/test_mutation.py +++ b/graphene/relay/tests/test_mutation.py @@ -3,6 +3,7 @@ import pytest from ...types import (AbstractType, Argument, Field, InputField, InputObjectType, NonNull, ObjectType, Schema) from ...types.scalars import String +from ..connection import Connection from ..mutation import ClientIDMutation from ..node import Node @@ -19,6 +20,11 @@ class MyNode(ObjectType): name = String() +class MyNodeConnection(Connection): + class Meta: + node = MyNode + + class SaySomething(ClientIDMutation): class Input: @@ -37,13 +43,13 @@ class OtherMutation(ClientIDMutation): additional_field = String() name = String() - my_node_edge = Field(MyNode.Connection.Edge) + my_node_edge = Field(MyNodeConnection.Edge) @classmethod def mutate_and_get_payload(cls, args, context, info): shared = args.get('shared', '') additionalField = args.get('additionalField', '') - edge_type = MyNode.Connection.Edge + edge_type = MyNodeConnection.Edge return OtherMutation(name=shared + additionalField, my_node_edge=edge_type( cursor='1', node=MyNode(name='name'))) diff --git a/graphene/relay/tests/test_node.py b/graphene/relay/tests/test_node.py index 12c7c2c2..d174d8aa 100644 --- a/graphene/relay/tests/test_node.py +++ b/graphene/relay/tests/test_node.py @@ -52,15 +52,6 @@ def test_node_good(): assert 'id' in MyNode._meta.fields -def test_node_get_connection(): - connection = MyNode.Connection - assert issubclass(connection, Connection) - - -def test_node_get_connection_dont_duplicate(): - assert MyNode.Connection == MyNode.Connection - - def test_node_query(): executed = schema.execute( '{ node(id:"%s") { ... on MyNode { name } } }' % to_global_id("MyNode", 1) diff --git a/graphene/types/interface.py b/graphene/types/interface.py index cc8361e6..c4ee8959 100644 --- a/graphene/types/interface.py +++ b/graphene/types/interface.py @@ -56,7 +56,3 @@ class Interface(six.with_metaclass(InterfaceMeta)): def __init__(self, *args, **kwargs): raise Exception("An Interface cannot be intitialized") - - @classmethod - def implements(cls, objecttype): - pass diff --git a/graphene/types/objecttype.py b/graphene/types/objecttype.py index f06dbf5e..c7bcc0a5 100644 --- a/graphene/types/objecttype.py +++ b/graphene/types/objecttype.py @@ -45,10 +45,6 @@ class ObjectTypeMeta(AbstractTypeMeta): ) cls = type.__new__(cls, name, bases, dict(attrs, _meta=options)) - - for interface in options.interfaces: - interface.implements(cls) - return cls def __str__(cls): # noqa: N802