use class-based generic views

[wolnelektury.git] / apps / reporting / views.py

diff --git a/apps/reporting/views.py b/apps/reporting/views.py
index 0203863..cb87f81 100644 (file)
--- a/apps/reporting/views.py
+++ b/apps/reporting/views.py
@@ -5,6 +5,7 @@
 import os.path
 from datetime import date
 from django.conf import settings
+from django.contrib.admin.views.decorators import staff_member_required
 from django.db.models import Count
 from django.shortcuts import render_to_response
 from django.template import RequestContext
@@ -13,13 +14,14 @@ from catalogue.models import Book, BookMedia
 from reporting.utils import render_to_pdf, generated_file_view
 
 
+@staff_member_required
 def stats_page(request):
     media = BookMedia.objects.count()
     media_types = BookMedia.objects.values('type').\
             annotate(count=Count('type')).\
             order_by('type')
     for mt in media_types:
-        mt['size'] = sum(b.file.size for b in BookMedia.objects.filter(type=mt['type']))
+        mt['size'] = sum(b.file.size for b in BookMedia.objects.filter(type=mt['type']).iterator())
         if mt['type'] in ('mp3', 'ogg'):
             deprecated = BookMedia.objects.filter(
                     type=mt['type'], source_sha1=None)
@@ -37,7 +39,6 @@ def stats_page(request):
         signals=[Book.published])
 def catalogue_pdf(path):
     books_by_author, orphans, books_by_parent = Book.book_list()
-    print books_by_parent
     render_to_pdf(path, 'reporting/catalogue.texml', locals(), {
             "wl-logo.png": os.path.join(settings.STATIC_ROOT, "img/logo-big.png"),
         })