django - 在Django Rest框架中将选择限制为外键

Question

如何限制request.user的图像与节点链接。我希望我可以做类似的事情:

photo = models.ForeignKey(
    Image,
    limit_choices_to={'owner': username},
)

但是request.user而不是username，并且我不想使用本地线程。

models.py

class Node(models.Model):
    owner = models.ForeignKey(User)
    content = models.TextField()
    photo = models.ForeignKey(Image)

class Image(models.Model):
    owner = models.ForeignKey(User)
    file = models.ImageField(upload_to=get_upload_file_name)

serializers.py

class ImageSerializer(serializers.ModelSerializer):
    owner = serializers.Field('owner.username')

    class Meta:
        model = Image
        fields = ('file', 'owner')

class NodeSerializer(serializers.HyperlinkedModelSerializer):

    class Meta:
        model = Node
        fields = ('content', 'photo', 'owner')

Answer 1

我将通过覆盖get_serializer_class来处理此问题，以在运行时动态返回序列化程序类，并在该字段上设置choices选项:

def get_serializer_class(self, ...):
    user = self.request.user
    owner_choices = ...  # However you want to restrict the choices

    class ImageSerializer(serializers.ModelSerializer):
        owner = serializers.Field('owner.username', choices=owner_choices)

        class Meta:
            model = Image
            fields = ('file', 'owner')

    return ImageSerializer