Fixed queryset manager issue (MongoEngine/mongoengine#52)

docs/changelog.rst

@@ -5,6 +5,7 @@ Changelog
 Changes in 0.6.X
 ================
 
+- Fixed queryset manager issue (MongoEngine/mongoengine#52)
 - Fixed FileField comparision (hmarr/mongoengine#547)
 
 Changes in 0.6.18

mongoengine/queryset.py

@@ -1881,10 +1881,10 @@ class QuerySetManager(object):
         queryset_class = owner._meta['queryset_class'] or QuerySet
         queryset = queryset_class(owner, owner._get_collection())
         if self.get_queryset:
-            var_names = self.get_queryset.func_code.co_varnames
-            if len(var_names) == 1:
+            arg_count = self.get_queryset.func_code.co_argcount
+            if arg_count == 1:
                 queryset = self.get_queryset(queryset)
-            elif len(var_names) == 2:
+            elif arg_count == 2:
                 queryset = self.get_queryset(owner, queryset)
             else:
                 queryset = partial(self.get_queryset, owner, queryset)

tests/test_queryset.py

@@ -2287,7 +2287,8 @@ class QuerySetTest(unittest.TestCase):
 
             @queryset_manager
             def objects(cls, qryset):
-                return qryset(deleted=False)
+                opts = {"deleted": False}
+                return qryset(**opts)
 
             @queryset_manager
             def music_posts(doc_cls, queryset, deleted=False):