[django] 장고 뷰에서 둘 이상의 쿼리 세트를 결합하는 방법은 무엇입니까?

QuerySetChain아래 클래스 를 사용할 수 있습니다 . Django의 paginator와 함께 사용하면 COUNT(*)모든 쿼리 세트에 대한 SELECT()쿼리 와 현재 페이지에 레코드가 표시되는 쿼리 세트에 대한 쿼리만으로 데이터베이스에 도달해야 합니다.

체인 쿼리 세트가 모두 동일한 모델을 사용하더라도 일반 뷰와 함께를 template_name=사용할 경우 지정해야합니다 QuerySetChain.

from itertools import islice, chain

class QuerySetChain(object):
    """
    Chains multiple subquerysets (possibly of different models) and behaves as
    one queryset.  Supports minimal methods needed for use with
    django.core.paginator.
    """

    def __init__(self, *subquerysets):
        self.querysets = subquerysets

    def count(self):
        """
        Performs a .count() for all subquerysets and returns the number of
        records as an integer.
        """
        return sum(qs.count() for qs in self.querysets)

    def _clone(self):
        "Returns a clone of this queryset chain"
        return self.__class__(*self.querysets)

    def _all(self):
        "Iterates records in all subquerysets"
        return chain(*self.querysets)

    def __getitem__(self, ndx):
        """
        Retrieves an item or slice from the chained set of results from all
        subquerysets.
        """
        if type(ndx) is slice:
            return list(islice(self._all(), ndx.start, ndx.stop, ndx.step or 1))
        else:
            return islice(self._all(), ndx, ndx+1).next()

귀하의 예에서 사용법은 다음과 같습니다.

pages = Page.objects.filter(Q(title__icontains=cleaned_search_term) |
                            Q(body__icontains=cleaned_search_term))
articles = Article.objects.filter(Q(title__icontains=cleaned_search_term) |
                                  Q(body__icontains=cleaned_search_term) |
                                  Q(tags__icontains=cleaned_search_term))
posts = Post.objects.filter(Q(title__icontains=cleaned_search_term) |
                            Q(body__icontains=cleaned_search_term) |
                            Q(tags__icontains=cleaned_search_term))
matches = QuerySetChain(pages, articles, posts)

그런 다음 예제에서 matches사용한 것처럼 페이지 매김 장치와 함께 사용 result_list하십시오.

이 itertools모듈은 Python 2.3에서 도입되었으므로 Django가 실행되는 모든 Python 버전에서 사용할 수 있어야합니다.

답변

from itertools import chain
result = list(chain(*docs))

DATE_FIELD_MAPPING = {
    Model1: 'date',
    Model2: 'pubdate',
}

def my_key_func(obj):
    return getattr(obj, DATE_FIELD_MAPPING[type(obj)])

And then sorted(chain(Model1.objects.all(), Model2.objects.all()), key=my_key_func)