inherit parent Document type _auto_id_field value

2014-02-23 19:37:42 -05:00 · 2014-02-23 19:37:42 -05:00 · 9d125c9e79
commit 9d125c9e79
parent d4b3649640
1 changed files with 2 additions and 1 deletions
--- a/mongoengine/base/metaclasses.py
+++ b/mongoengine/base/metaclasses.py
@ -359,7 +359,8 @@ class TopLevelDocumentMetaclass(DocumentMetaclass):
                    new_class.id = field

        # Set primary key if not defined by the document
-        new_class._auto_id_field = False
+        new_class._auto_id_field = getattr(parent_doc_cls,
+                                           '_auto_id_field', False)
        if not new_class._meta.get('id_field'):
            new_class._auto_id_field = True
            new_class._meta['id_field'] = 'id'