Skip to content

Commit cd8368b

Browse files
committed
If collection.objects is a partial function, call it first
1 parent e51ee40 commit cd8368b

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

mongoengine/dereference.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
from functools import partial
2+
13
from bson import SON, DBRef
24

35
from mongoengine.base import (
@@ -174,7 +176,10 @@ def _fetch_objects(self, doc_type=None):
174176
refs = [
175177
dbref for dbref in dbrefs if (col_name, dbref) not in object_map
176178
]
177-
references = collection.objects.in_bulk(refs)
179+
if isinstance(collection.objects, partial):
180+
references = collection.objects().in_bulk(refs)
181+
else:
182+
references = collection.objects.in_bulk(refs)
178183
for key, doc in references.items():
179184
object_map[(col_name, key)] = doc
180185
else: # Generic reference: use the refs data to convert to document

0 commit comments

Comments
 (0)