Как узнать количество продуктов в категории через django orm?

Question

[Gasoid]

вот есть такие модельки обычные

# coding: utf-8
from django.db import models


class Category(models.Model):
    name = models.CharField('Категория товара', max_length=64)


class Product(models.Model):
    category = models.ForeignKey(Category, verbose_name='Категория')
    name = models.CharField('Наименование товара', max_length=128)
    price = models.DecimalField('Цена единицы, руб.', max_digits=10, decimal_places=2)

как с помощью Django ORM выбрать товары, цена которых больше или равна 100 руб., сгруппировать по категориям и посчитать количество товаров в каждой категории?

делаю так:

Category.objects.filter(product__price__gte=100).annotate(product_name=F('product__name'), product_price=F('product__price'), count=Count('product'))

но выходит, что count 1 всегда

а есть сделать так

Category.objects.filter(product__price__gte=100).annotate(count=Count('product'))

то count показывает то что надо

Comments

[Oscar Django]

ну так и чем не устраивает
> Category.objects.filter(product__price__gte=100).annotate(count=Count('product'))
> то count показывает то что надо
?

[Gasoid]

Oscar Django: тут только категории будут, а не сами продукты)

Answer 1

[Артём Клименко]

рекомендую вам не парится с annotate, у него есть один нюанс(который у вас практически не проявляется) в GROUP BY будут добавлены все поля из модели, если их не ограничивать через only (у вас это [id, name]), что может оказаться крайне медленным, если также вытягиваете related связи они тоже будут в GROUP BY.

я надеюсь, что у вас не старая джанго и пострге в качестве базы

сategories = Category.objects.filter(
    product__price__gte=100
).prefetch_related(
    Prefetch('product_set', queryset=Product.objects.filter(price__gte=100))
).distinct('id')
for category in сategories:
    products = category.product_set.all()
    print(category.name, len(products), products)

будет ровно два запроса

Comments

[greatdvl]

Разве не лучше использовать count()
category.product_set.all()
чем использование len() для подсчета количества???
category.product_set.count()

Answer 2

[Виктор Мельников]

Хотите получить товары, а выборку делаете по категориям. Product.objects.filter ...

Comments

[Gasoid]

и напишите ваш вариант, а так непонятно что вы имеете ввиду

Answer 3

[Oscar Django]

category_products = Category.objects.annotate(c=Count('product')).order_by('-c')
for category in category_products:
     print(category.c, category.product_set.all())

Это не работает у вас?

Comments

[Gasoid]

в одном запросе надо, а так получается еще как минимум +1 запрос

[Oscar Django]

Gasoid запрос к бд?

[Gasoid]

Oscar Django: да,

[Oscar Django]

@Gasoid
type(category_products[0].product_set.all())
Это QuerySet -- запроса нет.

[Gasoid]

Oscar Django: это да, но гы когда продукты нужно будет получить, запрос то выполниться на каждую категорию

[Oscar Django]

Согласен. Может я чего то не понимаю, но вы можете составить sql запрос, который делает то что вы хотите? Без орм.

[Oscar Django]

Гарантировано одним запросом

Product.objects.filter(price__gte=100).values_list('name', 'price', 'category__name')

А дальше средствами python сортировать и считать что надо. Других способов не вижу.