Versione ORM di Django di SQL COUNT (DISTINCT <column>)

Question

Devo compilare un modello con un riepilogo dell'attività dell'utente in un semplice sistema di messaggistica. Per ciascun mittente del messaggio, desidero il numero di messaggi inviati e il numero di destinatari distinti.

Ecco una versione semplificata del modello:

class Message(models.Model): 
    sender = models.ForeignKey(User, related_name='messages_from') 
    recipient = models.ForeignKey(User, related_name='messages_to') 
    timestamp = models.DateTimeField(auto_now_add=True) 

Ecco come lo farei in SQL:

SELECT sender_id, COUNT(id), COUNT(DISTINCT recipient_id) 
    FROM myapp_messages 
    GROUP BY sender_id; 

Ho letto attraverso la documentazione sull'aggregazione nelle query ORM, e sebbene annotate() sia in grado di gestire la prima colonna COUNT, non vedo un modo per ottenere il risultato COUNT (DISTINCT) (anche extra (select = {}) non ha funzionato, anche se sembra come dovrebbe). Questo può essere tradotto in una query ORG di Django o dovrei limitarmi a SQL raw?

Answer 1

È infatti possibile utilizzare distinte e contare insieme, come si è visto su questa risposta: https://stackoverflow.com/a/13145407/237091

Nel tuo caso:

SELECT sender_id, COUNT(id), COUNT(DISTINCT recipient_id) 
FROM myapp_messages 
GROUP BY sender_id; 

sarebbe diventato:

Message.objects.values('sender').annotate(
    message_count=Count('sender'), 
    recipient_count=Count('recipient', distinct=True)) 

Answer 2

from django.db.models import Count 

messages = Message.objects.values('sender').annotate(message_count=Count('sender')) 

for m in messages: 
    m['recipient_count'] = len(Message.objects.filter(sender=m['sender']).\ 
           values_list('recipient', flat=True).distinct())