Python Django : Use Form Wizard to Edit Model and check if request user is allowed to edit

Question

我使用 from 向导不仅可以创建模型，还可以编辑模型(如下所述: Django Form Wizard to Edit Model )。这工作正常，但现在我需要检查是否允许请求用户进行编辑。在我的模型中，我有一个字段，因此只有所有者才有权编辑:

class Document(models.Model):
...
    owner = models.ForeignKey(User, editable=False)
...

你知道如何做到这一点吗？谢谢!

Answer 1

您可能需要将以下内容添加到向导的 get_form_initial 方法中:

from django.core import exceptions

class DocumentWizard(SessionWizardView):
    # ...

    def get_form_initial(self, step):
        # ... determine document_id
        document = Document.objects.get(id=document_id)
        if self.request.user == document.owner:
            document_dict = model_to_dict(document)
            return document_dict
        else:
            raise exceptions.PermissionDenied