EbookFoundation
/
regluit
mirror of https://github.com/EbookFoundation/regluit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'ry' of github.com:Gluejar/regluit into ry

pull/1/head
Raymond Yee 2012-02-01 09:44:37 -08:00
parent fb77d1a08d da552d75db
commit fe8c060750
1 changed files with 28 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
test/querytests.py
View File

@ -1,26 +1,39 @@
from regluit.core import models from regluit.core import models
from django.db.models import Q, Count, Sum from django.db.models import Count
from regluit.core import userlists
from itertools import izip from itertools import izip
def list_popular(): def list_popular():
work_set = models.Work.objects.annotate(wished=Count('wishlists')).order_by('-wished') """Compare calculating popular works using QuerySets + distinct() and order_by() with an alternate approach """
print work_set
counts={} w1 = models.Work.objects.filter(wishlists__isnull=False). \
counts['unglued'] = work_set.filter(editions__ebooks__isnull=False).distinct().count() distinct().annotate(wished=Count('wishlists')).order_by('-wished', 'id')
counts['unglueing'] = work_set.filter(campaigns__status='ACTIVE').count()
counts['wished'] = work_set.count() - counts['unglued'] - counts['unglueing']
print counts
ungluers = userlists.work_list_users(work_set,5) # create a list of tuples of Works + the wishlist count, filter by non-zero wishlist counts, sort the list by descending
print ungluers # number of wishlists + Work.id and then blot out the wishlist count
def list_new(): w0 = map (lambda x: x[0],
w1 = models.Work.objects.filter(wishlists__isnull=False).distinct().order_by('-created') sorted(
w0 = [w for w in models.Work.objects.order_by('-created') if w.wishlists.count()] filter(lambda x: x[1] > 0,
[(w, w.wishlists.count()) for w in models.Work.objects.all()]
) ,
key=lambda x: (-x[1],x[0].id)
)
)
print w1.count() print w1.count()
print len(w0) print len(w0)
print w0 == w1 print list(w1.all()) == w0
print "difference: ", filter(lambda item: item[1][0] != item[1][1], enumerate(izip(w0,w1)))
def list_new():
"""Compare calculating new works using QuerySets + distinct() and order_by() with an alternate approach """
w1 = models.Work.objects.filter(wishlists__isnull=False).distinct().order_by('-created', 'id')
w0 = [w for w in models.Work.objects.order_by('-created', 'id') if w.wishlists.count()]
print w1.count()
print len(w0)
print list(w1.all()) == w0
print "difference: ", filter(lambda item: item[1][0] != item[1][1], enumerate(izip(w0,w1)))