Allow documents to override the 'objects' QuerySetManager

2010-07-30 22:09:00 +10:00 · 2010-07-30 22:09:00 +10:00 · b96e27a7e4
commit b96e27a7e4
parent 196606438c
1 changed files with 4 additions and 1 deletions
--- a/mongoengine/base.py
+++ b/mongoengine/base.py
@ -252,6 +252,9 @@ class TopLevelDocumentMetaclass(DocumentMetaclass):
        # Set up collection manager, needs the class to have fields so use
        # DocumentMetaclass before instantiating CollectionManager object
        new_class = super_new(cls, name, bases, attrs)
+        
+        # Provide a default queryset unless one has been manually provided
+        if not 'objects' in dir(new_class):
            new_class.objects = QuerySetManager()

        user_indexes = [QuerySet._build_index_spec(new_class, spec)