Django 改进了更新视图？

Question

我有这个 UpDateView 类，我只需要文章的作者就可以编辑博客。我有 CreateView 类的解决方案（使用 def Form_valid），但它不适用于 UpdateView 类 :::

class ArticleUpdateView(LoginRequiredMixin,UpdateView):
    model = models.Article
    template_name = 'article_edit.html'
    fields = ['title','body']
    login_url = 'login'

class ArticleCreateView(LoginRequiredMixin,CreateView):
    model = models.Article
    template_name = 'article_new.html'
    fields = ['title','body',]
    login_url='login'


    def form_valid(self,form):
        form.instance.author = self.request.user
        return super().form_valid(form)

Answer 1

您可以在视图类中覆盖get_object 方法：

class ArticleUpdateView(LoginRequiredMixin,UpdateView):
    model = models.Article
    template_name = 'article_edit.html'
    fields = ['title','body']
    login_url = 'login'

    def get_object(self, *args, **kwargs):
        article = super().get_object(*args, **kwargs)

        if article.author != self.request.user:
            raise PermissionDenied('You should be the author of this article.')

        return article