book text template fix

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

diff --git a/apps/reporting/views.py b/apps/reporting/views.py
index 958e080..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)