From d5ec3c6a31f0409731f11a3e2fefc4b9b7184a4c Mon Sep 17 00:00:00 2001 From: Jorge Bastida Date: Fri, 7 Dec 2012 15:59:09 +0000 Subject: [PATCH] Add as_pymongo to __getitem__ and in_bulk --- mongoengine/queryset.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/mongoengine/queryset.py b/mongoengine/queryset.py index be605713..d12f3bb4 100644 --- a/mongoengine/queryset.py +++ b/mongoengine/queryset.py @@ -988,6 +988,9 @@ class QuerySet(object): for doc in docs: doc_map[doc['_id']] = self._get_scalar( self._document._from_son(doc)) + elif self._as_pymongo: + for doc in docs: + doc_map[doc['_id']] = self._get_as_pymongo(doc) else: for doc in docs: doc_map[doc['_id']] = self._document._from_son(doc) @@ -1189,6 +1192,8 @@ class QuerySet(object): if self._scalar: return self._get_scalar(self._document._from_son( self._cursor[key])) + if self._as_pymongo: + return self._get_as_pymongo(self._cursor.next()) return self._document._from_son(self._cursor[key]) raise AttributeError