Django:带有附加字段的CreateView?

时间:2013-01-20 00:06:25

标签: python django forms django-class-based-views class-based-views

我正在尝试编写一个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都应保存为Project实例的一部分。

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 个答案:

答案 0 :(得分:7)

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

请注意,您无需致电user.save()create_user调用已将用户添加到数据库。您也不必生成随机密码 - 如果passwordNone,则create_user将设置无法使用的密码。

最后,请确保您未在​​user中添加ProjectCreateForm字段。你可能没有,但你的代码说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)