oarepo-runtime 1.5.37__tar.gz → 1.5.38__tar.gz

{oarepo_runtime-1.5.37/oarepo_runtime.egg-info → oarepo_runtime-1.5.38}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oarepo-runtime
-Version: 1.5.37
+Version: 1.5.38
 Summary: A set of runtime extensions of Invenio repository
 Description-Content-Type: text/markdown
 License-File: LICENSE

{oarepo_runtime-1.5.37 → oarepo_runtime-1.5.38}/oarepo_runtime/services/search.py

@@ -12,6 +12,8 @@ from invenio_records_resources.services.records.params import (
     SortParam,
 )
 from invenio_records_resources.services.records.queryparser import SuggestQueryParser
+from invenio_search.engine import dsl
+
 
 # TODO: integrate this to invenio_records_resources.services.records and remove SearchOptions class
 from oarepo_runtime.i18n import lazy_gettext as _
@@ -25,6 +27,27 @@ except ImportError:
     from invenio_i18n.babel import get_locale
 
 
+class FuzzySuggestQueryParser(SuggestQueryParser):
+    def __init__(self, identity=None, extra_params=None, **kwargs):
+        """Constructor."""
+        super().__init__(identity=identity, extra_params=extra_params)
+        self.fields = self.extra_params.get("fields", [])
+        self.extra_params.setdefault("type", "bool_prefix")
+
+    def parse(self, query_str):
+        """Parse the query."""
+        # default behavior
+        multi_match_with_bool_prefix = dsl.Q(
+            "multi_match", query=query_str, **self.extra_params
+        )
+        # fuzziness does not seem to work with bool_prefix multimatch query, so we turn this into
+        # should multi match query with two clauses
+        multi_match_fuzzy = dsl.Q(
+            "multi_match", query=query_str, fields=self.fields, fuzziness="AUTO"
+        )
+        return dsl.Q("bool", should=[multi_match_with_bool_prefix, multi_match_fuzzy])
+
+
 class SearchOptions(InvenioSearchOptions):
     params_interpreters_cls = [
         QueryStrParam,
@@ -114,9 +137,7 @@ class ICUSuggestParser:
                     f"{fld.field}._index_prefix^{fld.boost * fld.boost_prefix}"
                 )
 
-        return SuggestQueryParser.factory(
-            fields=fields,
-        )
+        return FuzzySuggestQueryParser.factory(fields=fields)
 
 
 @dataclasses.dataclass

{oarepo_runtime-1.5.37 → oarepo_runtime-1.5.38/oarepo_runtime.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oarepo-runtime
-Version: 1.5.37
+Version: 1.5.38
 Summary: A set of runtime extensions of Invenio repository
 Description-Content-Type: text/markdown
 License-File: LICENSE

{oarepo_runtime-1.5.37 → oarepo_runtime-1.5.38}/setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = oarepo-runtime
-version = 1.5.37
+version = 1.5.38
 description = A set of runtime extensions of Invenio repository
 authors = Alzbeta Pokorna
 readme = README.md