Truy vấn ORM Django cho bạn bè của một người dùng

Question

Tôi gặp sự cố khi nhận truy vấn ORM Django để hoạt động chính xác. Tôi có mô hình Hữu nghị này:

class Friendship(models.Model): 
    user1 = models.ForeignKey(User, related_name='friendships1') 
    user2 = models.ForeignKey(User, related_name='friendships2') 
    class Meta: 
     unique_together = ('user1', 'user2',) 

Để tìm những người bạn cho một người dùng nhất định, chúng ta phải kiểm tra user1 và user2 bởi vì chúng tôi không bao giờ có thể chắc chắn những gì bên trong những mối quan hệ họ sẽ về. Vì vậy, để có được tất cả bạn bè cho một người dùng nhất định, tôi sử dụng truy vấn sau:

user = request.user 
User.objects.filter(
    Q(friendships1__user2=user, friendships1__status__in=statuses) | 
    Q(friendships2__user1=user, friendships2__status__in=statuses) 
) 

Điều này dường như tôi thích nó sẽ hoạt động, nhưng không. Nó mang lại cho tôi bản sao. Dưới đây là SQL mà nó tạo ra:

SELECT auth_user.* 
FROM auth_user 
LEFT OUTER JOIN profile_friendship ON (auth_user.id = profile_friendship.user1_id) 
LEFT OUTER JOIN profile_friendship T4 ON (auth_user.id = T4.user2_id) 
WHERE (
    (profile_friendship.status IN ('Accepted') AND profile_friendship.user2_id = 1) 
    OR (T4.user1_id = 1 AND T4.status IN ('Accepted')) 
); 

đây là SQL mà tôi muốn, mà tạo ra kết quả chính xác:

SELECT f1.id as f1id, f2.id AS f2id, u.* 
FROM auth_user u 
LEFT OUTER JOIN profile_friendship f1 ON (u.id = f1.user1_id AND f1.user2_id = 1 AND f1.status IN ('Accepted')) 
LEFT OUTER JOIN profile_friendship f2 ON (u.id = f2.user2_id AND f2.user1_id = 1 AND f2.status IN ('Accepted')) 
WHERE f1.id IS NOT NULL OR f2.id IS NOT NULL 

Tôi biết tôi có thể làm điều này trong một truy vấn liệu, nhưng sau đó tôi don Tôi nghĩ rằng tôi sẽ có thể chuỗi. Có cách nào tốt đẹp sạch để làm điều này mà không đi nguyên?

Answer 1

Giải pháp đơn giản:

user = request.user 
User.objects.filter(
    Q(friendships1__user2=user, friendships1__status__in=statuses) | 
    Q(friendships2__user1=user, friendships2__status__in=statuses) 
).distinct() 

Bất cứ ai biết bất cứ nhược điểm?

Answer 2

Bạn nên sử dụng ManyToManyField.symmetrical

https://docs.djangoproject.com/en/dev/ref/models/fields/#django.db.models.ManyToManyField.symmetrical