Fix.

[redakcja.git] / src / catalogue / views.py

diff --git a/src/catalogue/views.py b/src/catalogue/views.py
index 65b2980..097c549 100644 (file)
--- a/src/catalogue/views.py
+++ b/src/catalogue/views.py
@@ -1,20 +1,29 @@
 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #

+import json

 from django.apps import apps
 from django.apps import apps

-from django.db.models import Prefetch
-from django.http import Http404
+from django.db.models import Prefetch, Q
+from django.http import Http404, JsonResponse
+from django.urls import reverse

 from django.utils.formats import localize_input
 from django.utils.formats import localize_input

+from django.contrib.auth.decorators import login_required

 from django.contrib.auth.models import User
 from django.contrib.auth.models import User

+from django.shortcuts import get_object_or_404, redirect
+from django.views.decorators.http import require_POST

 from django.views.generic import DetailView, TemplateView
 from django.views.generic import DetailView, TemplateView

+import apiclient

 from . import models
 import documents.models
 from . import models
 import documents.models

-from rest_framework.generics import ListAPIView
+from rest_framework.generics import ListAPIView, RetrieveAPIView

 from rest_framework.filters import SearchFilter
 from rest_framework.permissions import IsAdminUser
 from rest_framework.response import Response
 from rest_framework.views import APIView
 from rest_framework import serializers
 from rest_framework.filters import SearchFilter
 from rest_framework.permissions import IsAdminUser
 from rest_framework.response import Response
 from rest_framework.views import APIView
 from rest_framework import serializers

+import depot.models
+
+

 
 
 class CatalogueView(TemplateView):
 
 
 class CatalogueView(TemplateView):


@@ -22,7 +31,7 @@ class CatalogueView(TemplateView):
 
     def get_context_data(self):
         ctx = super().get_context_data()
 
     def get_context_data(self):
         ctx = super().get_context_data()

-        ctx["authors"] = models.Author.objects.all().prefetch_related('book_set__book_set', 'translated_book_set__book_set')
+        ctx["authors"] = models.Author.objects.all().prefetch_related('book_set__document_books', 'translated_book_set__document_books')

 
         return ctx
 
 
         return ctx
 


@@ -44,6 +53,47 @@ class AuthorView(TemplateView):
 class BookView(DetailView):
     model = models.Book
 
 class BookView(DetailView):
     model = models.Book
 

+class BookAPIView(RetrieveAPIView):
+    queryset = models.Book.objects.all()
+    lookup_field = 'slug'
+
+    class serializer_class(serializers.ModelSerializer):
+        class AuthorSerializer(serializers.ModelSerializer):
+            literal = serializers.CharField(source='name')
+
+            class Meta:
+                model = models.Author
+                fields = ['literal']
+
+        def category_serializer(m):
+            class CategorySerializer(serializers.ModelSerializer):
+                literal = serializers.CharField(source='name')
+                class Meta:
+                    model = m
+                    fields = ['literal']
+            return CategorySerializer
+
+        authors = AuthorSerializer(many=True)
+        translators = AuthorSerializer(many=True)
+        epochs = category_serializer(models.Epoch)(many=True)
+        kinds = category_serializer(models.Kind)(many=True)
+        genres = category_serializer(models.Genre)(many=True)
+
+        class Meta:
+            model = models.Book
+            fields = [
+                'title',
+                'authors',
+                'translators',
+                'epochs',
+                'kinds',
+                'genres',
+                'scans_source',
+                'text_source',
+                'original_year',
+                'pd_year',
+            ]
+

 
 class TermSearchFilter(SearchFilter):
     search_param = 'term'
 
 class TermSearchFilter(SearchFilter):
     search_param = 'term'


@@ -57,6 +107,15 @@ class Terms(ListAPIView):
         label = serializers.CharField(source='name')
 
 
         label = serializers.CharField(source='name')
 
 

+class AudienceTerms(Terms):
+    queryset = models.Audience.objects.all()
+    search_fields = ['code', 'name', 'description']
+
+    class serializer_class(serializers.Serializer):
+        label = serializers.CharField(source='code')
+        name = serializers.CharField()
+        description = serializers.CharField()
+

 class EpochTerms(Terms):
     queryset = models.Epoch.objects.all()
 class KindTerms(Terms):
 class EpochTerms(Terms):
     queryset = models.Epoch.objects.all()
 class KindTerms(Terms):


@@ -77,7 +136,7 @@ class EditorTerms(Terms):
 
         def get_label(self, obj):
             return f'{obj.last_name}, {obj.first_name}'
 
         def get_label(self, obj):
             return f'{obj.last_name}, {obj.first_name}'

-    
+

 class BookTitleTerms(Terms):
     queryset = models.Book.objects.all()
     search_fields = ['title', 'slug']
 class BookTitleTerms(Terms):
     queryset = models.Book.objects.all()
     search_fields = ['title', 'slug']


