Equivalente Django per count e group by

Ho un modello che assomiglia a questo:

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

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

Voglio selezionare il conteggio (solo il conteggio) degli elementi per ogni categoria, quindi in SQL sarebbe semplice come questo:

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

C'è un equivalente di fare questo "alla maniera di Django"? O il semplice SQL è l'unica opzione? Conosco il metodo count () in Django, tuttavia non vedo come group by si adatterebbe lì.

Ecco, come ho appena scoperto, come farlo con l'API di aggregazione Django 1.1:

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

( Aggiornamento : il supporto completo per l'aggregazione ORM è ora incluso in Django 1.1 . Fedele all'avvertimento di seguito sull'uso di API private, il metodo qui documentato non funziona più nelle versioni successive alla 1.1 di Django. Non ho approfondito per capire perché; se sei su 1.1 o versioni successive dovresti comunque utilizzare l' API di aggregazione reale .

Il supporto per l'aggregazione principale era già presente nella 1.0; è solo non documentato, non supportato e non dispone ancora di un'API amichevole. Ma ecco come puoi usarlo comunque fino all'arrivo della 1.1 (a tuo rischio e con la piena consapevolezza che l'attributo query.group_by non fa parte di un'API pubblica e potrebbe cambiare):

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

Se poi si itera su query_set, ogni valore restituito sarà un dizionario con una chiave "categoria" e una chiave "conteggio".

Non devi ordinare per conteggio qui, è incluso solo per dimostrare come è fatto (deve essere fatto nella chiamata .extra (), non altrove nella catena di costruzione del set di query). Inoltre, potresti anche dire count (id) invece di count (1), ma quest'ultimo potrebbe essere più efficiente.

Nota anche che quando si imposta .query.group_by, i valori devono essere i nomi delle colonne DB effettivi ('category_id') e non i nomi dei campi Django ('category'). Questo perché stai modificando gli interni della query a un livello in cui tutto è in termini DB, non termini Django.

Dato che ero un po 'confuso su come funziona il raggruppamento in Django 1.1, ho pensato di approfondire qui come esattamente lo usi. Innanzitutto, per ripetere ciò che ha detto Michael:

Ecco, come ho appena scoperto, come farlo con l'API di aggregazione Django 1.1:

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

Nota anche che devi from django.db.models import Count!

Questo selezionerà solo le categorie e quindi aggiungerà un'annotazione chiamata category__count. A seconda dell'ordinamento predefinito, potrebbe essere tutto ciò di cui hai bisogno, ma se l'ordinamento predefinito utilizza un campo diverso da categoryquesto non funzionerà . La ragione di ciò è che anche i campi richiesti per l'ordinazione sono selezionati e rendono unica ogni riga, quindi non otterrai cose raggruppate come desideri. Un modo rapido per risolvere questo problema è ripristinare l'ordine:

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

Questo dovrebbe produrre esattamente i risultati desiderati. Per impostare il nome dell'annotazione puoi utilizzare:

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

Quindi avrai un'annotazione chiamata mycountnei risultati.

Tutto il resto sul raggruppamento è stato molto semplice per me. Assicurati di controllare l' API di aggregazione Django per informazioni più dettagliate.

Com'è questo? (Altro che lento.)

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

Ha il vantaggio di essere breve, anche se recupera molte righe.

Modificare.

L'unica versione della query. BTW, questo è spesso più veloce di SELECT COUNT (*) nel database. Provalo per vedere.

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