Django:如何在CreateView中添加额外的字段?

6

我正在尝试编写一个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' object has no attribute 'email'

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

1个回答

9
您会收到错误信息“'Solution'对象没有属性'email'”,因为form.email无效。验证后的数据永远不会作为表单或模型表单的属性存在。当表格(包括模型表格)是有效的时,成功验证的数据可在form.cleaned_data字典中获得。
请注意,您不需要调用user.save()create_user调用已将用户添加到数据库中。您也不必生成随机密码--如果passwordNone,那么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)

只是写这个评论,提醒您可以使用get_or_create方法。 - Aymane Shuichi
2
@AymaneShuichi 不,我们不能在这里使用 get_or_create - 我们正在使用 create_user 方法以便正确处理密码。 - Alasdair

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接