@@ -88,10 +147,116 @@ class BookTitleTerms(Terms):
 class WLURITerms(Terms):
     queryset = models.Book.objects.all()
     search_fields = ['title', 'slug']
 class WLURITerms(Terms):
     queryset = models.Book.objects.all()
     search_fields = ['title', 'slug']

-    
+

     class serializer_class(serializers.Serializer):
         label = serializers.CharField(source='wluri')
 
     class serializer_class(serializers.Serializer):
         label = serializers.CharField(source='wluri')
 

+class ThemaTerms(Terms):
+    queryset = models.Thema.objects.filter(usable=True, hidden=False)
+    search_fields = ['code', 'name', 'description']
+
+    class serializer_class(serializers.Serializer):
+        label = serializers.CharField(source='code')
+        name = serializers.CharField()
+        description = serializers.CharField()
+
+class MainThemaTerms(ThemaTerms):
+    queryset = models.Thema.objects.filter(usable=True, hidden=False, usable_as_main=True)
+
+
+
+class Chooser(APIView):
+    def get(self, request):
+        return Response([{
+            'value': 'x',
+            'name': 'name',
+            'description': 'desc',
+            'sub': [
+                {
+                    'value': 'y',
+                    'name': 'name y',
+                    'description': 'desc y',
+                }
+            ]
+        }])
+
+
+class ThemaChooser(Chooser):
+    queryset = models.Thema.objects.filter(usable=True, hidden=False)
+
+    def get(self, request):
+        tree = {}
+
+        def getitem(code):
+            if len(code) == 1:
+                parent = tree
+            else:
+                parent = getitem(code[:-1]).setdefault('sub', {})
+            return parent.setdefault(code, {})
+
+        def getmissing(t):
+            for k, v in t.items():
+                if 'name' not in v:
+                    yield k
+                if 'sub' in v:
+                    for c in getmissing(v['sub']):
+                        yield c
+
+        def populate(thema):
+            item = getitem(thema.code)
+            item['usable'] = thema.usable
+            item['hidden'] = thema.hidden
+            item['name'] = thema.name
+            item['description'] = thema.description
+
+        def order(tree):
+            res = []
+            for k, v in tree.items():
+                v.update(value=k)
+                if 'sub' in v:
+                    v['sub'] = order(v['sub'])
+                res.append(v)
+            while len(res) == 1 and 'name' not in res[0] and 'sub' in res[0]:
+                res = res[0]['sub']
+            return res
+
+        def apply_filter(filt):
+            if 'not' in filt:
+                return ~apply_filter(filt['not'])
+            if 'startswith' in filt:
+                q = None
+                for prefix in filt['startswith']:
+                    q2 = Q(code__startswith=prefix)
+                    if q:
+                        q |= q2
+                    else:
+                        q = q2
+                return q
+            assert False
+
+        qs = self.queryset
+        try:
+            filt = json.loads(request.GET.get('filter'))
+        except:
+            pass
+        else:
+            qs = qs.filter(apply_filter(filt))
+
+        for thema in qs:
+            populate(thema)
+
+        missing = list(getmissing(tree))
+        for thema in models.Thema.objects.filter(code__in=missing):
+            populate(thema)
+
+        tree = order(tree)
+
+        return Response(tree)
+
+
+class MainThemaChooser(ThemaChooser):
+    queryset = models.Thema.objects.filter(usable=True, hidden=False, usable_as_main=True)[:1000]
+

 
 class WikidataView(APIView):
     permission_classes = [IsAdminUser]
 
 class WikidataView(APIView):
     permission_classes = [IsAdminUser]


@@ -110,28 +275,151 @@ class WikidataView(APIView):
         if not obj.pk and save:
             obj.save()
         else:
         if not obj.pk and save:
             obj.save()
         else:

-            obj.wikidata_populate(save=False)
+            obj.wikidata_populate(save=False, force=True)

         d = {
             "id": obj.pk,
         d = {
             "id": obj.pk,

+            "__str__": str(obj),

         }
         for attname in dir(Model.Wikidata):
             if attname.startswith("_"):
                 continue
             for fieldname, lang in obj.wikidata_fields_for_attribute(attname):
         }
         for attname in dir(Model.Wikidata):
             if attname.startswith("_"):
                 continue
             for fieldname, lang in obj.wikidata_fields_for_attribute(attname):

-                d[fieldname] = getattr(obj, fieldname)
+                try:
+                    d[fieldname] = getattr(obj, fieldname)
+                except ValueError:
+                    # Like accessing related field on non-saved object.
+                    continue

 
                 if isinstance(d[fieldname], models.WikidataModel):
 
                 if isinstance(d[fieldname], models.WikidataModel):

