Django เลือกเฉพาะแถวที่มีค่าฟิลด์ซ้ำกัน

สมมติว่าเรามีโมเดลใน django ที่กำหนดไว้ดังนี้:

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

ฟิลด์ชื่อไม่ซ้ำกันและอาจมีค่าที่ซ้ำกันได้ ฉันต้องการทำงานต่อไปนี้ให้สำเร็จ: เลือกแถวทั้งหมดจากโมเดลที่มีค่าที่ซ้ำกันอย่างน้อยหนึ่งค่าของnameฟิลด์

ฉันรู้วิธีทำโดยใช้ SQL ธรรมดา (อาจไม่ใช่วิธีแก้ปัญหาที่ดีที่สุด):

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

ดังนั้นเป็นไปได้ไหมที่จะเลือกสิ่งนี้โดยใช้ django ORM? หรือโซลูชัน SQL ที่ดีกว่า?

ลอง:

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

ใกล้ที่สุดเท่าที่จะทำได้กับ Django ปัญหาคือสิ่งนี้จะส่งคืน a ที่ValuesQuerySetมีเพียงnameและcount. อย่างไรก็ตามคุณสามารถใช้สิ่งนี้เพื่อสร้างแบบปกติQuerySetโดยป้อนกลับเข้าไปในแบบสอบถามอื่น:

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

สิ่งนี้ถูกปฏิเสธเป็นการแก้ไข ดังนั้นนี่คือคำตอบที่ดีกว่า

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

สิ่งนี้จะส่งคืน a ValuesQuerySetพร้อมกับชื่อที่ซ้ำกันทั้งหมด อย่างไรก็ตามคุณสามารถใช้สิ่งนี้เพื่อสร้างแบบปกติQuerySetโดยป้อนกลับเข้าไปในแบบสอบถามอื่น django ORM ฉลาดพอที่จะรวมสิ่งเหล่านี้ไว้ในแบบสอบถามเดียว:

Literal.objects.filter(name__in=dups)

การเรียกพิเศษ.values('name')หลังจากการเรียกคำอธิบายประกอบดูแปลก ๆ เล็กน้อย หากไม่มีสิ่งนี้แบบสอบถามย่อยจะล้มเหลว ค่าพิเศษจะหลอกให้ ORM เลือกเฉพาะคอลัมน์ชื่อสำหรับเคียวรีย่อย

ลองใช้การรวม

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

ในกรณีที่คุณใช้ PostgreSQL คุณสามารถทำสิ่งนี้ได้:

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

ผลลัพธ์ในแบบสอบถาม SQL ที่ค่อนข้างเรียบง่ายนี้:

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

ถ้าคุณต้องการผลลัพธ์เฉพาะรายการชื่อ แต่ไม่ใช่วัตถุคุณสามารถใช้แบบสอบถามต่อไปนี้

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