Obey length limits for wikidata import.

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

diff --git a/src/catalogue/views.py b/src/catalogue/views.py
index 44ad305..097c549 100644 (file)
--- a/src/catalogue/views.py
+++ b/src/catalogue/views.py
@@ -1,8 +1,9 @@
 # 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.db.models import Prefetch
+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
@@ -20,7 +21,7 @@ from rest_framework.permissions import IsAdminUser
 from rest_framework.response import Response
 from rest_framework.views import APIView
 from rest_framework import serializers
-from depot.woblink import get_woblink_session
+import depot.models
 
 
 
@@ -92,7 +93,7 @@ class BookAPIView(RetrieveAPIView):
                 'original_year',
                 'pd_year',
             ]
-    
+
 
 class TermSearchFilter(SearchFilter):
     search_param = 'term'
@@ -135,7 +136,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']
@@ -146,7 +147,7 @@ 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')
 
@@ -163,6 +164,100 @@ 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]
 
@@ -218,7 +313,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)
 
@@ -311,20 +406,20 @@ def publish_collection(request, pk):
 
 
 @login_required
-def woblink_author_autocomplete(request):
-    session = get_woblink_session()
+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({})
-    response = session.get(
-        'https://publisher.woblink.com/author/autocomplete/' + term
-    ).json()
-    return JsonResponse({
-        "results": [
-            {
-                "id": item['autId'],
-                "text": item['autFullname'],
-            }
-            for item in response
-        ],
-    })
+
+    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})