Fixed Doc.objects(read_preference=X) not setting read preference (#352)

2013-06-03 15:40:54 +00:00
parent ceece5a7e2
commit 4c8dfc3fc2
3 changed files with 12 additions and 4 deletions
--- a/mongoengine/queryset/queryset.py
+++ b/mongoengine/queryset/queryset.py
@@ -104,13 +104,17 @@ class QuerySet(object):
                raise InvalidQueryError(msg)
            query &= q_obj

-        queryset = self.clone()
+        if read_preference is None:
+            queryset = self.clone()
+        else:
+            # Use the clone provided when setting read_preference
+            queryset = self.read_preference(read_preference)
+
        queryset._query_obj &= query
        queryset._mongo_query = None
        queryset._cursor_obj = None
-        if read_preference is not None:
-            queryset.read_preference(read_preference)
        queryset._class_check = class_check
+
        return queryset

    def __len__(self):