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

fix validatione error for invalid embedded document instance #1067

Browse Source
This commit is contained in:
Erdenezul Batmunkh
2017-11-21 21:56:26 +08:00
committed by Erdenezul
parent c45dfacb41
commit b35efb9f72
2 changed files with 40 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mongoengine/fields.py
View File

@@ -689,6 +689,11 @@ class GenericEmbeddedDocumentField(BaseField):
return value
def validate(self, value, clean=True):
if self.choices and isinstance(value, SON):
for choice in self.choices:
if value['_cls'] == choice._class_name:
return True
if not isinstance(value, EmbeddedDocument):
self.error('Invalid embedded document instance provided to an '
'GenericEmbeddedDocumentField')
@@ -706,7 +711,6 @@ class GenericEmbeddedDocumentField(BaseField):
def to_mongo(self, document, use_db_field=True, fields=None):
if document is None:
return None
data = document.to_mongo(use_db_field, fields)
if '_cls' not in data:
data['_cls'] = document._class_name