File size: 1,967 Bytes
146bbac d456722 146bbac d456722 146bbac d456722 146bbac d456722 146bbac d456722 146bbac d456722 146bbac | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
from __future__ import annotations
def build_aggs(config):
aggs={}
for ff in config.facet_fields:
name=getattr(ff,"field",None) or getattr(ff,"dest",None); size=getattr(ff,"facet_size",10); key=name.replace(".","_")
if getattr(ff,"facet_histogram",False): aggs[f"por_{key}"]={"date_histogram":{"field":name,"calendar_interval":"year","min_doc_count":1}}
else:
af=f"{name}.raw" if ff.es_type=="keyword_text" else name; aggs[f"por_{key}"]={"terms":{"field":af,"size":size}}
return aggs
def build_search_query(q,filters=None,page=1,size=10,com_facets=True,config=None):
fields=config.search_fields_es if config else ["*"]; excl=config.source_excludes if config else []
fc=[]
for campo,valor in (filters or {}).items():
fd=next((f for f in (config.campos_filter if config else []) if getattr(f,"field",None)==campo or getattr(f,"dest",None)==campo),None)
fc.append({"term":{f"{campo}.raw":valor}} if fd and fd.es_type=="keyword_text" else {"term":{campo:valor}})
qb={"from":(page-1)*size,"size":size,"query":{"bool":{"must":{"multi_match":{"query":q,"fields":fields,"type":"best_fields","fuzziness":"AUTO","prefix_length":2}},"filter":fc}},"_source":{"excludes":excl} if excl else {}}
if config:
hl={sf.field:{"number_of_fragments":2,"fragment_size":150} for sf in config.search_fields if any(fd.field==sf.field and fd.es_type in("text","keyword_text") for fd in config.campos_filter)}
if hl: qb["highlight"]={"fields":hl,"pre_tags":["<mark>"],"post_tags":["</mark>"]}
if com_facets and config: qb["aggs"]=build_aggs(config)
return qb
def build_autocomplete_query(q,size,config):
sf=config.suggest_fields or ([config.search_fields[0].field] if config.search_fields else [])
sh=[{"match":{f"{f}.autocomplete":{"query":q,"boost":2.0 if i==0 else 1.0}}} for i,f in enumerate(sf[:3])]
return {"size":size,"_source":sf[:3],"query":{"bool":{"should":sh}}}
|