Skip to content

Commit f5f765b

Browse files
Merge branch 'feat-pagination-performance'
2 parents 861c601 + 926832b commit f5f765b

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

graphene_mongo/fields.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,10 @@ def get_queryset(self, model, info, required_fields=list(), skip=None, limit=Non
202202
for arg_name, arg in args.copy().items():
203203
if arg_name in reference_fields and not isinstance(arg,
204204
mongoengine.base.metaclasses.TopLevelDocumentMetaclass):
205-
reference_obj = reference_fields[arg_name].document_type(pk=from_global_id(arg)[1])
205+
try:
206+
reference_obj = reference_fields[arg_name].document_type(pk=from_global_id(arg)[1])
207+
except TypeError:
208+
reference_obj = reference_fields[arg_name].document_type(pk=arg)
206209
hydrated_references[arg_name] = reference_obj
207210
args.update(hydrated_references)
208211

0 commit comments

Comments
 (0)