Django-Äquivalent für Zählung und Gruppierung nach

Ich habe ein Modell, das so aussieht:

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

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

Ich möchte die Anzahl (nur die Anzahl) der Elemente für jede Kategorie auswählen, daher wäre dies in SQL so einfach:

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

Gibt es ein Äquivalent zu diesem "Django-Weg"? Oder ist einfaches SQL die einzige Option? Ich bin mit der count () -Methode in Django vertraut , sehe jedoch nicht, wie group by dort passen würde.

Wie ich gerade herausgefunden habe, geht hier Folgendes mit der Django 1.1-Aggregations-API vor:

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

( Update : Die vollständige Unterstützung der ORM-Aggregation ist jetzt in Django 1.1 enthalten . Gemäß der folgenden Warnung zur Verwendung privater APIs funktioniert die hier dokumentierte Methode in Post-1.1-Versionen von Django nicht mehr. Ich habe nicht nachgeforscht, um herauszufinden, warum; Wenn Sie 1.1 oder höher verwenden, sollten Sie trotzdem die echte Aggregations-API verwenden .)

Die Unterstützung für die Kernaggregation war bereits in Version 1.0 vorhanden. Es ist nur undokumentiert, nicht unterstützt und verfügt noch nicht über eine benutzerfreundliche API. Aber so können Sie es trotzdem verwenden, bis 1.1 eintrifft (auf eigenes Risiko und in vollem Wissen, dass das Attribut query.group_by nicht Teil einer öffentlichen API ist und sich ändern könnte):

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

Wenn Sie dann über query_set iterieren, ist jeder zurückgegebene Wert ein Wörterbuch mit einem "Kategorie" -Schlüssel und einem "Zähl" -Schlüssel.

Sie müssen hier nicht nach Anzahl bestellen, dies ist nur enthalten, um zu demonstrieren, wie es gemacht wird (es muss im Aufruf von .extra () gemacht werden, nicht anderswo in der Queryset-Konstruktionskette). Sie können auch genauso gut count (id) anstelle von count (1) sagen, aber letzteres ist möglicherweise effizienter.

Beachten Sie auch, dass beim Festlegen von .query.group_by die Werte tatsächliche DB-Spaltennamen ('category_id') und keine Django-Feldnamen ('category') sein müssen. Dies liegt daran, dass Sie die internen Abfragen auf einer Ebene optimieren, auf der alles in DB-Begriffen und nicht in Django-Begriffen angegeben ist.

Da ich ein wenig verwirrt darüber war, wie das Gruppieren in Django 1.1 funktioniert, dachte ich, ich würde hier näher erläutern, wie genau Sie es verwenden. Um zu wiederholen, was Michael gesagt hat:

Wie ich gerade herausgefunden habe, geht hier Folgendes mit der Django 1.1-Aggregations-API vor:

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

Beachten Sie auch, dass Sie müssen from django.db.models import Count!

Dadurch werden nur die Kategorien ausgewählt und anschließend eine aufgerufene Anmerkung hinzugefügt category__count. Abhängig von der Standardreihenfolge ist dies möglicherweise alles, wascategory Sie benötigen. Wenn die Standardreihenfolge jedoch ein anderes Feld verwendet, funktioniert dies nicht . Der Grund dafür ist, dass die für die Bestellung erforderlichen Felder ebenfalls ausgewählt sind und jede Zeile eindeutig machen, sodass Sie nicht gruppiert werden, wie Sie es möchten. Eine schnelle Möglichkeit, dies zu beheben, besteht darin, die Bestellung zurückzusetzen:

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

Dies sollte genau die gewünschten Ergebnisse liefern. So legen Sie den Namen der Anmerkung fest:

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

Dann wird mycountin den Ergebnissen eine Anmerkung aufgerufen .

Alles andere an der Gruppierung war für mich sehr einfach. Weitere Informationen finden Sie in der Django-Aggregations-API .

Wie ist das? (Anders als langsam.)

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

Es hat den Vorteil, dass es kurz ist, auch wenn es viele Zeilen abruft.

Bearbeiten.

Die Version mit einer Abfrage. Übrigens ist dies oft schneller als SELECT COUNT (*) in der Datenbank. Probieren Sie es aus, um zu sehen.

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