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

Merge pull request #1859 from bagerard/fix_EmbeddedDocumentField_init_with_Document

Browse Source 
Ensures EmbeddedDocumentField does not accepts references to Document
This commit is contained in:
erdenezul
2018-09-03 17:22:08 +08:00
committed by GitHub
parent 5a091956ef 5dbee2a270
commit 3574e21e4f
3 changed files with 63 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
mongoengine/fields.py
View File

@@ -647,9 +647,17 @@ class EmbeddedDocumentField(BaseField):
def document_type(self):
if isinstance(self.document_type_obj, six.string_types):
if self.document_type_obj == RECURSIVE_REFERENCE_CONSTANT:
self.document_type_obj = self.owner_document
resolved_document_type = self.owner_document
else:
self.document_type_obj = get_document(self.document_type_obj)
resolved_document_type = get_document(self.document_type_obj)
if not issubclass(resolved_document_type, EmbeddedDocument):
# Due to the late resolution of the document_type
# There is a chance that it won't be an EmbeddedDocument (#1661)
self.error('Invalid embedded document class provided to an '
'EmbeddedDocumentField')
self.document_type_obj = resolved_document_type
return self.document_type_obj
def to_python(self, value):