src/catalogue/views.py

   1 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   3 #
   4 from django.db.models import Prefetch
   5 from django.views.generic import DetailView, TemplateView
   6 from . import models
   7 import documents.models
   8
   9
  10 class CatalogueView(TemplateView):
  11     template_name = "catalogue/catalogue.html"
  12
  13     def get_context_data(self):
  14         ctx = super().get_context_data()
  15         ctx["authors"] = models.Author.objects.all().prefetch_related('book_set__book_set', 'translated_book_set__book_set')
  16
  17         return ctx
  18
  19
  20 class AuthorView(TemplateView):
  21     model = models.Author
  22     template_name = "catalogue/author_detail.html"
  23
  24     def get_context_data(self, slug):
  25         ctx = super().get_context_data()
  26         authors = models.Author.objects.filter(slug=slug).prefetch_related(
  27             Prefetch("book_set"),
  28             Prefetch("translated_book_set"),
  29         )
  30         ctx["author"] = authors.first()
  31         return ctx
  32
  33
  34 class BookView(DetailView):
  35     model = models.Book