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

Relaxed field name checking on embedded documents

Browse Source
This commit is contained in:
Ross Lawley 2011-09-20 03:45:11 -07:00
parent 89c44cd14e
commit f7fbb3d2f6
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mongoengine/base.py
View File

@ -391,7 +391,7 @@ class DocumentMetaclass(type):
attrs['_db_field_map'] = dict([(k, v.db_field) for k, v in doc_fields.items() if k!=v.db_field])
attrs['_reverse_db_field_map'] = dict([(v, k) for k, v in attrs['_db_field_map'].items()])
from mongoengine import Document
from mongoengine import Document, EmbeddedDocument
new_class = super_new(cls, name, bases, attrs)
for field in new_class._fields.values():
@ -401,7 +401,7 @@ class DocumentMetaclass(type):
field.document_type.register_delete_rule(new_class, field.name,
delete_rule)
if field.name and hasattr(Document, field.name):
if field.name and hasattr(Document, field.name) and EmbeddedDocument not in new_class.mro():
raise InvalidDocumentError("%s is a document method and not a valid field name" % field.name)
module = attrs.get('__module__')

2
tests/document.py
View File

@ -2336,7 +2336,7 @@ class DocumentTest(unittest.TestCase):
pickle_doc.reload()
self.assertEquals(resurrected, pickle_doc)
def throw_invalid_document_error(self):
def test_throw_invalid_document_error(self):
# test handles people trying to upsert
def throw_invalid_document_error():