def get_inner_hits(self, obj):
inner_hits = getattr(obj.meta, 'inner_hits', None)
if inner_hits:
sections = inner_hits.sections or []
domains = inner_hits.domains or []
all_results = itertools.chain(sections, domains)
sorted_results = utils._get_sorted_results(
results=all_results,
source_key='_source',
)
log.debug('[API] Sorted Results: %s', sorted_results)
return sorted_results
def elastic_search(request, project_slug=None):
"""
Global user search on the dashboard.
This is for both the main search and project search.
:param project_slug: Sent when the view is a project search
"""
request_type = None
if project_slug:
queryset = Project.objects.protected(request.user)
project_obj = get_object_or_404(queryset, slug=project_slug)
request_type = request.GET.get('type', 'file')
user_input = UserInput(
query=request.GET.get('q'),
type=request_type or request.GET.get('type', 'project'),
project=project_slug or request.GET.get('project'),
version=request.GET.get('version', LATEST),
language=request.GET.get('language'),
role_name=request.GET.get('role_name'),
index=request.GET.get('index'),
)
search_facets = collections.defaultdict(lambda: ProjectSearch, {
'project': ProjectSearch,
'file': PageSearch,
})
results = None
facets = {}
if user_input.query:
kwargs = {}
for avail_facet in ALL_FACETS:
value = getattr(user_input, avail_facet, None)
if value:
kwargs[avail_facet] = value
search = search_facets[user_input.type](query=user_input.query,
user=request.user,
**kwargs)
results = search[:50].execute()
facets = results.facets
log.info(
LOG_TEMPLATE, {
'user': request.user,
'project': user_input.project or '',
'type': user_input.type or '',
'version': user_input.version or '',
'language': user_input.language or '',
'msg': user_input.query or '',
})
# Make sure our selected facets are displayed even when they return 0 results
for avail_facet in ALL_FACETS:
value = getattr(user_input, avail_facet, None)
if not value or avail_facet not in facets:
continue
if value not in [val[0] for val in facets[avail_facet]]:
facets[avail_facet].insert(0, (value, 0, True))
if results:
# sorting inner_hits (if present)
if user_input.type == 'file':
try:
for result in results:
inner_hits = result.meta.inner_hits
sections = inner_hits.sections or []
domains = inner_hits.domains or []
all_results = itertools.chain(sections, domains)
sorted_results = utils._get_sorted_results(
results=all_results,
source_key='source',
)
result.meta.inner_hits = sorted_results
except Exception:
log.exception('Error while sorting the results (inner_hits).')
log.debug('Search results: %s', results.to_dict())
log.debug('Search facets: %s', results.facets.to_dict())
template_vars = user_input._asdict()
template_vars.update({
'results': results,
'facets': facets,
})
if project_slug:
template_vars.update({'project_obj': project_obj})
return render(
request,
'search/elastic_search.html',
template_vars,
)
