Added support for args / kwargs and queryset_manager
Closes MongoEngine/mongoengine#37
This commit is contained in:
parent
c1621ee39c
commit
8879d5560b
2 changed files with 11 additions and 2 deletions
|
|
@ -2,6 +2,10 @@
|
||||||
Changelog
|
Changelog
|
||||||
=========
|
=========
|
||||||
|
|
||||||
|
Changes in 0.6.15
|
||||||
|
=================
|
||||||
|
- Deref list custom id fix
|
||||||
|
|
||||||
Changes in 0.6.14
|
Changes in 0.6.14
|
||||||
=================
|
=================
|
||||||
- Fixed error dict with nested validation
|
- Fixed error dict with nested validation
|
||||||
|
|
|
||||||
|
|
@ -4,6 +4,8 @@ import copy
|
||||||
import itertools
|
import itertools
|
||||||
import operator
|
import operator
|
||||||
|
|
||||||
|
from functools import partial
|
||||||
|
|
||||||
import pymongo
|
import pymongo
|
||||||
from bson.code import Code
|
from bson.code import Code
|
||||||
|
|
||||||
|
|
@ -1871,10 +1873,13 @@ class QuerySetManager(object):
|
||||||
queryset_class = owner._meta['queryset_class'] or QuerySet
|
queryset_class = owner._meta['queryset_class'] or QuerySet
|
||||||
queryset = queryset_class(owner, owner._get_collection())
|
queryset = queryset_class(owner, owner._get_collection())
|
||||||
if self.get_queryset:
|
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)
|
queryset = self.get_queryset(queryset)
|
||||||
else:
|
elif var_names == ('doc_cls', 'queryset',):
|
||||||
queryset = self.get_queryset(owner, queryset)
|
queryset = self.get_queryset(owner, queryset)
|
||||||
|
else:
|
||||||
|
queryset = partial(self.get_queryset, owner, queryset)
|
||||||
return queryset
|
return queryset
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue