Preserve sort order

scheb · Mar 8, 2024 · 2e554be · 2e554be
1 parent dd95170
commit 2e554be
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/recipe_db/search/elasticsearch.py b/recipe_db/search/elasticsearch.py
@@ -36,10 +36,10 @@ def __init__(self, result: ObjectApiResponse):
 
     @property
     def recipes(self) -> Iterable[Recipe]:
-        # TODO: preserve score
-        ids = map(lambda r: r['_id'], self._result)
-        recipes = Recipe.objects.filter(uid__in=ids)
+        ids = list(map(lambda r: r['_id'], self._result))
         builder = RecipeResultBuilder()
+        recipes = list(Recipe.objects.filter(uid__in=ids))
+        recipes = sorted(recipes, key=lambda r: ids.index(r.uid))
         for recipe in recipes:
             yield builder.create_recipe_result(recipe)