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

Merge branch 'only_fields' of git://github.com/blackbrrr/mongoengine

Browse Source
This commit is contained in:
Harry Marr 2010-02-24 20:23:59 +00:00
commit 200e9eca92
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
mongoengine/queryset.py
View File

@ -120,6 +120,7 @@ class QuerySet(object):
self._accessed_collection = False
self._query = {}
self._where_clause = None
self._loaded_fields = []
# If inheritance is allowed, only return instances and instances of
# subclasses of the class being used
@ -222,7 +223,11 @@ class QuerySet(object):
@property
def _cursor(self):
if self._cursor_obj is None:
self._cursor_obj = self._collection.find(self._query)
cursor_args = {}
if self._loaded_fields:
cursor_args = {'fields': self._loaded_fields}
self._cursor_obj = self._collection.find(self._query,
**cursor_args)
# Apply where clauses to cursor
if self._where_clause:
self._cursor_obj.where(self._where_clause)
@ -444,6 +449,16 @@ class QuerySet(object):
elif isinstance(key, int):
return self._document._from_son(self._cursor[key])
def only(self, *fields):
"""Load only a subset of this document's fields. ::
post = BlogPost.objects(...).only("title")
:param *fields: fields to include
"""
self._loaded_fields = list(fields)
return self
def order_by(self, *keys):
"""Order the :class:`~mongoengine.queryset.QuerySet` by the keys. The
order may be specified by prepending each of the keys by a + or a -.