Allow node interface to be used in ConnectionField

This commit is contained in:
Markus Padourek 2016-06-07 10:07:49 +01:00
parent 631fa23249
commit 226b27fc4e

View File

@ -58,10 +58,10 @@ class ConnectionField(Field):
return edge_type.for_node(node)
def get_type(self, schema):
from graphene.relay.utils import is_node
from graphene.relay.utils import is_node, is_node_type
type = schema.T(self.type)
node = schema.objecttype(type)
assert is_node(node), 'Only nodes have connections.'
assert is_node(node) or is_node_type(node), 'Only nodes have connections.'
schema.register(node)
connection_type = self.get_connection_type(node)