fix bug query subfield in generic_embedded_document #1651

This commit is contained in:
Erdenezul 2017-09-15 11:18:24 +08:00
parent be8f1b9fdd
commit aa4996ef28

View File

@ -692,6 +692,14 @@ class GenericEmbeddedDocumentField(BaseField):
value.validate(clean=clean) value.validate(clean=clean)
def lookup_member(self, member_name):
if self.choices:
for choice in self.choices:
field = choice._fields.get(member_name)
if field:
return field
return None
def to_mongo(self, document, use_db_field=True, fields=None): def to_mongo(self, document, use_db_field=True, fields=None):
if document is None: if document is None:
return None return None