Skip to content

Commit 802bc24

Browse files
authored
Merge pull request #4 from sjhewitt/connection-get-query
make SQLAlchemyConnectionField.get_query a classmethod
2 parents 2bced89 + 04a648b commit 802bc24

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

graphene_sqlalchemy/fields.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,11 +15,15 @@ class SQLAlchemyConnectionField(ConnectionField):
1515
def model(self):
1616
return self.type._meta.node._meta.model
1717

18-
@staticmethod
19-
def connection_resolver(resolver, connection, model, root, args, context, info):
18+
@classmethod
19+
def get_query(cls, model, context, info, args):
20+
return get_query(model, context)
21+
22+
@classmethod
23+
def connection_resolver(cls, resolver, connection, model, root, args, context, info):
2024
iterable = resolver(root, args, context, info)
2125
if iterable is None:
22-
iterable = get_query(model, context)
26+
iterable = cls.get_query(model, context, info, args)
2327
if isinstance(iterable, Query):
2428
_len = iterable.count()
2529
else:

0 commit comments

Comments
 (0)