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

Question

我的自定义 Manager 方法中的查询延迟执行有问题。 在其中我想通过 model 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 调试工具栏结果：这里

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

如果我错误地使用了“懒惰”这个词，我感到非常抱歉。

Answer 1

问题不在于懒惰，问题在于您为每个PRODUCT_STATUS创建了一个QuerySet 。

您可以使用itertools模块 [Python-doc]的groupby function [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
        }