Django创建自定义UserCreationForm

Question

我在Django中启用了用户auth模块,但是当我使用时,UserCreationForm他只询问用户名和两个密码/密码确认字段.我还想要电子邮件和全名字段,并设置为必填字段.

我这样做了:

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

class RegisterForm(UserCreationForm):
    email = forms.EmailField(label = "Email")
    fullname = forms.CharField(label = "Full name")

    class Meta:
        model = User
        fields = ("username", "fullname", "email", )

现在表单显示新字段但不保存在数据库中.

我该怎么办？

Answer 1

fullname用户模型中没有调用此类字段.

如果您希望使用原始模型存储名称,则必须将其作为名字和姓氏单独存储.

编辑:如果您只想在表单中使用一个字段并仍使用原始用户模型,请使用以下命令:

你可以这样做:

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

class RegisterForm(UserCreationForm):
    email = forms.EmailField(label = "Email")
    fullname = forms.CharField(label = "First name")

    class Meta:
        model = User
        fields = ("username", "fullname", "email", )

现在你必须做manji所说的并覆盖save方法,但是由于User模型没有fullname字段,它应该如下所示:

def save(self, commit=True):
        user = super(RegisterForm, self).save(commit=False)
        first_name, last_name = self.cleaned_data["fullname"].split()
        user.first_name = first_name
        user.last_name = last_name
        user.email = self.cleaned_data["email"]
        if commit:
            user.save()
        return user

注意:您应该为fullname字段添加一个clean方法,以确保输入的fullname只包含两个部分,第一个名称和姓氏,以及它具有其他有效字符.

用户模型的参考源代码:

http://code.djangoproject.com/browser/django/trunk/django/contrib/auth/models.py#L201

Answer 2

你必须重写UserCreationForm.save()方法：

    def save(self, commit=True):
        user = super(RegisterForm, self).save(commit=False)
        user.fullname = self.cleaned_data["fullname"]
        user.email = self.cleaned_data["email"]
        if commit:
            user.save()
        return user

http://code.djangoproject.com/browser/django/trunk/django/contrib/auth/forms.py#L10