Adding _collection to _cls

This commit is contained in:
Ross Lawley
2012-11-27 14:02:50 +00:00
parent f9dd051ec9
commit 3598fe0fb4
2 changed files with 28 additions and 0 deletions

View File

@@ -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