Added support for args / kwargs and queryset_manager

Closes MongoEngine/mongoengine#37
This commit is contained in:
Ross Lawley 2012-07-19 10:32:33 +01:00
parent c1621ee39c
commit 8879d5560b
2 changed files with 11 additions and 2 deletions

View File

@ -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

View File

@ -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