pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix read_preference for PyMongo 3+

Browse Source
This commit is contained in:
Stefan Wojcik 2015-07-05 15:59:50 -07:00
parent a7aead5138
commit 54975de0f3
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mongoengine/queryset/base.py
View File

@ -1444,6 +1444,14 @@ class BaseQuerySet(object):
def _cursor(self): def _cursor(self):
if self._cursor_obj is None: if self._cursor_obj is None:
# In PyMongo 3+, we define the read preference on a collection
# level, not a cursor level. Thus, we need to get a cloned
# collection object using `with_options` first.
if IS_PYMONGO_3 and self._read_preference is not None:
self._cursor_obj = self._collection\
.with_options(read_preference=self._read_preference)\
.find(self._query, **self._cursor_args)
else:
self._cursor_obj = self._collection.find(self._query, self._cursor_obj = self._collection.find(self._query,
**self._cursor_args) **self._cursor_args)
# Apply where clauses to cursor # Apply where clauses to cursor