Django chỉ chọn các hàng có giá trị trường trùng lặp

giả sử chúng ta có một mô hình trong django được định nghĩa như sau:

class Literal:
    name = models.CharField(...)
    ...

Trường tên không phải là duy nhất và do đó có thể có các giá trị trùng lặp. Tôi cần hoàn thành nhiệm vụ sau: Chọn tất cả các hàng từ mô hình có ít nhất một giá trị trùng lặp của nametrường.

Tôi biết cách thực hiện bằng cách sử dụng SQL thuần túy (có thể không phải là giải pháp tốt nhất):

select * from literal where name IN (
    select name from literal group by name having count((name)) > 1
);

Vì vậy, có thể chọn điều này bằng cách sử dụng django ORM không? Hay giải pháp SQL tốt hơn?

Thử:

from django.db.models import Count
Literal.objects.values('name')
               .annotate(Count('id')) 
               .order_by()
               .filter(id__count__gt=1)

Điều này gần như bạn có thể đạt được với Django. Vấn đề là điều này sẽ trả về a ValuesQuerySetonly nameand count. Tuy nhiên, sau đó bạn có thể sử dụng điều này để tạo một quy tắc QuerySetbằng cách đưa nó trở lại vào một truy vấn khác:

dupes = Literal.objects.values('name')
                       .annotate(Count('id'))
                       .order_by()
                       .filter(id__count__gt=1)
Literal.objects.filter(name__in=[item['name'] for item in dupes])

Điều này đã bị từ chối dưới dạng một bản chỉnh sửa. Vì vậy, đây là một câu trả lời tốt hơn

dups = (
    Literal.objects.values('name')
    .annotate(count=Count('id'))
    .values('name')
    .order_by()
    .filter(count__gt=1)
)

Điều này sẽ trả về a ValuesQuerySetvới tất cả các tên trùng lặp. Tuy nhiên, sau đó bạn có thể sử dụng nó để xây dựng một thông thường QuerySetbằng cách đưa nó trở lại vào một truy vấn khác. ORM django đủ thông minh để kết hợp chúng thành một truy vấn duy nhất:

Literal.objects.filter(name__in=dups)

Cuộc gọi bổ sung tới .values('name')sau cuộc gọi chú thích trông hơi lạ. Nếu không có điều này, truy vấn con không thành công. Các giá trị phụ đánh lừa ORM chỉ chọn cột tên cho truy vấn con.

thử sử dụng tổng hợp

Literal.objects.values('name').annotate(name_count=Count('name')).exclude(name_count=1)

Trong trường hợp bạn sử dụng PostgreSQL, bạn có thể làm như sau:

from django.contrib.postgres.aggregates import ArrayAgg
from django.db.models import Func, Value

duplicate_ids = (Literal.objects.values('name')
                 .annotate(ids=ArrayAgg('id'))
                 .annotate(c=Func('ids', Value(1), function='array_length'))
                 .filter(c__gt=1)
                 .annotate(ids=Func('ids', function='unnest'))
                 .values_list('ids', flat=True))

Nó dẫn đến truy vấn SQL khá đơn giản này:

SELECT unnest(ARRAY_AGG("app_literal"."id")) AS "ids"
FROM "app_literal"
GROUP BY "app_literal"."name"
HAVING array_length(ARRAY_AGG("app_literal"."id"), 1) > 1

Nếu bạn chỉ muốn kết quả danh sách tên chứ không phải các đối tượng, bạn có thể sử dụng truy vấn sau

repeated_names = Literal.objects.values('name').annotate(Count('id')).order_by().filter(id__count__gt=1).values_list('name', flat='true')