pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed issue with custom queryset manager expecting explict variable names

Browse Source 
If using / expecting kwargs you have to call the queryset manager
explicitly.
This commit is contained in:
Ross Lawley
2012-07-25 14:55:10 +01:00
parent 598ffd3e5c
commit 6459d4c0b6
3 changed files with 21 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mongoengine/queryset.py
View File

@@ -1882,9 +1882,9 @@ class QuerySetManager(object):
queryset = queryset_class(owner, owner._get_collection())
if self.get_queryset:
var_names = self.get_queryset.func_code.co_varnames
if var_names == ('queryset',):
if len(var_names) == 1:
queryset = self.get_queryset(queryset)
elif var_names == ('doc_cls', 'queryset',):
elif len(var_names) == 2:
queryset = self.get_queryset(owner, queryset)
else:
queryset = partial(self.get_queryset, owner, queryset)