Skip to content

Commit 030e5e1

Browse files
authored
Merge pull request #36 from yfilali/master
Adding iterable and length to SQLAlchemyConnectionField
2 parents 2de70b4 + d17c284 commit 030e5e1

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

graphene_sqlalchemy/fields.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def connection_resolver(cls, resolver, connection, model, root, args, context, i
2828
_len = iterable.count()
2929
else:
3030
_len = len(iterable)
31-
return connection_from_list_slice(
31+
connection = connection_from_list_slice(
3232
iterable,
3333
args,
3434
slice_start=0,
@@ -38,6 +38,9 @@ def connection_resolver(cls, resolver, connection, model, root, args, context, i
3838
pageinfo_type=PageInfo,
3939
edge_type=connection.Edge,
4040
)
41+
connection.iterable = iterable
42+
connection.length = _len
43+
return connection
4144

4245
def get_resolver(self, parent_resolver):
4346
return partial(self.connection_resolver, parent_resolver, self.type, self.model)

0 commit comments

Comments
 (0)