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.apps import apps
   5 from django.db.models import Prefetch
   6 from django.http import Http404
   7 from django.urls import reverse
   8 from django.utils.formats import localize_input
   9 from django.contrib.auth.decorators import login_required
  10 from django.contrib.auth.models import User
  11 from django.shortcuts import get_object_or_404, redirect
  12 from django.views.decorators.http import require_POST
  13 from django.views.generic import DetailView, TemplateView
  14 import apiclient
  15 from . import models
  16 import documents.models
  17 from rest_framework.generics import ListAPIView, RetrieveAPIView
  18 from rest_framework.filters import SearchFilter
  19 from rest_framework.permissions import IsAdminUser
  20 from rest_framework.response import Response
  21 from rest_framework.views import APIView
  22 from rest_framework import serializers
  23
  24
  25
  26
  27
  28 class CatalogueView(TemplateView):
  29     template_name = "catalogue/catalogue.html"
  30
  31     def get_context_data(self):
  32         ctx = super().get_context_data()
  33         ctx["authors"] = models.Author.objects.all().prefetch_related('book_set__document_books', 'translated_book_set__document_books')
  34
  35         return ctx
  36
  37
  38 class AuthorView(TemplateView):
  39     model = models.Author
  40     template_name = "catalogue/author_detail.html"
  41
  42     def get_context_data(self, slug):
  43         ctx = super().get_context_data()
  44         authors = models.Author.objects.filter(slug=slug).prefetch_related(
  45             Prefetch("book_set"),
  46             Prefetch("translated_book_set"),
  47         )
  48         ctx["author"] = authors.first()
  49         return ctx
  50
  51
  52 class BookView(DetailView):
  53     model = models.Book
  54
  55 class BookAPIView(RetrieveAPIView):
  56     queryset = models.Book.objects.all()
  57     lookup_field = 'slug'
  58
  59     class serializer_class(serializers.ModelSerializer):
  60         class AuthorSerializer(serializers.ModelSerializer):
  61             literal = serializers.CharField(source='name')
  62
  63             class Meta:
  64                 model = models.Author
  65                 fields = ['literal']
  66
  67         def category_serializer(m):
  68             class CategorySerializer(serializers.ModelSerializer):
  69                 literal = serializers.CharField(source='name')
  70                 class Meta:
  71                     model = m
  72                     fields = ['literal']
  73             return CategorySerializer
  74
  75         authors = AuthorSerializer(many=True)
  76         translators = AuthorSerializer(many=True)
  77         epochs = category_serializer(models.Epoch)(many=True)
  78         kinds = category_serializer(models.Kind)(many=True)
  79         genres = category_serializer(models.Genre)(many=True)
  80
  81         class Meta:
  82             model = models.Book
  83             fields = [
  84                 'title',
  85                 'authors',
  86                 'translators',
  87                 'epochs',
  88                 'kinds',
  89                 'genres',
  90                 'scans_source',
  91                 'text_source',
  92                 'original_year',
  93                 'pd_year',
  94             ]
  95
  96
  97 class TermSearchFilter(SearchFilter):
  98     search_param = 'term'
  99
 100
 101 class Terms(ListAPIView):
 102     filter_backends = [TermSearchFilter]
 103     search_fields = ['name']
 104
 105     class serializer_class(serializers.Serializer):
 106         label = serializers.CharField(source='name')
 107
 108
 109 class EpochTerms(Terms):
 110     queryset = models.Epoch.objects.all()
 111 class KindTerms(Terms):
 112     queryset = models.Kind.objects.all()
 113 class GenreTerms(Terms):
 114     queryset = models.Genre.objects.all()
 115
 116 class AuthorTerms(Terms):
 117     search_fields = ['first_name', 'last_name']
 118     queryset = models.Author.objects.all()
 119
 120 class EditorTerms(Terms):
 121     search_fields = ['first_name', 'last_name', 'username']
 122     queryset = User.objects.all()
 123
 124     class serializer_class(serializers.Serializer):
 125         label = serializers.SerializerMethodField()
 126
 127         def get_label(self, obj):
 128             return f'{obj.last_name}, {obj.first_name}'
 129
 130 class BookTitleTerms(Terms):
 131     queryset = models.Book.objects.all()
 132     search_fields = ['title', 'slug']
 133
 134     class serializer_class(serializers.Serializer):
 135         label = serializers.CharField(source='title')
 136
 137 class WLURITerms(Terms):
 138     queryset = models.Book.objects.all()
 139     search_fields = ['title', 'slug']
 140
 141     class serializer_class(serializers.Serializer):
 142         label = serializers.CharField(source='wluri')
 143
 144 class ThemaTerms(Terms):
 145     queryset = models.Thema.objects.filter(usable=True, hidden=False)
 146     search_fields = ['code', 'name', 'description']
 147
 148     class serializer_class(serializers.Serializer):
 149         label = serializers.CharField(source='code')
 150         name = serializers.CharField()
 151         description = serializers.CharField()
 152
 153
 154 class WikidataView(APIView):
 155     permission_classes = [IsAdminUser]
 156
 157     def get_object(self, model, qid, save):
 158         try:
 159             Model = apps.get_model('catalogue', model)
 160         except LookupError:
 161             raise Http404
 162         if not issubclass(Model, models.WikidataModel):
 163             raise Http404
 164
 165         obj = Model.objects.filter(wikidata=qid).first()
 166         if obj is None:
 167             obj = Model(wikidata=qid)
 168         if not obj.pk and save:
 169             obj.save()
 170         else:
 171             obj.wikidata_populate(save=False, force=True)
 172         d = {
 173             "id": obj.pk,
 174             "__str__": str(obj),
 175         }
 176         for attname in dir(Model.Wikidata):
 177             if attname.startswith("_"):
 178                 continue
 179             for fieldname, lang in obj.wikidata_fields_for_attribute(attname):
 180                 try:
 181                     d[fieldname] = getattr(obj, fieldname)
 182                 except ValueError:
 183                     # Like accessing related field on non-saved object.
 184                     continue
 185
 186                 if isinstance(d[fieldname], models.WikidataModel):
 187                     d[fieldname] = {
 188                         "model": type(d[fieldname])._meta.model_name,
 189                         "id": d[fieldname].pk,
 190                         "wd": d[fieldname].wikidata,
 191                         "label": str(d[fieldname]) or d[fieldname]._wikidata_label,
 192                     }
 193                 elif hasattr(d[fieldname], 'all'):
 194                     d[fieldname] = [
 195                         {
 196                             "model": type(item)._meta.model_name,
 197                             "id": item.pk,
 198                             "wd": item.wikidata,
 199                             "label": str(item) or item._wikidata_label
 200                         } for item in d[fieldname].all()
 201                     ]
 202                 elif hasattr(d[fieldname], 'as_hint_json'):
 203                     d[fieldname] = d[fieldname].as_hint_json()
 204                 elif hasattr(d[fieldname], 'storage'):
 205                     d[fieldname] = d[fieldname].url if d[fieldname] else None
 206                 else:
 207                     d[fieldname] = localize_input(d[fieldname])
 208         return Response(d)
 209
 210     def get(self, request, model, qid):
 211         return self.get_object(model, qid, save=False)
 212
 213     def post(self, request, model, qid):
 214         return self.get_object(model, qid, save=True)
 215
 216
 217 @require_POST
 218 @login_required
 219 def publish_author(request, pk):
 220     author = get_object_or_404(models.Author, pk=pk)
 221     data = {
 222         "description_pl": author.generate_description(),
 223     }
 224     apiclient.api_call(request.user, f"authors/{author.slug}/", data)
 225     return redirect(reverse('admin:catalogue_author_change', args=[author.pk]))
 226
 227
 228 @require_POST
 229 @login_required
 230 def publish_genre(request, pk):
 231     obj = get_object_or_404(models.Genre, pk=pk)
 232     data = {
 233         "description_pl": obj.description,
 234         "plural": obj.plural,
 235         "is_epoch_specific": obj.is_epoch_specific,
 236     }
 237     apiclient.api_call(request.user, f"genres/{obj.slug}/", data)
 238     return redirect(reverse('admin:catalogue_genre_change', args=[obj.pk]))
 239
 240
 241 @require_POST
 242 @login_required
 243 def publish_kind(request, pk):
 244     obj = get_object_or_404(models.Kind, pk=pk)
 245     data = {
 246         "description_pl": obj.description,
 247         "collective_noun": obj.collective_noun,
 248     }
 249     apiclient.api_call(request.user, f"kinds/{obj.slug}/", data)
 250     return redirect(reverse('admin:catalogue_kind_change', args=[obj.pk]))
 251
 252
 253 @require_POST
 254 @login_required
 255 def publish_epoch(request, pk):
 256     obj = get_object_or_404(models.Epoch, pk=pk)
 257     data = {
 258         "description_pl": obj.description,
 259         "adjective_feminine_singular": obj.adjective_feminine_singular,
 260         "adjective_nonmasculine_plural": obj.adjective_feminine_singular,
 261     }
 262     apiclient.api_call(request.user, f"epochs/{obj.slug}/", data)
 263     return redirect(reverse('admin:catalogue_epoch_change', args=[obj.pk]))
 264
 265
 266 @require_POST
 267 @login_required
 268 def publish_collection(request, pk):
 269     collection = get_object_or_404(models.Collection, pk=pk)
 270     data = {
 271         "title": collection.name,
 272         "description": collection.description,
 273         "book_slugs": "\n".join(
 274             book.slug
 275             for book in collection.book_set.exclude(slug=None).exclude(slug='')
 276         )
 277     }
 278     apiclient.api_call(
 279         request.user,
 280         f"collections/{collection.slug}/",
 281         data,
 282         method='PUT',
 283         as_json=True,
 284     )
 285     return redirect(reverse(
 286         'admin:catalogue_collection_change', args=[collection.pk]
 287     ))