X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/a9b4700e3cd649a6dddc3b72dc0bd632f7841c74..ad422879d55a62e02c71024531aa4a2277dedaf3:/src/catalogue/views.py diff --git a/src/catalogue/views.py b/src/catalogue/views.py index 5be41462..2ed4216d 100644 --- a/src/catalogue/views.py +++ b/src/catalogue/views.py @@ -3,7 +3,7 @@ # from django.apps import apps from django.db.models import Prefetch -from django.http import Http404 +from django.http import Http404, JsonResponse from django.urls import reverse from django.utils.formats import localize_input from django.contrib.auth.decorators import login_required @@ -20,7 +20,7 @@ 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 @@ -92,7 +92,7 @@ class BookAPIView(RetrieveAPIView): 'original_year', 'pd_year', ] - + class TermSearchFilter(SearchFilter): search_param = 'term' @@ -106,6 +106,15 @@ class Terms(ListAPIView): 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): @@ -126,7 +135,7 @@ class EditorTerms(Terms): 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'] @@ -137,10 +146,94 @@ class BookTitleTerms(Terms): class WLURITerms(Terms): queryset = models.Book.objects.all() search_fields = ['title', 'slug'] - + 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 + + for thema in self.queryset.all(): + 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] @@ -197,7 +290,7 @@ class WikidataView(APIView): else: d[fieldname] = localize_input(d[fieldname]) return Response(d) - + def get(self, request, model, qid): return self.get_object(model, qid, save=False) @@ -210,7 +303,15 @@ class WikidataView(APIView): 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])) @@ -221,6 +322,7 @@ def publish_author(request, pk): 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, @@ -234,6 +336,7 @@ def publish_genre(request, pk): 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, } @@ -246,6 +349,7 @@ def publish_kind(request, pk): 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, @@ -276,3 +380,23 @@ def publish_collection(request, pk): 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})