首页 > 解决方案 > 验证实例是否存在 django

问题描述

我正在 DRF 上构建一个 REST Api。我有一个来自 rest_framework.viewsets 的 ModelViewSet 端点。

我有一个帖子和一个评论模型。每条评论都属于一个帖子。所以我定义了 2 个端点,如下所示:

1) router.register(r"posts", views.PostView
2) router.register(r"(?P<pk>[^/.]+)/comments", views.CommentView

它们都继承自 ModelViewSet,因此我可以对它们执行 CRUD 操作。

我对第二个端点有疑问。由于我们为帖子创建评论,因此我从 posts/int:pk/comments 中获得了一个帖子 pk。但问题是当我在此端点上执行 GET 请求时,它会返回所有评论的列表,但我需要那些属于帖子(url 中的 ID)。当我尝试在此端点上发出 POST 请求时,如果帖子不存在,它会引发 DoesNotExist 错误,这是合乎逻辑的。

到目前为止我所做的是:

  1. 重新定义了 get_queryset() 以仅检索属于特定帖子的对象。如果帖子不存在,它会返回一个空列表(尽管我认为它应该引发 404)
  2. 在我的序列化程序中重新定义 validate() 以检查帖子是否存在。如果没有则返回 404

但是当我检查posts/int:pk/comments/int:pk 上的PUT、DELETE、PATCH 方法时,它也不会考虑post 实例,即即使post 不存在,它也会允许这些操作我错了相信。

ModelViewSet 类中是否有一个方法可以在每个请求上触发并检查 post 实例是否存在?这样我就不必为了检查它而重新定义每个 CRUD 方法。

编辑

评论视图

class CommentView(ModelViewSet):
    queryset = Comment.objects.all()
    serializer_class = CommentSerializer
    lookup_url_kwarg = 'pk2'

    def get_queryset(self):
        queryset = Comment.objects.filter(post=self.kwargs.get('pk'))
        return queryset

    def perform_create(self, serializer, **kwargs):
        post = get_object_or_404(Post, pk=self.kwargs.get('pk'))
        serializer.save(author=self.request.user, post=post)

标签: djangodjango-rest-framework

解决方案


也许看看drf-nested-routers package

如果它不能解决您的问题,也许您想要的是标准路由器注册:

router.register(r"posts", views.PostView)
router.register(r"comments", views.CommentView)

然后在列出评论时使用django-filter 之/comments/?post=<id>类的包通过查询参数进行过滤: . 创建评论时,只需在请求正文中发送帖子 ID(不要忘记在 CommentSerializer 中包含“帖子”字段)


推荐阅读