src/reporting/views.py

   1 # This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
   3 #
   4 import os.path
   5 from datetime import date
   6 from django.conf import settings
   7 from django.contrib.admin.views.decorators import staff_member_required
   8 from django.db.models import Count
   9 from django.shortcuts import render
  10
  11 from catalogue.models import Book, BookMedia
  12 from reporting.utils import render_to_pdf, render_to_csv, generated_file_view
  13
  14
  15 @staff_member_required
  16 def stats_page(request):
  17     media_types = BookMedia.objects.values('type').annotate(count=Count('type')).order_by('type')
  18     for mt in media_types:
  19         mt['size'] = sum(b.file.size for b in BookMedia.objects.filter(type=mt['type']).iterator())
  20         if mt['type'] in ('mp3', 'ogg'):
  21             deprecated = BookMedia.objects.filter(type=mt['type'], source_sha1=None)
  22             mt['deprecated'] = deprecated.count()
  23             mt['deprecated_files'] = deprecated.order_by('book', 'name')
  24         else:
  25             mt['deprecated'] = '-'
  26
  27     licenses = set()
  28     for b in Book.objects.all().iterator():
  29         extra_info = b.get_extra_info_json()
  30         if extra_info.get('license'):
  31             licenses.add((extra_info.get('license'), extra_info.get('license_description')))
  32
  33     etags = []
  34     all_books = Book.objects.all()
  35     all_books_count = all_books.count()
  36     noparent_books = Book.objects.filter(children=None)
  37     noparent_books_count = noparent_books.count()
  38     for field in Book._meta.fields:
  39         if not getattr(field, 'with_etag', None): continue
  40         etag = field.get_current_etag()
  41         d = {
  42             'field': field.name,
  43             'etag': etag,
  44         }
  45         if field.for_parents:
  46             books = all_books
  47             n_books = all_books_count
  48         else:
  49             books = noparent_books
  50             n_books = noparent_books_count
  51         tags = books.values_list(field.etag_field.name).order_by(
  52             '-' + field.etag_field.name).distinct().annotate(c=Count('*'))
  53         d['tags'] = [
  54             {
  55                 'tag': t[0],
  56                 'count': t[1],
  57                 'perc': round(100 * t[1] / n_books, 2)
  58             }
  59             for t in tags
  60         ]
  61         etags.append(d)
  62
  63     return render(request, 'reporting/main.html', {
  64         'media_types': media_types,
  65         'licenses': licenses,
  66         'etags': etags,
  67     })
  68
  69
  70 @generated_file_view('reports/katalog.pdf', 'application/pdf',
  71                      send_name=lambda: 'wolnelektury_%s.pdf' % date.today(), signals=[Book.published])
  72 def catalogue_pdf(path):
  73     books_by_author, orphans, books_by_parent = Book.book_list()
  74     render_to_pdf(path, 'reporting/catalogue.texml', {
  75         'books_by_author': books_by_author,
  76         'orphans': orphans,
  77         'books_by_parent': books_by_parent,
  78     }, {
  79         "wl-logo.png": os.path.join(settings.STATIC_ROOT, "img/logo-big.png"),
  80     })
  81
  82
  83 @generated_file_view('reports/katalog.csv', 'application/csv',
  84                      send_name=lambda: 'wolnelektury_%s.csv' % date.today(), signals=[Book.published])
  85 def catalogue_csv(path):
  86     books_by_author, orphans, books_by_parent = Book.book_list()
  87     render_to_csv(path, 'reporting/catalogue.csv', {
  88         'books_by_author': books_by_author,
  89         'orphans': orphans,
  90         'books_by_parent': books_by_parent,
  91     })