django：如何使查询不被懒惰地执行？

Question

我的自定义管理器方法中的查询延迟执行有问题。在其中我想通过模型 CharField 选择来分离查询并返回 dict[choice, QuerySet]。

model.py 部分：

...

PRODUCT_STATUS = [
    ('pn', 'Запланировано'),
    ('ga', 'В процессе (Ознакомляюсь)'),
    ('rv', 'Повтор'),
    ('ac', 'Завершено'),
    ('ab', 'Брошено'),
    ('pp', 'Отложено'),
]


class ExtendedManager(models.Manager):
    def separated_by_status(self, product_type):
        query = super().get_queryset().all()
        dict_ = {}
        for status in PRODUCT_STATUS:
            dict_.update({status[1]: query.filter(status=status[0]).all()})
        return dict_

...

views.py 部分与经理使用：

...

class ProductListView(ContextMixin, View):
    template_name = 'myList/myList.html'

    def get(self, request: HttpRequest, *args: Any, **kwargs: Any) -> HttpResponse:
        product = kwargs.get('product')
        if product not in {'film', 'game', 'series', 'book'}:
            raise Http404
        context = self.get_context_data()
        context['title'] = f'Список {product}'
        context['dict_queryset'] = Product.objects.separated_by_status(product)
        
        return render(request, self.template_name, context)

...

django 调试工具栏结果： here

问题是查询在管理器中是惰性执行的，但在模板中它已经分别为每个 PRODUCT_STATUS 元素完全执行。 cat 如何优化执行 1 次？

如果我错误地使用了“懒惰”这个词，我深表歉意。

Answer 1

问题不在于懒惰，问题在于你做了一个QuerySet per PRODUCT_STATUS。

您可以使用groupby function [Python-doc] 的itertools module [Python-doc] 单遍制作这样的字典：

from itertools import groupby
from operator import attrgetter

class ExtendedManager(models.Manager):
    def separated_by_status(self, product_type):
        query = super().get_queryset().order_by('status')
        dict_ = {
            k: list(vs)
            for k, vs in groupby(query, attrgetter('status'))
        }
        return {
            status1: dict_.get(status0, [])
            for status0, status1 in PRODUCT_STATUS
        }