python - Django:带有附加字段的CreateView？

Question

我正在尝试编写一个 Django CreateView (CBV)，它使用用户电子邮件代替用户 ID，并根据电子邮件确定(或创建)用户。

我的模型不包含任何特殊内容:

class Project(models.Model):
    name = models.CharField(_('Title'), max_length=100,)
    user = models.ForeignKey(User, verbose_name=_('user'),)
    ...

我的 forms.py 将额外的电子邮件字段添加到表单中:

class ProjectCreateForm(forms.ModelForm):
    email = forms.EmailField(required=True, )

    class Meta:
        model = Project
        fields = ('name', ...,)

在我的 views.py 中，我试图确定用户是否存在或是否应该创建。在这两种情况下，用户 ID 都应保存为项目实例的一部分。

class ProjectCreateDetails(CreateView):
    form_class = ProjectCreateForm
    template_name = '...'
    success_url = reverse_lazy('login') 
    model = Project

    def form_valid(self, form):
        try:
            user = User.objects.get(email=form.email)
        except User.DoesNotExist:
            user = User.objects.create_user(form.email, form.email, ''.join([random.choice(string.digits + string.letters) for i in range(0, 10)]))
            user.save()
        form.instance.user = user
        return super(ProjectCreateDetails, self).form_valid(form)

但是我遇到了一个错误，即 'Solution' 对象没有属性 'email'。

我需要切换到 FormView 而不是 CreateView 吗？

Answer 1

您收到错误 'Solution' object has no attribute 'email' 因为 form.email 无效。经过验证的数据永远不能用作表单或模型表单的属性。当表单(包括模型表单)有效时，成功验证的数据在 form.cleaned_data 中可用。词典。

请注意，您不需要调用 user.save()。 create_user 调用已经将用户添加到数据库中。您也不必生成随 secret 码 - 如果 password 为 None，则 create_user将设置一个无法使用的密码。

最后，确保您没有在 ProjectCreateForm 中包含 user 字段。你可能不知道，但你的代码说 fields = ('name', ...,) 所以我不能确定。

将其放在一起，您将获得以下(未经测试的)代码:

def form_valid(self, form):
    try:
        user = User.objects.get(email=form.cleaned_data['email'])
    except User.DoesNotExist:
        user = User.objects.create_user(form.cleaned_data['email'], form.cleaned_data['email']) 
    form.instance.user = user
    return super(ProjectCreateDetails, self).form_valid(form)