Django通过不同的相关对象注释QuerySet中的几个相同对象

Question

我明白了：

# models

class Building(models.Model):
    ...


class Flat(models.Model):
    building = models.ForeignKey(Building)


class Profile(models.Model):
    flats = models.ManyToManyField(Flat)

# logic

building = Building.objects.create()
flat_1 = Flat.objects.create(building=building)
flat_2 = Flat.objects.create(building=building)

profile = Profile.objects.create()
profile.flats.add(flat_1)
profile.flats.add(flat_2)

profiles = Profile.objects.filter(flats__building=building)  

我获得了profiles 2 个相同的个人资料。我如何用不同的flat 注释它们中的每一个：profiles.first().flat == flat_1 和profiles.last().flat == flat_2？

也许是Subquery()，但如何？

UPD 我在某些 DRF 列表视图中需要这个。 JSON 格式的输出必须类似于：

[
  {
    "profile_id": 1,
    "flat_id": 2
  },
  {
    "profile_id": 1,
    "flat_id": 3
  }
]

Answer 1

要获得该输出，您可以这样做：

data = Profile.objects.all().values('flats', 'id')
return Response(data=data)

在您的 DRF 视图中。

Answer 2

您不必分析实例...

我最后为您的确切需求编写了代码，但首先写了一些可能感兴趣的东西。

在您的代码示例中，您只创建了一个配置文件，我确定您没有获得 2 个相等的配置文件实例，而只有一个。

问题是，如果您有一个只有一个条目的 QuerySet，那么：

profiles.first() == profiles.last()  # True

因为profile.first() 和profiles.last() 是同一个实例。

您应该尝试创建 2 个 Profile 实例：

building = Building.objects.create()

flat_1 = Flat.objects.create(building=building)
flat_2 = Flat.objects.create(building=building)

profile_1 = Profile.objects.create()  # You coud/should use bulk_create here.
profile_2 = Profile.objects.create()

profile_1.flats.add(flat_1)
profile_2.flats.add(flat_2)

然后

profiles = Profile.objects.filter(flats__building=building)  

将返回两个不同的配置文件对象。

另一方面，获取你想要的 JSON ...

按照示例，您发布，按配置文件过滤公寓并获取值（如果您有多个配置文件，这也适用）。

Flat.objects.filter(profile=profile_1).values('profile__id', 'id')

这将返回类似（“id”代表公寓ids）：

[
  {
    "profile__id": 1,
    "id": 1
  },
  {
    "profile__id": 1,
    "id": 3
  }
]

如果您不按个人资料过滤（并且您有多个），您可能会得到如下信息：

[
  {
    "profile__id": 1,
    "id": 1
  },
  {
    "profile__id": 2,
    "id": 3
  },
  {
    "profile__id": 2,
    "id": 4
  },
  ...
]

注释以获得您想要的 EXACT json：

过滤如上图注解，得到想要的值：

Flat.objects.filter(profile=profile_1).annotate(
    flat_id=F('id')
).annotate(
    profile_id=F('profile__id')
).values(
    'profile_id', 'flat_id'
)

会给出你想要的：

[
  {
    "profile_id": 1,
    "flat_id": 2
  },
  {
    "profile_id": 1,
    "flat_id": 3
  }
]

Answer 3

您可以使用正确的序列化程序和正确的注释来做到这一点：

序列化器：

class FlatSerializer(serializers.ModelSerializer):
    class Meta:
        model = Flat
        fields = ('flat_id', 'building_id')

    flat_id = serializers.CharField(read_only=True)

然后我会简单地查询 Flats 而不是配置文件并序列化：

flats = Flat.objects \
    .annotate(flat_id=F('id')) \
    .filter(building=building)

serialized = FlatSerializer(flats, many=True)
print(serialized.data) # [ { flat_id: 1, building_id: 1 }, { flat_id: 2, building_id: 1 } ]

让我知道这是否适合你