在 DRF 中为 PrimaryKeyRelatedField 动态创建的查询集

在 Django REST Framework 中，Primary Key Related Field 是用于表示资源之间关系的常见字段类型之一。它允许我们通过表示资源之间的主键关系来构建资源的序列化器。但是，在某些情况下，我们需要为这些字段动态创建查询集。这篇文章将会介绍如何为 DRF 中的 Primary Key Related Field 动态创建查询集。

创建动态查询集

我们可以通过传递 queryset 参数来动态创建 Primary Key Related Field 的查询集。这个 queryset 可以是一个函数，它将根据请求来返回不同的结果。让我们通过一个例子来看看：

from django.db.models import Q
from rest_framework import serializers
from .models import Book, Author


class AuthorSerializer(serializers.ModelSerializer):
    class Meta:
        model = Author
        fields = '__all__'


class BookSerializer(serializers.ModelSerializer):
    author = serializers.PrimaryKeyRelatedField(
        queryset=lambda: Author.objects.filter(
            Q(is_published=True) | Q(user=self.context['request'].user)
        )
    )

    class Meta:
        model = Book
        fields = '__all__'

在上面的例子中，我们定义了两个序列化器： AuthorSerializer 和 BookSerializer。在 BookSerializer 中，我们通过传递一个 lambda 函数来动态创建 Primary Key Related Field 的查询集。在这个 lambda 函数中，我们使用了 self.context['request'].user 来获取当前请求的用户，并将其用于构建查询集。

解释代码

代码中主要通过serializers.PrimaryKeyRelatedField来实现查询集。该字段表示与另一个模型（模型关系）的主键关系。当在其父级模型中使用时，本字段通常包括在与父级模型的 serializer（用于表示引用）中，而不是与在它中引用的模型的 serializer 中。

然后，我们使用了 lambda 函数来动态创建查询集。通过使用 lambda 函数，我们可以在声明 BookSerializer 类时动态构建查询集，而不是在运行时使用硬编码的查询集。在 lambda 函数中，我们使用 Q 对象和 self.context['request'].user 来根据请求构建查询集。这有助于确保我们只返回当前用户或已发布的图书。最后，我们返回查询集并将其传递给 Primary Key Related Field。

总结

在 DRF 中，Primary Key Related Field 是用于表示资源之间关系的常见字段类型之一。我们可以使用动态查询集为 Primary Key Related Field 构造查询集。通过传递一个 lambda 函数作为查询集参数，我们可以根据请求动态构建查询集。