Django 自定义 UserCreationForm

Question

我在Django里启用了用户认证模块，但是当我使用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

在django 1.10中，我在admin.py文件里写了这些代码，以便在默认的django用户创建表单中添加名字、电子邮件和姓氏这几个字段。

from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
from django.contrib import admin
from django.contrib.auth.models import Group, User

# first unregister the existing useradmin...
admin.site.unregister(User)

class UserAdmin(BaseUserAdmin):
    # The forms to add and change user instances
    # The fields to be used in displaying the User model.
    # These override the definitions on the base UserAdmin
    # that reference specific fields on auth.User.
    list_display = ('username', 'email', 'first_name', 'last_name', 'is_staff')
    fieldsets = (
    (None, {'fields': ('username', 'password')}),
    ('Personal info', {'fields': ('first_name', 'last_name', 'email',)}),
    ('Permissions', {'fields': ('is_active', 'is_staff', 'is_superuser', 'groups', 'user_permissions')}),
    ('Important dates', {'fields': ('last_login', 'date_joined')}),)
    # add_fieldsets is not a standard ModelAdmin attribute. UserAdmin
    # overrides get_fieldsets to use this attribute when creating a user.
    add_fieldsets = (
    (None, {
        'classes': ('wide',),
        'fields': ('username', 'email', 'first_name', 'last_name', 'password1', 'password2')}),)
    list_filter = ('is_staff', 'is_superuser', 'is_active', 'groups')
    search_fields = ('username', 'first_name', 'last_name', 'email')
    ordering = ('username',)
    filter_horizontal = ('groups', 'user_permissions',)

# Now register the new UserAdmin...
admin.site.register(User, UserAdmin)

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

Answer 3

在用户模型里没有叫做 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说的去做，重写保存方法。不过因为用户模型里没有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字段添加一个清理方法，确保输入的fullname只包含两个部分，名字和姓氏，并且这些字符是有效的。

用户模型的参考源代码：

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