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

Add as_pymongo to __getitem__ and in_bulk

Browse Source
This commit is contained in:
Jorge Bastida 2012-12-07 15:59:09 +00:00
parent ad983dc279
commit d5ec3c6a31
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mongoengine/queryset.py
View File

@ -988,6 +988,9 @@ class QuerySet(object):
for doc in docs: for doc in docs:
doc_map[doc['_id']] = self._get_scalar( doc_map[doc['_id']] = self._get_scalar(
self._document._from_son(doc)) self._document._from_son(doc))
elif self._as_pymongo:
for doc in docs:
doc_map[doc['_id']] = self._get_as_pymongo(doc)
else: else:
for doc in docs: for doc in docs:
doc_map[doc['_id']] = self._document._from_son(doc) doc_map[doc['_id']] = self._document._from_son(doc)
@ -1189,6 +1192,8 @@ class QuerySet(object):
if self._scalar: if self._scalar:
return self._get_scalar(self._document._from_son( return self._get_scalar(self._document._from_son(
self._cursor[key])) self._cursor[key]))
if self._as_pymongo:
return self._get_as_pymongo(self._cursor.next())
return self._document._from_son(self._cursor[key]) return self._document._from_son(self._cursor[key])
raise AttributeError raise AttributeError