如何使用UserProfile中的字段扩展UserCreationForm

时间:2012-08-11 23:03:54

标签: django django-models django-forms django-views django-users

我发现这个 post关于如何使用“email”等额外字段扩展UserCreationForm。但是,电子邮件字段已在预构建的用户模型中定义。

我创建了一个额外的模型(称为UserProfile),进一步扩展了Django预构建的User类。如何将我在UserProfile中定义的这些字段显示在UserCreationForm中?

1 个答案:

答案 0 :(得分:9)

为您的UserProfile模型添加适当的字段(不幸的是,使用ModelForm并不太容易避免重复自己),然后创建并保存新的UserProfile实例过度使用save()功能。改编自您链接到的帖子:

from django import forms
from django.contrib.auth.models import User
from django.contrib.auth.forms import UserCreationForm


class UserCreateForm(UserCreationForm):
    job_title = forms.CharField(max_length=100, required=True)
    age = forms.IntegerField(required=True)

    class Meta:
        model = User

    def save(self, commit=True):
        if not commit:
            raise NotImplementedError("Can't create User and UserProfile without database save")
        user = super(UserCreateForm, self).save(commit=True)
        user_profile = UserProfile(user=user, job_title=self.cleaned_data['job_title'], 
            age=self.cleaned_data['age'])
        user_profile.save()
        return user, user_profile