Integrates Document with search form

jlariza · May 18, 2024 · ab2ff7c · ab2ff7c
1 parent bd35144
commit ab2ff7c
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/django_project/opensearch_workshop/pokemons/views.py b/django_project/opensearch_workshop/pokemons/views.py
@@ -6,6 +6,7 @@
 
 from .forms import SearchPokemonForm
 from .models import Pokemon
+from .documents import PokemonDocument
 
 
 # Create your views here.
@@ -24,5 +25,20 @@ def get_queryset(self) -> QuerySet[Any]:
         queryset = super().get_queryset()
         if form.is_valid():
             data = form.cleaned_data
-            queryset = queryset.filter(name__icontains=data["name"])
+            name = data["name"]
+            os_query = (
+                PokemonDocument.search()
+                .filter(
+                    "query_string",
+                    query=f"*{name}*",
+                    fields=[
+                        "name",
+                    ],
+                )
+                .sort("pokemon_id")
+                # los resultados de opensearch se deben paginar
+                # pero está fuera del scope de este taller
+                .extra(size=1000, track_total_hits=True)
+            )
+            queryset = os_query.to_queryset()
         return queryset