Phiên bản Django ORM của SQL COUNT (DISTINCT <column>)

Question

Tôi cần điền vào mẫu có bản tóm tắt hoạt động của người dùng trong một hệ thống nhắn tin đơn giản. Đối với mỗi người gửi thư, tôi muốn số lượng thư được gửi và số người nhận riêng biệt.

Dưới đây là một phiên bản đơn giản của mô hình:

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) 

Đây là cách tôi muốn làm điều đó trong SQL:

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

Tôi đã đọc qua tài liệu về tập hợp trong các truy vấn ORM, và mặc dù chú thích() có thể xử lý cột COUNT đầu tiên, nhưng tôi không thấy cách nào để có được kết quả COUNT (DISTINCT) (thậm chí thêm (select = {}) đã không hoạt động, mặc dù có vẻ như nó cần). Điều này có thể được dịch sang một truy vấn ORM Django hay tôi chỉ nên gắn bó với SQL thô?

Answer 1

Bạn thực sự có thể sử dụng riêng biệt và đếm với nhau, như đã thấy trên câu trả lời này: https://stackoverflow.com/a/13145407/237091

Trong trường hợp của bạn:

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

sẽ trở thành:

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