Django: как сделать так, чтобы запрос не был лениво выполнен?
У меня проблема с ленивым выполнением запроса в моем пользовательском методе Manager. В нем я хочу разделить запрос по модели CharField choices и вернуть dict[choice, QuerySet].
model.py part:
...
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 отдельно. Как его можно оптимизировать для выполнения 1 раз?
Мне очень жаль, если я неправильно использую термин "ленивый".
Проблема не в лени, проблема в том, что вы делаете QuerySet
на PRODUCT_STATUS
.
Вы можете составить такой словарь за один проход с помощью функции groupby
[Python-doc] модуля itertools
[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
}