summaryrefslogtreecommitdiff
path: root/todolists/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'todolists/utils.py')
-rw-r--r--todolists/utils.py62
1 files changed, 48 insertions, 14 deletions
diff --git a/todolists/utils.py b/todolists/utils.py
index 24101e86..7b98c887 100644
--- a/todolists/utils.py
+++ b/todolists/utils.py
@@ -1,23 +1,57 @@
-from django.db.models import Count
+from django.db import connections, router
-from main.models import Todolist
+from .models import Todolist, TodolistPackage
+from packages.models import Package
-def get_annotated_todolists():
- qs = Todolist.objects.all()
- 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(
- pkg_count=Count('todolistpkg')).order_by('-date_added')
- incomplete = qs.filter(todolistpkg__complete=False).annotate(
- Count('todolistpkg')).values_list('id', 'todolistpkg__count')
+def todo_counts():
+ sql = """
+SELECT todolist_id, count(*), sum(CASE WHEN status = %s THEN 1 ELSE 0 END)
+ FROM todolists_todolistpackage
+ WHERE removed IS NULL
+ GROUP BY todolist_id
+ """
+ database = router.db_for_write(TodolistPackage)
+ connection = connections[database]
+ cursor = connection.cursor()
+ cursor.execute(sql, [TodolistPackage.COMPLETE])
+ results = cursor.fetchall()
+ return {row[0]: (row[1], row[2]) for row in results}
- # tag each list with an incomplete package count
- lookup = dict(incomplete)
+
+def get_annotated_todolists(incomplete_only=False):
+ lists = Todolist.objects.all().defer('raw').select_related(
+ 'creator').order_by('-created')
+ lookup = todo_counts()
+
+ # tag each list with package counts
for todolist in lists:
- todolist.incomplete_count = lookup.get(todolist.id, 0)
+ counts = lookup.get(todolist.id, (0, 0))
+ todolist.pkg_count = counts[0]
+ todolist.complete_count = counts[1]
+ todolist.incomplete_count = counts[0] - counts[1]
+
+ if incomplete_only:
+ lists = [l for l in lists if l.incomplete_count > 0]
return lists
+
+def attach_staging(packages, list_id):
+ '''Look for any staging version of the packages provided and attach them
+ to the 'staging' attribute on each package if found.'''
+ pkgnames = TodolistPackage.objects.filter(
+ todolist_id=list_id).values('pkgname')
+ staging_pkgs = Package.objects.normal().filter(repo__staging=True,
+ pkgname__in=pkgnames)
+ # now build a lookup dict to attach to the correct package
+ lookup = {(p.pkgname, p.arch): p for p in staging_pkgs}
+
+ annotated = []
+ for package in packages:
+ in_staging = lookup.get((package.pkgname, package.arch), None)
+ package.staging = in_staging
+
+ return annotated
+
# vim: set ts=4 sw=4 et: