From d45dd31e9c1d025818733990b6e08bde59d48254 Mon Sep 17 00:00:00 2001
From: Birger Schacht <birger.schacht@oeaw.ac.at>
Date: Tue, 14 Nov 2023 12:34:25 +0100
Subject: [PATCH] feat(autocomplete): add custom querysets

Create querysets.py file, add autocomplete
adapters for places, persons, institutions.
---
 apis_ontology/querysets.py | 45 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 apis_ontology/querysets.py

diff --git a/apis_ontology/querysets.py b/apis_ontology/querysets.py
new file mode 100644
index 0000000..c8e65d9
--- /dev/null
+++ b/apis_ontology/querysets.py
@@ -0,0 +1,45 @@
+import logging
+import os
+from django.db.models.functions import Collate
+from django.conf import settings
+
+from apis_core.utils.autocomplete import ExternalAutocomplete, TypeSenseAutocompleteAdapter, LobidAutocompleteAdapter
+
+from .models import Person
+
+logger = logging.getLogger(__name__)
+
+DB_COLLATION = ("binary" if "sqlite" in settings.DATABASES["default"]["ENGINE"] else "en-x-icu")
+
+
+def PersonListViewQueryset(*args):
+    return Person.objects.all().order_by(Collate("surname", DB_COLLATION), Collate("forename", DB_COLLATION))
+
+
+class PlaceExternalAutocomplete(ExternalAutocomplete):
+    adapters = [
+            TypeSenseAutocompleteAdapter(
+                collections=["prosnet-wikidata-place-index", "prosnet-geonames-place-index"],
+                template="apis_ontology/place_external_autocomplete_result.html",
+                token=os.getenv("TYPESENSE_TOKEN", None),
+                server=os.getenv("TYPESENSE_SERVER", None)),
+            LobidAutocompleteAdapter(params={"filter": "type:PlaceOrGeographicName", "format": "json:preferredName"})
+            ]
+
+
+class PersonExternalAutocomplete(ExternalAutocomplete):
+    adapters = [
+            TypeSenseAutocompleteAdapter(collections="prosnet-wikidata-person-index",
+                                         token=os.getenv("TYPESENSE_TOKEN", None),
+                                         server=os.getenv("TYPESENSE_SERVER", None)),
+            LobidAutocompleteAdapter(params={"filter": "type:Person", "format": "json:preferredName,professionOrOccupation"})
+            ]
+
+
+class InstitutionExternalAutocomplete(ExternalAutocomplete):
+    adapters = [
+            TypeSenseAutocompleteAdapter(collections="prosnet-wikidata-organization-index",
+                                         token=os.getenv("TYPESENSE_TOKEN", None),
+                                         server=os.getenv("TYPESENSE_SERVER", None)),
+            LobidAutocompleteAdapter(params={"filter": "type:CorporateBody", "format": "json:preferredName"})
+            ]