Django ORM wersja SQL COUNT (DISTINCT)

Question

Muszę wypełnić szablon podsumowaniem aktywności użytkownika w prostym systemie przesyłania wiadomości. Dla każdego nadawcy wiadomości chcę liczbę wysłanych wiadomości i liczbę różnych odbiorców.

Oto uproszczona wersja modelu:

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) 

Oto jak zrobiłbym to w SQL:

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

Czytałem poprzez dokumentację na agregację w zapytaniach ORM, i chociaż funkcja annotate() może obsłużyć pierwszą kolumnę COUNT, nie widzę sposobu na uzyskanie wyniku COUNT (DISTINCT) (nawet dodatkowy (select = {}) nie działa, chociaż wydaje się, że powinien). Czy można to przetłumaczyć na zapytanie ORM Django, czy powinienem trzymać się surowego kodu SQL?

Answer 1

Można rzeczywiście wykorzystywać odrębne i liczyć razem, jak widać na tej odpowiedzi: https://stackoverflow.com/a/13145407/237091

W twoim przypadku:

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

staną:

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())