2011-04-09 14:29:38 -07:00
|
|
|
from django.db.models import Count
|
|
|
|
|
|
|
|
from main.models import Todolist
|
|
|
|
|
2012-04-07 12:51:25 -07:00
|
|
|
|
2012-04-27 06:59:00 -07:00
|
|
|
def get_annotated_todolists(incomplete_only=False):
|
2011-04-09 14:29:38 -07:00
|
|
|
qs = Todolist.objects.all()
|
2012-04-07 12:51:25 -07:00
|
|
|
lists = qs.select_related('creator').defer(
|
|
|
|
'creator__email', 'creator__password', 'creator__is_staff',
|
|
|
|
'creator__is_active', 'creator__is_superuser',
|
|
|
|
'creator__last_login', 'creator__date_joined').annotate(
|
2011-04-09 14:29:38 -07:00
|
|
|
pkg_count=Count('todolistpkg')).order_by('-date_added')
|
|
|
|
incomplete = qs.filter(todolistpkg__complete=False).annotate(
|
|
|
|
Count('todolistpkg')).values_list('id', 'todolistpkg__count')
|
|
|
|
|
|
|
|
lookup = dict(incomplete)
|
2012-04-27 06:59:00 -07:00
|
|
|
|
|
|
|
if incomplete_only:
|
|
|
|
lists = lists.filter(id__in=lookup.keys())
|
|
|
|
|
|
|
|
# tag each list with an incomplete package count
|
2011-04-09 14:29:38 -07:00
|
|
|
for todolist in lists:
|
|
|
|
todolist.incomplete_count = lookup.get(todolist.id, 0)
|
|
|
|
|
|
|
|
return lists
|
|
|
|
|
|
|
|
# vim: set ts=4 sw=4 et:
|