创建时向django-rest-framework ModelViewSet添加其他字段的最佳方法

Question

我有一个带有外键的Book模型给用户(书的所有者):

class Book(models.Model):
    owner = models.ForiegnKey(User)
    ...

我已经为Book创建了一个ModelViewSet,它显示了登录用户拥有的书籍:

class BookViewSet(viewsets.ModelViewSet):
    model = Book
    serializer_class = BookSerializer
    def get_queryset(self):
        return Book.objects.filter(owner=self.request.user)

现在要创建一本新书,我想用request.user保存用户字段,而不是从其他客户端发送数据(为了更高的安全性).例如:

def create(self, request, *args, **kwargs):
    request.DATA['user'] = request.user
    ... (some code to create new Book from request.DATA using serialize class)

但是我收到了这个错误:这个QueryDict实例是不可变的.(表示request.DATA是一个不可变的QueryDict,无法更改)

在使用django rest框架创建对象时,您是否知道添加其他字段的更好方法？

Answer 1

更新:从v3开始,您需要这样做:

def perform_create(self, serializer):
    serializer.save(owner=self.request.user)

原则保持不变.

您希望将owner图书序列化程序的字段设置为只读,然后设置与用户的关联pre_save().

就像是:

def pre_save(self, obj):
    obj.owner = self.request.user

请参阅"将代码段与用户关联"的教程部分.

我希望有所帮助.

Answer 2

对于它的价值,这在Django REST Framework 3中已经发生了变化.现在有一个perform_create()替换上一个答案中建议的旧版pre_save()和post_save()钩子版.例如:

from rest_framework import viewsets

class MyViewSet(viewsets.ModelViewSet):
    queryset = MyModel.objects.all()
    serializer_class = MyModelSerializer

    def perform_create(self, serializer):
        serializer.save(owner=self.request.user)

希望这能节省一些时间.