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

Added support for multiple slices

Browse Source 
Also made slicing chainable. (#170) (#190) (#191)
This commit is contained in:
Ross Lawley
2012-12-19 14:56:15 +00:00
parent 3074dad293
commit 1c10f3020b
3 changed files with 50 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
mongoengine/queryset/field_list.py
View File

@@ -12,20 +12,31 @@ class QueryFieldList(object):
self.fields = set(fields)
self.always_include = set(always_include)
self._id = None
self.slice = {}
def __add__(self, f):
if not self.fields:
if isinstance(f.value, dict):
for field in f.fields:
self.slice[field] = f.value
if not self.fields:
self.fields = f.fields
elif not self.fields:
self.fields = f.fields
self.value = f.value
self.slice = {}
elif self.value is self.ONLY and f.value is self.ONLY:
self._clean_slice()
self.fields = self.fields.intersection(f.fields)
elif self.value is self.EXCLUDE and f.value is self.EXCLUDE:
self.fields = self.fields.union(f.fields)
self._clean_slice()
elif self.value is self.ONLY and f.value is self.EXCLUDE:
self.fields -= f.fields
self._clean_slice()
elif self.value is self.EXCLUDE and f.value is self.ONLY:
self.value = self.ONLY
self.fields = f.fields - self.fields
self._clean_slice()
if '_id' in f.fields:
self._id = f.value
@@ -42,10 +53,18 @@ class QueryFieldList(object):
def as_dict(self):
field_list = dict((field, self.value) for field in self.fields)
if self.slice:
field_list.update(self.slice)
if self._id is not None:
field_list['_id'] = self._id
return field_list
def reset(self):
self.fields = set([])
self.slice = {}
self.value = self.ONLY
def _clean_slice(self):
if self.slice:
for field in set(self.slice.keys()) - self.fields:
del self.slice[field]