如何在 DRF 视图集上设置 cookie？

Question

我想知道如何在 Django rest 框架 ViewSet/ModelViewSet 上设置 cookie。

我读了一些SO帖子说你可以使用Response({serializer.data}) ，但是在检索它可能没问题，但是对于像下面这样的 create() ， Response() 不影响它的后续处理吗？

实际上，当我使用 HttpResponse() 之类的HttpResponse(status=401, reason="you cannot post anymore.")替换下面的raise PermissionDenied()行时，我收到了 500 错误（错误显示AttributeError: 'collections.OrderedDict' object has no attribute 'pk'等）但没有指定的 401 代码。

所以基本上下面解释了我想要做什么，即当用户的帖子计数> = 4时拒绝，否则继续 perform_create() 和下面的 DRF 脚本，附加一个 cookie，告诉浏览器“你有 n 个帖子槽剩下”。

class PostViewSet(viewsets.ModelViewSet):
    ..

  def perform_create(self, serializer):

      role = self.request.user.userproperty.role
      if role == "guest":
          post_count = self.request.user.posts.count()
          if post_count >= 4:
              raise PermissionDenied({"message": "you cannot post anymore."})
          else:
              response = // anyhow gets response here //
              response.set_cookie("notify", "your post slot is " + str(3 - post_count) + "left now.")

      serializer.save(user=self.request.user)

但我不知道如何才能从某事中获得response ，或者调用Response()而不损害后果脚本。

我试过了，但没有奏效（正如我所料）：

1. self.request.COOKIES["new_cookie"] = "test."
2. self.response 。

Answer 1

看来我的研究还不够。 我找到了解决办法：

    def create(self, request, *args, **kwargs):
        response = super().create(request, args, kwargs)

        // whatever 

        response.set_cookie("_hoge", "hogee.")

        return response

关键是 perform_create() 不处理 Response()，而 create() 处理； 由于 create() 调用 perform_create()，所以将块移动到 create() 是没有问题的。

参考： 将额外数据添加到 Django Rest 整个结果集的框架结果 - 代码日志