119 lines
4.3 KiB
Python
119 lines
4.3 KiB
Python
from __future__ import absolute_import
|
|
|
|
import urlparse
|
|
import logging
|
|
|
|
from django.shortcuts import render_to_response
|
|
from django.template import RequestContext
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from django.contrib import messages
|
|
from django.conf import settings
|
|
from django.http import HttpResponseRedirect
|
|
from django.core.urlresolvers import reverse
|
|
from django.utils.http import urlencode
|
|
|
|
from .conf.settings import SHOW_OBJECT_TYPE
|
|
from .conf.settings import LIMIT
|
|
from .forms import SearchForm, AdvancedSearchForm
|
|
from .models import RecentSearch
|
|
from .classes import SearchModel
|
|
|
|
logger = logging.getLogger(__name__)
|
|
document_search = SearchModel.get('documents.Document')
|
|
|
|
|
|
def results(request, extra_context=None):
|
|
context = {
|
|
'query_string': request.GET,
|
|
#'hide_header': True,
|
|
'hide_links': True,
|
|
'multi_select_as_buttons': True,
|
|
'search_results_limit': LIMIT,
|
|
}
|
|
|
|
if request.GET:
|
|
# Only do search if there is user input, otherwise just render
|
|
# the template with the extra_context
|
|
|
|
if 'q' in request.GET:
|
|
# Simple query
|
|
logger.debug('simple search')
|
|
query_string = request.GET.get('q', u'').strip()
|
|
model_list, flat_list, shown_result_count, result_count, elapsed_time = document_search.simple_search(query_string)
|
|
else:
|
|
# Advanced search
|
|
logger.debug('advanced search')
|
|
model_list, flat_list, shown_result_count, result_count, elapsed_time = document_search.advanced_search(request.GET)
|
|
|
|
if shown_result_count != result_count:
|
|
title = _(u'results, (showing only %(shown_result_count)s out of %(result_count)s)') % {
|
|
'shown_result_count': shown_result_count,
|
|
'result_count': result_count}
|
|
|
|
else:
|
|
title = _(u'results')
|
|
|
|
# Update the context with the search results
|
|
context.update({
|
|
'found_entries': model_list,
|
|
'object_list': flat_list,
|
|
'title': title,
|
|
'time_delta': elapsed_time,
|
|
})
|
|
|
|
RecentSearch.objects.add_query_for_user(request.user, request.GET, result_count)
|
|
|
|
if extra_context:
|
|
context.update(extra_context)
|
|
|
|
if SHOW_OBJECT_TYPE:
|
|
context.update({'extra_columns':
|
|
[{'name': _(u'type'), 'attribute': lambda x: x._meta.verbose_name[0].upper() + x._meta.verbose_name[1:]}]})
|
|
|
|
return render_to_response('search_results.html', context,
|
|
context_instance=RequestContext(request))
|
|
|
|
|
|
def search(request, advanced=False):
|
|
if advanced:
|
|
form = AdvancedSearchForm(data=request.GET, search_model=document_search)
|
|
return render_to_response('generic_form.html',
|
|
{
|
|
'form': form,
|
|
'title': _(u'advanced search'),
|
|
'form_action': reverse('results'),
|
|
'submit_method': 'GET',
|
|
'search_results_limit': LIMIT,
|
|
'submit_label': _(u'Search'),
|
|
'submit_icon_famfam': 'zoom',
|
|
},
|
|
context_instance=RequestContext(request)
|
|
)
|
|
else:
|
|
if request.GET.get('source') != 'sidebar':
|
|
# Don't include a form a top of the results if the search
|
|
# was originated from the sidebar search form
|
|
extra_context = {
|
|
'submit_label': _(u'Search'),
|
|
'submit_icon_famfam': 'zoom',
|
|
'form_title': _(u'Search'),
|
|
'form_hide_required_text': True,
|
|
}
|
|
if ('q' in request.GET) and request.GET['q'].strip():
|
|
query_string = request.GET['q']
|
|
form = SearchForm(initial={'q': query_string})
|
|
extra_context.update({'form': form})
|
|
return results(request, extra_context=extra_context)
|
|
else:
|
|
form = SearchForm()
|
|
extra_context.update({'form': form})
|
|
return results(request, extra_context=extra_context)
|
|
else:
|
|
# Already has a form with data, go to results
|
|
return results(request)
|
|
|
|
|
|
def search_again(request):
|
|
query = urlparse.urlparse(request.META.get('HTTP_REFERER', u'/')).query
|
|
return HttpResponseRedirect('%s?%s' % (reverse('search_advanced'), query))
|