X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/282654ea252af7e2d30740b40bccfc9be61dd3a8..7720832e5e083dda48bc99b9f07b377b9d11c2cb:/src/catalogue/views.py

diff --git a/src/catalogue/views.py b/src/catalogue/views.py
index 95f33fd0..4ae5c742 100644
--- a/src/catalogue/views.py
+++ b/src/catalogue/views.py
@@ -22,12 +22,15 @@ from rest_framework.views import APIView
 from rest_framework import serializers
 
 
+
+
+
 class CatalogueView(TemplateView):
     template_name = "catalogue/catalogue.html"
 
     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
 
@@ -138,6 +141,15 @@ class WLURITerms(Terms):
     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 WikidataView(APIView):
     permission_classes = [IsAdminUser]
@@ -165,7 +177,11 @@ class WikidataView(APIView):
             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):
                     d[fieldname] = {
@@ -178,6 +194,7 @@ class WikidataView(APIView):
                     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()
@@ -208,6 +225,44 @@ def publish_author(request, pk):
     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 = {
+        "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 = {
+        "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 = {
+        "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):