Question

我正在尝试获取Json个元素及其相关元素

我有两个表，Service和Room。一种服务有很多房间。我想在有room_id = x的地方获得服务。

模型

class Service(models.Model):
    name = models.CharField(max_length=255, blank=True, null=True)
    class Meta:
        managed = True
        db_table = 'Service'

class Room(models.Model):
    name = models.CharField(max_length=255, blank=True, null=True)
    service = models.ForeignKey(Service, models.DO_NOTHING, blank=True, 
    null=True)
    class Meta:
        managed = True
        db_table = 'Room'

序列化器

class ServiceSerializer(serializers.ModelSerializer):
    room_set = RoomSerializer(many=True, read_only=True)
    class Meta:
       model = Service
       fields = ('name','room_set')
class RoomSerializer(serializers.ModelSerializer):
    class Meta:
        model = Room        
        fields = '__all__'

查看

queryset = Service.objects.filter(room__id=1)
serializer = ServiceSerializer(queryset, many=True)
return JsonResponse(serializer.data, safe=False)

我希望这样的Json：

{
     "name": "Hotel1",
     "room_set": [
     {
           "id": 1,
           "name": "Room1"
      },

但是我明白了：

 {
     "name": "Hotel1",
     "room_set": [
     {
           "id": 1,
           "name": "Room1",
      },
      {
            "id": 2,
            "name": "Room2",
      },
      {
             "id": 3,
             "name": "Room3",
      }
  }

是否有可能获得我期望的json？

Answer 1

您可以通过添加带有过滤查询集的 custom Prefetch object [Django-doc]来修补集合，例如：

from django.db.models import Prefetch

queryset = Service.objects.filter(
    room__id=1
).prefetch_related(
    Prefetch('room_set', queryset=Room.objects.filter(id=1), to_attr='room_set1')
)
serializer = ServiceSerializer(queryset, many=True)
return JsonResponse(serializer.data, safe=False)

，然后让Serializer解析新的相关经理：

class ServiceSerializer(serializers.ModelSerializer):
    room_set = RoomSerializer(many=True, read_only=True, source='room_set1')
    class Meta:
       model = Service
       fields = ('name','room_set1')
class RoomSerializer(serializers.ModelSerializer):
    class Meta:
        model = Room        
        fields = '__all__'

Answer 2

您可以通过序列化器上下文传递房间ID，并在SerializerMethodField()

中进行相应的过滤

class ServiceSerializer(serializers.ModelSerializer):
    rooms = serializers.SerializerMethodField()
    class Meta:
       model = Service
       fields = ('name','rooms')

    get_rooms(self,service):
       room_id = self.get_context('room')
       if room_id:
         queryset = service.rooms_set.filter(id=room_id)
         return RoomSerializer(queryset,many=True).data
       return RoomSerializer(service.rooms_set.all(),many=True).data

serializer = ServiceSerializer(queryset, many=True,context={'room':1})
return JsonResponse(serializer.data, safe=False)

这是通过序列化器进行操作的方法，它是高度可定制的，Willem Van Onsem的答案很简短，但也需要两个与我相同的查询。

一对多的查询集

2 个答案: