|
@@ -1,19 +1,109 @@
|
|
|
+from django.core.urlresolvers import reverse
|
|
|
+from django.shortcuts import redirect
|
|
|
from django.template import RequestContext
|
|
|
+from django.utils import timezone
|
|
|
+from django.utils.translation import ugettext as _
|
|
|
from haystack.query import RelatedSearchQuerySet
|
|
|
-from misago.models import Post
|
|
|
+from misago.decorators import block_crawlers
|
|
|
+from misago.forms import FormFields
|
|
|
+from misago.models import Forum, Thread, Post, User
|
|
|
+from misago.search import SearchException
|
|
|
+from misago.apps.errors import error403, error404
|
|
|
+from misago.apps.profiles.views import list as users_list
|
|
|
+from misago.apps.search.forms import QuickSearchForm
|
|
|
|
|
|
-def do_search(request):
|
|
|
- query = request.POST.get('search_query')
|
|
|
- sqs = RelatedSearchQuerySet().auto_query(query).order_by('-id').load_all()
|
|
|
- sqs = sqs.load_all_queryset(Post, Post.objects.all().select_related('thread', 'forum'))
|
|
|
+@block_crawlers
|
|
|
+def search(request):
|
|
|
+ queryset = Post.objects.filter(forum__in=Forum.objects.readable_forums(request.acl))
|
|
|
+ return do_search(request, queryset)
|
|
|
|
|
|
+
|
|
|
+def do_search(request, queryset, search_route=None):
|
|
|
+ if not request.acl.search.can_search():
|
|
|
+ return error403(request, _("You don't have permission to search community."))
|
|
|
+
|
|
|
+ search_route = search_route or 'search'
|
|
|
+
|
|
|
+ if request.method != "POST":
|
|
|
+ form = QuickSearchForm(request=request)
|
|
|
+ return request.theme.render_to_response('search/home.html',
|
|
|
+ {
|
|
|
+ 'form': FormFields(form),
|
|
|
+ 'search_route': search_route,
|
|
|
+ 'search_result': request.session.get('%s_result' % search_route),
|
|
|
+ 'disable_search': True,
|
|
|
+ },
|
|
|
+ context_instance=RequestContext(request))
|
|
|
+
|
|
|
+ try:
|
|
|
+ form = QuickSearchForm(request.POST, request=request)
|
|
|
+ if form.is_valid():
|
|
|
+ if form.mode == 'forum':
|
|
|
+ jump_to = Forum.objects.forum_by_name(form.target, request.acl)
|
|
|
+ if jump_to:
|
|
|
+ if jump_to.level == 1:
|
|
|
+ return redirect(reverse('index') + ('#%s' % jump_to.slug))
|
|
|
+ return redirect(jump_to.url)
|
|
|
+ else:
|
|
|
+ raise SearchException(_('Forum "%(forum)s" could not be found.') % {'forum': form.target})
|
|
|
+ if form.mode == 'user':
|
|
|
+ request.POST = request.POST.copy()
|
|
|
+ request.POST['username'] = form.target
|
|
|
+ return users_list(request)
|
|
|
+ sqs = RelatedSearchQuerySet().auto_query(form.cleaned_data['search_query']).order_by('-id').load_all()
|
|
|
+ sqs = sqs.load_all_queryset(Post, queryset.select_related('thread', 'forum'))[:24]
|
|
|
+ request.user.last_search = timezone.now()
|
|
|
+ request.user.save(force_update=True)
|
|
|
+ if not sqs:
|
|
|
+ raise SearchException(_("Search returned no results. Change search query and try again."))
|
|
|
+ request.session['%s_result' % search_route] = {
|
|
|
+ 'search_query': form.cleaned_data['search_query'],
|
|
|
+ 'search_results': [p.object.pk for p in sqs],
|
|
|
+ }
|
|
|
+ return redirect(reverse('%s_results' % search_route))
|
|
|
+ else:
|
|
|
+ raise SearchException(form.errors['search_query'][0])
|
|
|
+ except SearchException as e:
|
|
|
+ return request.theme.render_to_response('search/error.html',
|
|
|
+ {
|
|
|
+ 'form': FormFields(form),
|
|
|
+ 'search_route': search_route,
|
|
|
+ 'message': unicode(e),
|
|
|
+ 'disable_search': True,
|
|
|
+ },
|
|
|
+ context_instance=RequestContext(request))
|
|
|
+
|
|
|
+
|
|
|
+@block_crawlers
|
|
|
+def show_results(request):
|
|
|
+ return results(request)
|
|
|
+
|
|
|
+
|
|
|
+def results(request, search_route=None):
|
|
|
+ if not request.acl.search.can_search():
|
|
|
+ return error403(request, _("You don't have permission to search community."))
|
|
|
+
|
|
|
+ search_route = search_route or 'search'
|
|
|
+ result = request.session.get('%s_result' % search_route)
|
|
|
+ if not result:
|
|
|
+ form = QuickSearchForm(request=request)
|
|
|
+ return request.theme.render_to_response('search/error.html',
|
|
|
+ {
|
|
|
+ 'form': FormFields(form),
|
|
|
+ 'search_route': search_route,
|
|
|
+ 'message': _("No search results were found."),
|
|
|
+ 'disable_search': True,
|
|
|
+ },
|
|
|
+ context_instance=RequestContext(request))
|
|
|
+
|
|
|
+ form = QuickSearchForm(request=request, initial={'search_query': result['search_query']})
|
|
|
return request.theme.render_to_response('search/results.html',
|
|
|
{
|
|
|
- 'search_phrase': query,
|
|
|
- 'results': sqs,
|
|
|
+ 'form': FormFields(form),
|
|
|
+ 'search_route': search_route,
|
|
|
+ 'search_query': result['search_query'],
|
|
|
+ 'results': Post.objects.filter(id__in=result['search_results']).select_related('thread', 'forum', 'user'),
|
|
|
+ 'disable_search': True,
|
|
|
},
|
|
|
context_instance=RequestContext(request))
|
|
|
|
|
|
-
|
|
|
-def show_sesults(request):
|
|
|
- pass
|