Adding _collection to _cls

2012-11-27 14:02:50 +00:00
parent f9dd051ec9
commit 3598fe0fb4
2 changed files with 28 additions and 0 deletions
--- a/mongoengine/base/metaclasses.py
+++ b/mongoengine/base/metaclasses.py
@@ -169,6 +169,9 @@ class DocumentMetaclass(type):
                       "field name" % field.name)
                raise InvalidDocumentError(msg)

+        if issubclass(new_class, Document):
+            new_class._collection = None
+
        # Add class to the _document_registry
        _document_registry[new_class._class_name] = new_class