group results by document_id

Author: ciur

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "salinic"
-version = "0.4.0"
+version = "0.5.0"
 description = "Search abstraction layer"
 authors = ["Eugen Ciur <[email protected]>"]
 readme = "README.md"

salinic/backends/solr/index.py

@@ -16,6 +16,9 @@ def __init__(self, client, schema):
         self.schema = schema
 
     def search(self, sq: SearchQuery):
+        """
+
+        """
         result = self.client.search(sq)
         if result['response']['numFound'] == 0:
             return []

salinic/backends/solr/schema_manager.py

@@ -1,7 +1,7 @@
 import logging
 from typing import List, Tuple
 
-from salinic.field import Field, NumericField
+from salinic.field import Field, NumericField, StringField
 from salinic.utils import first
 
 from .types import CopyFieldDump, FieldDump, FieldType
@@ -109,6 +109,10 @@ def _normal_fields(self) -> List[Tuple[str, FieldDump]]:
 
             if isinstance(field_instance, NumericField):
                 _type = FieldType.pint
+            elif isinstance(field_instance, StringField):
+                _type = FieldType.string
+            elif field_instance.group:
+                _type = FieldType.text_gen_sort
             else:
                 _type = FieldType.text_general
 

salinic/backends/solr/types.py

@@ -11,6 +11,8 @@ def to_dash(string: str) -> str:
 
 class FieldType(str, Enum):
     text_general = 'text_general'
+    text_gen_sort = 'text_gen_sort'
+    string = 'string'
     pint = 'pint'
 
 

salinic/field.py

@@ -9,6 +9,8 @@ class Field(NamedTuple):
     default: any = None
     multi_value: bool = False
     multi_lang: bool = False
+    # enables grouping by this field
+    group: bool = False
 
 
 class KeywordField(Field):
@@ -23,6 +25,10 @@ class NumericField(Field):
     pass
 
 
+class StringField(Field):
+    pass
+
+
 class IdField(Field):
     pass