Haystack objects in one query

 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
def haystack_monkey_load(sqs):
    """
    it loads all Items for given SearchQuerySet and attaches them as "object" in each SearchResult
    """
    model_dict = {'app.item': Item, 'another_app.comment': Comment}
    for model_key, model in model_dict.items():
        items_id = [sr.pk for sr in sqs if sr.app_label+"."+sr.model_name == model_key]
        items = model.objects.filter(id__in=items_id).select_related()
        for item in items:
            for sr in sqs:
                if item.id == sr.pk and sr.app_label+"."+sr.model_name == model_key:
                    sr.object = item
    return sqs


class SpecialSearchView(SearchView):
    def build_page(self):
        """ for example special SearchView that uses haystack_monkey_load for pagination """
        paginator = paginator(self.results, self.results_per_page)
        try:
            page = paginator.page(self.request.GET.get('page', 1))
            page.object_list = haystack_monkey_load(page.object_list)
        except InvalidPage:
            raise Http404
        return (paginator, page)

More like this

  1. Dynamic query builder with AND/OR by bobwaycott 3 years, 8 months ago
  2. url extension mechanism by showell 4 years, 1 month ago
  3. One step up from __icontains by peterbe 2 years, 8 months ago
  4. Smart {% if %} template tag by SmileyChris 4 years, 2 months ago
  5. plaintext filter by onelson 3 years, 1 month ago

Comments

(Forgotten your password?)