fix(sample_project): add autocomplete querysets for person and place

Closes: #900
acdh-oeaw · Jul 30, 2024 · 8d6d8dd · 8d6d8dd
1 parent e8de605
commit 8d6d8dd
Showing 1 changed file with 41 additions and 0 deletions.
diff --git a/sample_project/querysets.py b/sample_project/querysets.py
@@ -0,0 +1,41 @@
+import os
+from apis_core.utils.autocomplete import (
+    ExternalAutocomplete,
+    TypeSenseAutocompleteAdapter,
+    LobidAutocompleteAdapter,
+)
+
+
+class PlaceExternalAutocomplete(ExternalAutocomplete):
+    adapters = [
+        TypeSenseAutocompleteAdapter(
+            collections=[
+                "prosnet-wikidata-place-index",
+                "prosnet-geonames-place-index",
+            ],
+            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",
+            }
+        ),
+    ]