Django-如何将DeleteView限制为对象所有者

时间:2018-09-11 12:35:58

标签: python django view

我正在尝试实现DeleteView并防止如果用户不是所有者则删除模型。我确定这是一项微不足道的任务,但我无法找到明确的答案。到目前为止,我可以看到我的视图,但是当前每个人都可以删除该对象。

这是我的代码:

views.py:

@method_decorator(login_required, name='dispatch')
class RestaurantDeleteView(DeleteView):
    model = Restaurant
    template_name = 'restaurant/delete_form.html'
    success_url = '/'

models.py:

class Restaurant(models.Model):
    owner = models.ForeignKey(User, on_delete=models.CASCADE)
    name = models.CharField(max_length=200)
    category = models.CharField(max_length=200)
    description = models.TextField()
    capacity = models.IntegerField(default=0)

def get_absolute_url(self):
    return reverse('detail', kwargs={'pk': self.pk})

提前谢谢!

1 个答案:

答案 0 :(得分:0)

这可能不是最好的方法,但它说明了这一点。

在您的views.py中,您可以访问self.get_object()以获取参考的模型实例,然后检查是否为instance.owner == self.request.user。查看源代码,并尝试尽可能忠实于原始代码。您所需要的只是条件检查。

类似这样的东西:

@method_decorator(login_required, name='dispatch')
class FooDeleteView(DeleteView):
    model = Foo
    success_url = reverse_lazy('index')

    def get(self, request, *args, **kwargs):
        self.object = self.get_object()
        if self.object.owner != self.request.user:
            return redirect(self.success_url)
        return super().post(request, *args, **kwargs)

如果您好奇并想了解更多信息,可以import pdb并将pdb.set_trace()放在方法的第一行,然后探究self是什么以及可用的方法对此。进入方法调用后,可以通过调用dir(self)来完成此操作。您会对发现的事物感到惊讶。