-                    d[attname] = {
+                    d[fieldname] = {

                         "model": type(d[fieldname])._meta.model_name,
                         "model": type(d[fieldname])._meta.model_name,

+                        "id": d[fieldname].pk,

                         "wd": d[fieldname].wikidata,
                         "label": str(d[fieldname]) or d[fieldname]._wikidata_label,
                     }
                         "wd": d[fieldname].wikidata,
                         "label": str(d[fieldname]) or d[fieldname]._wikidata_label,
                     }

+                elif hasattr(d[fieldname], 'all'):
+                    d[fieldname] = [
+                        {
+                            "model": type(item)._meta.model_name,
+                            "id": item.pk,
+                            "wd": item.wikidata,
+                            "label": str(item) or item._wikidata_label
+                        } for item in d[fieldname].all()
+                    ]
+                elif hasattr(d[fieldname], 'as_hint_json'):
+                    d[fieldname] = d[fieldname].as_hint_json()
+                elif hasattr(d[fieldname], 'storage'):
+                    d[fieldname] = d[fieldname].url if d[fieldname] else None

                 else:
                     d[fieldname] = localize_input(d[fieldname])
         return Response(d)
                 else:
                     d[fieldname] = localize_input(d[fieldname])
         return Response(d)

-    
+

     def get(self, request, model, qid):
         return self.get_object(model, qid, save=False)
 
     def post(self, request, model, qid):
         return self.get_object(model, qid, save=True)
     def get(self, request, model, qid):
         return self.get_object(model, qid, save=False)
 
     def post(self, request, model, qid):
         return self.get_object(model, qid, save=True)

+
+
+@require_POST
+@login_required
+def publish_author(request, pk):
+    author = get_object_or_404(models.Author, pk=pk)
+    data = {
+        "name_pl": author.name,
+        "description_pl": author.generate_description(),
+        "genitive": author.genitive,
+        "gazeta_link": author.gazeta_link,
+        "culturepl_link": author.culturepl_link,
+        "wiki_link_pl": author.plwiki,
+        "photo": request.build_absolute_uri(author.photo.url) if author.photo else None,
+        "photo_source": author.photo_source,
+        "photo_attribution": author.photo_attribution,
+    }
+    apiclient.api_call(request.user, f"authors/{author.slug}/", data)
+    return redirect(reverse('admin:catalogue_author_change', args=[author.pk]))
+
+
+@require_POST
+@login_required
+def publish_genre(request, pk):
+    obj = get_object_or_404(models.Genre, pk=pk)
+    data = {
+        "name_pl": obj.name,
+        "description_pl": obj.description,
+        "plural": obj.plural,
+        "is_epoch_specific": obj.is_epoch_specific,
+    }
+    apiclient.api_call(request.user, f"genres/{obj.slug}/", data)
+    return redirect(reverse('admin:catalogue_genre_change', args=[obj.pk]))
+
+
+@require_POST
+@login_required
+def publish_kind(request, pk):
+    obj = get_object_or_404(models.Kind, pk=pk)
+    data = {
+        "name_pl": obj.name,
+        "description_pl": obj.description,
+        "collective_noun": obj.collective_noun,
+    }
+    apiclient.api_call(request.user, f"kinds/{obj.slug}/", data)
+    return redirect(reverse('admin:catalogue_kind_change', args=[obj.pk]))
+
+
+@require_POST
+@login_required
+def publish_epoch(request, pk):
+    obj = get_object_or_404(models.Epoch, pk=pk)
+    data = {
+        "name_pl": obj.name,
+        "description_pl": obj.description,
+        "adjective_feminine_singular": obj.adjective_feminine_singular,
+        "adjective_nonmasculine_plural": obj.adjective_feminine_singular,
+    }
+    apiclient.api_call(request.user, f"epochs/{obj.slug}/", data)
+    return redirect(reverse('admin:catalogue_epoch_change', args=[obj.pk]))
+
+
+@require_POST
+@login_required
+def publish_collection(request, pk):
+    collection = get_object_or_404(models.Collection, pk=pk)
+    data = {
+        "title": collection.name,
+        "description": collection.description,
+        "book_slugs": "\n".join(
+            book.slug
+            for book in collection.book_set.exclude(slug=None).exclude(slug='')
+        )
+    }
+    apiclient.api_call(
+        request.user,
+        f"collections/{collection.slug}/",
+        data,
+        method='PUT',
+        as_json=True,
+    )
+    return redirect(reverse(
+        'admin:catalogue_collection_change', args=[collection.pk]
+    ))
+
+
+@login_required
+def woblink_autocomplete(request, category):
+    site = depot.models.Site.objects.filter(site_type='woblink').first()
+    if site is None:
+        return JsonResponse({})
+    woblink = site.get_publisher()
+    term = request.GET.get('term')
+    if not term:
+        return JsonResponse({})
+
+    if category == 'author':
+        results = woblink.search_author_catalogue(term)
+    elif category == 'series':
+        results = woblink.search_series_catalogue(term)
+    else:
+        raise Http404
+
+    return JsonResponse({"results": results})