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

Merge pull request #1765 from kushalmitruka/master

Browse Source 
fixed : `pull` not working for EmbeddedDocumentListField, only working for ListFields #1534
This commit is contained in:
erdenezul
2018-05-03 09:44:54 +08:00
committed by GitHub
parent 2121387aa2 b1f62a2735
commit 317e844886
2 changed files with 36 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
mongoengine/queryset/transform.py
View File

@@ -314,11 +314,17 @@ def update(_doc_cls=None, **update):
field_classes = [c.__class__ for c in cleaned_fields]
field_classes.reverse()
ListField = _import_class('ListField')
if ListField in field_classes:
# Join all fields via dot notation to the last ListField
EmbeddedDocumentListField = _import_class('EmbeddedDocumentListField')
if ListField in field_classes or EmbeddedDocumentListField in field_classes:
# Join all fields via dot notation to the last ListField or EmbeddedDocumentListField
# Then process as normal
if ListField in field_classes:
_check_field = ListField
else:
_check_field = EmbeddedDocumentListField
last_listField = len(
cleaned_fields) - field_classes.index(ListField)
cleaned_fields) - field_classes.index(_check_field)
key = '.'.join(parts[:last_listField])
parts = parts[last_listField:]
parts.insert(0, key)