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

Ensures EmbeddedDocumentField does not accepts references to Document classes in its constructor

Browse Source
This commit is contained in:
Bastien Gérard
2018-08-30 16:03:16 +02:00
parent bf2de81873
commit 5dbee2a270
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

@@ -645,9 +645,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):