Added support for args / kwargs and queryset_manager
Closes MongoEngine/mongoengine#37
This commit is contained in:
parent
c1621ee39c
commit
8879d5560b
@ -2,6 +2,10 @@
|
||||
Changelog
|
||||
=========
|
||||
|
||||
Changes in 0.6.15
|
||||
=================
|
||||
- Deref list custom id fix
|
||||
|
||||
Changes in 0.6.14
|
||||
=================
|
||||
- Fixed error dict with nested validation
|
||||
|
@ -4,6 +4,8 @@ import copy
|
||||
import itertools
|
||||
import operator
|
||||
|
||||
from functools import partial
|
||||
|
||||
import pymongo
|
||||
from bson.code import Code
|
||||
|
||||
@ -1871,10 +1873,13 @@ class QuerySetManager(object):
|
||||
queryset_class = owner._meta['queryset_class'] or QuerySet
|
||||
queryset = queryset_class(owner, owner._get_collection())
|
||||
if self.get_queryset:
|
||||
if self.get_queryset.func_code.co_argcount == 1:
|
||||
var_names = self.get_queryset.func_code.co_varnames
|
||||
if var_names == ('queryset',):
|
||||
queryset = self.get_queryset(queryset)
|
||||
else:
|
||||
elif var_names == ('doc_cls', 'queryset',):
|
||||
queryset = self.get_queryset(owner, queryset)
|
||||
else:
|
||||
queryset = partial(self.get_queryset, owner, queryset)
|
||||
return queryset
|
||||
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user