在哪里检查 Django CBV 中的 403？

Question

我正在制作一个教初学者的基本应用程序。每个用户都可以写笔记，但我想让用户无法查看或更新其他用户的笔记。

我有以下观点，但我不得不重复自己。

from django.core.exceptions import PermissionDenied

...

class NoteUpdate(LoginRequiredMixin, UpdateView):

    ...

    def get(self, request, *args, **kwargs):
        self.object = self.get_object()

        if self.object.owner != self.request.user:
            raise PermissionDenied

        return super(NoteUpdate, self).get(request, *args, **kwargs)

    def post(self, request, *args, **kwargs):
        self.object = self.get_object()

        if self.object.owner != self.request.user:
            raise PermissionDenied

        return super(NoteUpdate, self).post(request, *args, **kwargs)

我觉得可能有一种方法可以做到这一点而无需重复自己。是的，我可以编写这样的方法并从两者中调用它：

def check_permission(self):
   if self.object.owner != self.request.user:
        raise PermissionDenied 

但我真正的意思是我是否覆盖了错误的方法？有没有更传统的方法来做到这一点？覆盖 .get() 和 .post() 感觉有点奇怪

Answer 1

回答您的问题：覆盖 .get() 和 .post() 很好，因为出于安全和完整性原因，您希望在显示和特别是修改数据之前验证您的 get() 和 post() 视图。现在，如果您想在 get 或 post 中进行重构，有两种简单的方法：

初级（模型方法）：

models.py

class Model(models.Model):
    owner = models.ForeignKey(User)
    ...

    def deny_if_not_owner(self, user):
        if self.owner != user:
            raise PermissionDenied
        return self.owner

views.py

class NoteUpdate(LoginRequiredMixin, UpdateView):
    ...

    def get(self, request, *args, **kwargs):
        self.object = self.get_object()

        self.object.deny_if_not_owner(request.user)

        return super(NoteUpdate, self).get(request, *args, **kwargs)

    def post(self, request, *args, **kwargs):
        self.object = self.get_object()

        self.object.deny_if_not_owner(request.user)

        return super(NoteUpdate, self).post(request, *args, **kwargs)

///////

替代（混合）：

如果您以后看到自己再次使用此验证，创建一个 Mixin 将允许您轻松将此代码添加到许多类中。

class DenyWrongUserMixin(object):
    def get(self):
        if self.object.owner != self.request.user:
            raise PermissionDenied
        return super(DenyWrongUserMixin, self).get(*args, **kwargs)

    def post(self):
        if self.object.owner != self.request.user:
            raise PermissionDenied
        return super(DenyWrongUserMixin, self).post(*args, **kwargs)

然后：

class NoteUpdate(LoginRequiredMixin, DenyWrongUserMixin, UpdateView):

    ...

    def get(self, request, *args, **kwargs):
        ...

    def post(self, request, *args, **kwargs):
        ...

Answer 2

您可以覆盖 get 方法或 get_queryset 方法。如果登录用户不是所有者，get_queryset 将引发 404。

def get_queryset(self):
    qs = super(NoteUpdate, self).get_queryset()
    return qs.filter(owner=self.request.user)

或者你可以重写 get 方法，因为它会先被调用，然后引发 PermissionDenied，所以没有理由也重写 post 方法。

def get(self, request, *args, **kwargs):
    self.object = self.get_object()
    if self.object.owner != self.request.user:
        raise PermissionDenied
    return super(NoteUpdate, self).get(request, *args, **kwargs)

然后你可以创建一个 mixin 并从 mixin 扩展你的视图以避免重复。