如何在两个字段上查找 django 模型的副本

发布于04月12日

考虑以下Django模型设置:

from django.db import models

Class Foo(models.Model):
    field_foo = models.charField(max_length=20,
                                 null=False,
                                 blank=False,
                                 unique=True,)

Class Bar(models.Model):
    field_bar = models.charField(max_length=20,
                                 null=False,
                                 blank=False,
                                 unique=True,)

Class Foobar(models.Model):
    field_foo = models.ForeignKey(foo,on_delete=models.CASCADE)
    field_bar = models.ForeignKey(bar,on_delete=models.CASCADE)

我希望查找具有相同的field_foo和field_bar值的两行.我可以手动完成此操作，但我想知道Django是否有一个功能可以解决这一问题.我现在这样做的方式是:

for f in Foo.objects.all():
    for b in Bar.objects.all():
        fb = Foobar.objects.filter(foo=f,bar=f)
        if len(fb)>2:
            something='do'

推荐答案

您可以同时使用annotate()和Filter()方法来筛选具有与关系相同的Foo和Bar实例的FooBar实例.

试一试这个

duplicate_foobars = Foobar.objects.values('field_foo', 'field_bar').annotate(
    count=Count('id')
).filter(
    count__gt=1
).values_list('field_foo', 'field_bar')

for field_foo, field_bar in duplicate_foobars:
    print(f"Duplicate Foobar: field_foo={field_foo}, field_bar={field_bar}")

输出

Duplicate Foobar: id=, field_foo=1, field_bar=1

经过测试的模型

class Foo(models.Model):
    field_foo = models.CharField(
        max_length=20,
        null=False,
        blank=False,
        unique=True,
    )


class Bar(models.Model):
    field_bar = models.CharField(
        max_length=20,
        null=False,
        blank=False,
        unique=True,
    )


class Foobar(models.Model):
    field_foo = models.ForeignKey(Foo, on_delete=models.CASCADE)
    field_bar = models.ForeignKey(Bar, on_delete=models.CASCADE)