Еквівалент Django для підрахунку та групування за

У мене є модель, яка виглядає так:

class Category(models.Model):
    name = models.CharField(max_length=60)

class Item(models.Model):
    name = models.CharField(max_length=60)
    category = models.ForeignKey(Category)

Я хочу вибрати кількість (лише кількість) елементів для кожної категорії, тому в SQL це було б так просто, як це:

select category_id, count(id) from item group by category_id

Чи є еквівалент виконання цього "шляху Джанго"? Або звичайний SQL - єдиний варіант? Я знайомий з методом count () у Django, однак не бачу, як би там вмістилася група за .

Ось, як я щойно виявив, ось як це зробити за допомогою агрегаційного API Django 1.1:

from django.db.models import Count
theanswer = Item.objects.values('category').annotate(Count('category'))

( Оновлення : Повна підтримка агрегації ORM тепер включена в Django 1.1 . Згідно з наведеним нижче застереженням про використання приватних API, метод, задокументований тут, більше не працює у версіях Django після 1.1. Я не розбирався, щоб з’ясувати, чому; якщо ви використовуєте 1.1 або пізнішу версію, ви все одно повинні використовувати реальний API агрегування .)

Основна підтримка агрегації була вже в 1.0; він просто недокументований, не підтримується і ще не має дружнього API поверх нього. Але ось як ви можете використовувати його в будь-якому випадку до появи 1.1 (на свій страх і ризик, і знаючи, що атрибут query.group_by не є частиною загальнодоступного API і може змінитися):

query_set = Item.objects.extra(select={'count': 'count(1)'}, 
                               order_by=['-count']).values('count', 'category')
query_set.query.group_by = ['category_id']

Якщо потім виконати ітерацію над query_set, кожне повернене значення буде словником із ключем "категорія" та ключем "count".

Тут вам не потрібно замовляти за -count, це просто включено, щоб продемонструвати, як це робиться (це має бути зроблено у виклику .extra (), а не деінде в ланцюжку побудови набору запитів). Крім того, ви можете так само сказати count (id) замість count (1), але останній може бути більш ефективним.

Також зауважте, що при встановленні .query.group_by значеннями повинні бути фактичні імена стовпців БД ('category_id'), а не імена полів Django ('категорія'). Це пов’язано з тим, що ви налаштовуєте внутрішні елементи запитів на рівні, де все в термінах БД, а не в термінах Django.

Оскільки я був трохи заплутаний щодо того, як працює групування в Django 1.1, я подумав, що я детальніше розкажу тут, як саме ви використовуєте його. По-перше, повторити те, що сказав Майкл:

Ось, як я щойно виявив, ось як це зробити за допомогою агрегаційного API Django 1.1:

from django.db.models import Count
theanswer = Item.objects.values('category').annotate(Count('category'))

Також зверніть увагу, що вам потрібно from django.db.models import Count!

Буде вибрано лише категорії, а потім додано анотацію з назвою category__count. Залежно від замовлення за замовчуванням, це може бути все, що вам потрібно, але якщо в порядку замовлення за замовчуванням використовується інше поле, ніж categoryце не буде працювати . Причиною цього є те, що поля, необхідні для впорядкування, також вибираються і роблять кожен рядок унікальним, тому ви не отримуватимете речі, згруповані за вашим бажанням. Один із швидких способів виправити це - скинути порядок:

Item.objects.values('category').annotate(Count('category')).order_by()

Це має дати саме ті результати, яких ви хочете. Щоб встановити назву анотації, ви можете використовувати:

...annotate(mycount = Count('category'))...

Тоді у вас буде анотація, що називається mycountв результатах.

Все інше щодо групування було для мене дуже простим. Для отримання більш детальної інформації обов’язково перегляньте API агрегування Django .

Як це? (Крім повільного.)

counts= [ (c, Item.filter( category=c.id ).count()) for c in Category.objects.all() ]

Ця перевага полягає в тому, що він короткий, навіть якщо він отримує багато рядків.

Редагувати.

Версія одного запиту. До речі, це часто швидше, ніж SELECT COUNT (*) у базі даних. Спробуйте побачити.

counts = defaultdict(int)
for i in Item.objects.all():
    counts[i.category] += 1