arruma o processo de busca textual nos diarios

diarios/search_service.py

@@ -0,0 +1,67 @@
+from elasticsearch_dsl import Q, Search
+from .documents import DiarioOficialDocument
+
+class DiarioOficialSearchService:
+    @staticmethod
+    def search(query, highlight=True, fuzziness=1, page=1, page_size=10, tipos=None, data_inicio=None, data_fim=None):
+        # Configura a busca básica
+        s = DiarioOficialDocument.search().source(excludes=['page_content.content'])
+        
+        # Filtros
+        if tipos:
+            s = s.filter('terms', tipo_nome=tipos)
+        if data_inicio and data_fim:
+            s = s.filter('range', data={'gte': data_inicio, 'lte': data_fim})
+
+        # Query principal com fuzziness e sinônimos
+        main_query = Q(
+            'multi_match',
+            query=query,
+            fields=[
+                'numero^3',         # Maior peso para o número
+                'tipo_nome^2',      # Peso médio para o tipo
+                'page_content.content'  # Peso padrão para o conteúdo
+            ],
+            fuzziness=fuzziness,
+            analyzer='portuguese_synonyms'
+        )
+        s = s.query(main_query)
+
+        # Highlighting
+        if highlight:
+            s = s.highlight(
+                'page_content.content',
+                fragment_size=150,
+                number_of_fragments=3,
+                pre_tags=['<mark>'],
+                post_tags=['</mark>']
+            )
+
+        # Paginação
+        start = (page - 1) * page_size
+        end = start + page_size
+        s = s[start:end]
+
+        # Executa a busca
+        response = s.execute()
+
+        # Formata os resultados
+        results = []
+        for hit in response:
+            result = {
+                'id': hit.id,
+                'numero': hit.numero,
+                'data': hit.data,
+                'link': hit.link,
+                'tipo_nome': hit.tipo_nome,
+                'score': hit.meta.score
+            }
+            if highlight and hasattr(hit.meta, 'highlight'):
+                result['highlights'] = hit.meta.highlight['page_content.content'].to_dict()
+            results.append(result)
+
+        return {
+            'total': response.hits.total.value,
+            'results': results
+        }
+