jyapayne/mongoengine
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

queryset_manager funcs now accept doc as arg

Browse source
This commit is contained in:
Harry Marr 2010-01-23 17:16:01 +00:00
commit 2585f1b724
5 changed files with 69 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/queryset.py
View file

@ -146,7 +146,7 @@ class QuerySetTest(unittest.TestCase):
published_date = DateTimeField()
@queryset_manager
def published(queryset):
def published(doc_cls, queryset):
return queryset(is_published=True)
blog_post_1 = BlogPost(title="Blog Post #1",
@ -444,7 +444,7 @@ class QuerySetTest(unittest.TestCase):
tags = ListField(StringField())
@queryset_manager
def music_posts(queryset):
def music_posts(doc_cls, queryset):
return queryset(tags='music')
BlogPost.drop_collection()