Question

我是 django 的新手，我正在使用 Django REST 框架开发我的第一个项目。

型号：

class Sticker(Model):
    image = models.ImageField(upload_to=sticker_image_directory_path, null=False)
    title = models.CharField(max_length=150)

    def __str__(self):
        return self.title

class StickerTag(Model):
    name = models.CharField(max_length=100)
    sticker = models.ForeignKey(Sticker, on_delete=models.RESTRICT, related_name='tags')

    def __str__(self):
        return self.name

序列化器：

class StickerSerializer(serializers.ModelSerializer):
    tags = serializers.StringRelatedField(many=True)
    
    class Meta:
        model = models.Sticker
        fields = ['id', 'image', 'title', 'tags', 'countries']

查看：

class StickerView(mixins.ListModelMixin,
                  viewsets.GenericViewSet):
    serializer_class = serializers.StickerSerializer
    queryset = models.Sticker.objects.all()

我正在使用 http://127.0.0.1:8000/stickers/ 调试对 django-debug-toolbar 的调用。我很惊讶地看到，对于每个贴纸实例，Django 都会像这样查询贴纸标签表：

SELECT `stickerapp_stickertag`.`id`,
       `stickerapp_stickertag`.`name`,
       `stickerapp_stickertag`.`sticker_id`
  FROM `stickerapp_stickertag`
 WHERE `stickerapp_stickertag`.`sticker_id` = 1

意味着它正在为每个贴纸一张一张地抓取贴纸标签。因此，如果有 10 个贴纸，它将使用上述查询进行 10 个数据库调用。但我认为这些查询太多，可以通过使用 MYSQL 的“IN”子句来减少，例如：

SELECT * FROM stickerapp_stickertag where sticker_id in (1,2,3,4,5);

我不知道如何在 Django REST 框架中做到这一点。请指导！

Answer 1

您可以.prefetch_related(…) [Django-doc]相关的Tag对象：

class StickerView(mixins.ListModelMixin, viewsets.GenericViewSet):
    serializer_class = serializers.StickerSerializer
    queryset = models.Sticker.objects.prefetch_related('tags')

这将进行两次查询，首先是获取所有 Sticker 对象的查询，然后是针对每个 Sticker 获取相关标签的第二次查询。这是在对 all Sticker 的单个查询中完成的。

Django REST framework 一对多关系和数据库查询

1 个答案: