graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2025-07-30 17:59:49 +03:00
Code Issues Packages Projects Releases Wiki Activity

minor changes

Browse Source
This commit is contained in:
Kacppian 2018-07-28 15:30:08 +05:30
parent 2ce38844a2
commit 863bfcf6b4
1 changed files with 35 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
graphene/relay/connection.py
View File

@ -9,6 +9,7 @@ from ..types import Boolean, Enum, Int, Interface, List, NonNull, Scalar, String
from ..types.field import Field from ..types.field import Field
from ..types.objecttype import ObjectType, ObjectTypeOptions from ..types.objecttype import ObjectType, ObjectTypeOptions
from .node import is_node from .node import is_node
from ..utils.comparison_helper import raise_assertion_if
class PageInfo(ObjectType): class PageInfo(ObjectType):
@ -46,16 +47,17 @@ class Connection(ObjectType):
@classmethod @classmethod
def __init_subclass_with_meta__(cls, node=None, name=None, **options): def __init_subclass_with_meta__(cls, node=None, name=None, **options):
_meta = ConnectionOptions(cls) _meta = ConnectionOptions(cls)
if not node: error_message = "You have to provide a node in {}.Meta".format(cls.__name__)
raise AssertionError( raise_assertion_if(condition=not node, message=error_message)
"You have to provide a node in {}.Meta".format(cls.__name__)
)
if not issubclass(node, (Scalar, Enum, ObjectType, Interface, Union, NonNull)): error_message = 'Received incompatible node "{}" for Connection {}.'.format(
raise AssertionError(
'Received incompatible node "{}" for Connection {}.'.format(
node, cls.__name__ node, cls.__name__
) )
raise_assertion_if(
condition=not issubclass(
node, (Scalar, Enum, ObjectType, Interface, Union, NonNull)
),
error_message=error_message,
) )
base_name = re.sub("Connection$", "", name or cls.__name__) or node._meta.name base_name = re.sub("Connection$", "", name or cls.__name__) or node._meta.name
@ -106,17 +108,19 @@ class IterableConnectionField(Field):
if isinstance(type, NonNull): if isinstance(type, NonNull):
connection_type = type.of_type connection_type = type.of_type
if is_node(connection_type): error_message = """
raise Exception( ConnectionField's now need a explicit ConnectionType for Nodes.
"ConnectionField's now need a explicit ConnectionType for Nodes.\n" Read more: https://github.com/graphql-python/graphene/blob/v2.0.0/UPGRADE-v2.0.md#node-connections
"Read more: https://github.com/graphql-python/graphene/blob/v2.0.0/UPGRADE-v2.0.md#node-connections" """
raise_assertion_if(
condition=is_node(connection_type), error_message=error_message
) )
if not issubclass(connection_type, Connection): error_message = '{} type have to be a subclass of Connection. Received "{}".'.format(
raise AssertionError(
'{} type have to be a subclass of Connection. Received "{}".'.format(
self.__class__.__name__, connection_type self.__class__.__name__, connection_type
) )
raise_assertion_if(
condition=not issubclass(connection_type, Connection), message=error_message
) )
return type return type
@ -125,15 +129,16 @@ class IterableConnectionField(Field):
if isinstance(resolved, connection_type): if isinstance(resolved, connection_type):
return resolved return resolved
if not isinstance(resolved, Iterable): error_message = """
raise AssertionError(
"""
Resolved value from the connection field have to be iterable or instance of {}. Resolved value from the connection field have to be iterable or instance of {}.
Received "{}" Received "{}"
""".format( """.format(
connection_type, resolved connection_type, resolved
) )
raise_assertion_if(
condition=not isinstance(resolved, Iterable), message=error_message
) )
connection = connection_from_list( connection = connection_from_list(
resolved, resolved,
args, args,