Added support for using custom QuerySet classes

2010-09-15 09:47:13 +01:00
parent 2af5f3c56e
commit f11ee1f9cf
3 changed files with 23 additions and 1 deletions
--- a/mongoengine/queryset.py
+++ b/mongoengine/queryset.py
@@ -992,7 +992,8 @@ class QuerySetManager(object):
                self._collection = db[collection]

        # owner is the document that contains the QuerySetManager
-        queryset = QuerySet(owner, self._collection)
+        queryset_class = owner._meta['queryset_class'] or QuerySet
+        queryset = queryset_class(owner, self._collection)
        if self._manager_func:
            if self._manager_func.func_code.co_argcount == 1:
                queryset = self._manager_func(queryset)