Django equivalente para el recuento y el grupo de

Question

que tiene un modelo que se parece a esto:

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

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

Quiero select count (sólo el recuento) de artículos para cada categoría, por lo que en SQL sería tan simple como esto:

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

¿Hay un equivalente de hacer esto "el camino de Django"? ¿O es simple SQL la única opción? Estoy familiarizado con el método count() en Django, sin embargo, no veo cómo cabría el grupo por.

Answer 1

Aquí, como acabo de descubrir, es cómo hacer esto con el Django 1.1 agregación API:

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

Answer 2

(actualización:.. La agregación de apoyo ORM completa Ahora se incluye en Django 1.1 Fiel a la continuación de advertencia sobre el uso de las API privadas, el método documentado aquí ya no trabaja en la post-1.1 versiones de Django no he cavado en para averiguar por qué, si está en 1.1 o posterior debe utilizar el aggregation API real de todos modos.)

El soporte de agregación de núcleo ya estaba allí en 1.0; simplemente no está documentado, no es compatible y aún no cuenta con una API amigable. Pero aquí es cómo se puede utilizar de todos modos hasta que llega 1.1 (a su propio riesgo, y en pleno conocimiento de que el atributo query.group_by no es parte de una API pública y podría cambiar):

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

Si a continuación iterar sobre query_set, cada valor devuelto será un diccionario con una clave de "categoría" y una clave de "conteo".

No tiene que hacer un pedido por -count aquí, solo se incluye para demostrar cómo se hace (tiene que hacerse en la llamada .extra(), no en otra parte de la cadena de construcción de queryset). Además, podría decir count (id) en lugar de count (1), pero este último puede ser más eficiente.

Tenga en cuenta también que al establecer .query.group_by, los valores deben ser nombres de columna DB reales ('category_id') no nombres de campo Django ('categoría'). Esto se debe a que está ajustando las consultas internas en un nivel donde todo está en términos de DB, no en términos de Django.

Answer 3

¿Cómo es esto? (Aparte lento.)

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

Tiene la ventaja de ser corto, incluso si lo hace traer un montón de registros.

---

Editar.

La única versión de consulta. Por cierto, esto es a menudo más rápido que SELECCIONAR COUNT (*) en la base de datos. Pruébalo para ver.

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

Answer 4

Desde que era un poco confundido acerca de cómo agrupar en Django 1.1 trabajos que pensé que' Elabora aquí cómo exactamente vas sobre usarlo. Primero, para repetir lo que Michael dijo:

Aquí, como acabo de descubrir, es cómo hacer esto con el Django 1.1 agregación API:

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

Tenga en cuenta también que es necesario from django.db.models import Count!

Esto seleccionará solo las categorías y luego agregará una anotación llamada category__count. Según el orden predeterminado, esto puede ser todo lo que necesita, , pero si el pedido predeterminado utiliza un campo que no sea category, esto no funcionará. La razón de esto es que los campos necesarios para realizar el pedido también se seleccionan y hacen que cada fila sea única, por lo que no se agruparán las cosas como se desee. Una forma rápida de solucionar este problema es restablecer el orden:

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

Esto debe producir exactamente los resultados que desea. Para establecer el nombre de la anotación que puede utilizar:

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

, entonces tendrá una anotación llamada mycount en los resultados.

Todo lo demás sobre la agrupación fue muy sencillo para mí. Asegúrese de consultar el Django aggregation API para obtener información más detallada.