Added iterable and length to the SQLAlchemyConnectionField

Author: yfilali

graphene_sqlalchemy/fields.py

@@ -1,16 +1,14 @@
 from functools import partial
 
-from sqlalchemy.orm.query import Query
-
 from graphene.relay import ConnectionField
 from graphene.relay.connection import PageInfo
 from graphql_relay.connection.arrayconnection import connection_from_list_slice
+from sqlalchemy.orm.query import Query
 
 from .utils import get_query
 
 
 class SQLAlchemyConnectionField(ConnectionField):
-
     @property
     def model(self):
         return self.type._meta.node._meta.model
@@ -28,7 +26,7 @@ def connection_resolver(cls, resolver, connection, model, root, args, context, i
             _len = iterable.count()
         else:
             _len = len(iterable)
-        return connection_from_list_slice(
+        connection = connection_from_list_slice(
             iterable,
             args,
             slice_start=0,
@@ -38,6 +36,9 @@ def connection_resolver(cls, resolver, connection, model, root, args, context, i
             pageinfo_type=PageInfo,
             edge_type=connection.Edge,
         )
+        connection.iterable = iterable
+        connection.length = _len
+        return connection
 
     def get_resolver(self, parent_resolver):
         return partial(self.connection_resolver, parent_resolver, self.type, self.model)