Django创建自定义UserCreationForm

Question

Django创建自定义UserCreationForm

26

我在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", )

现在表单展示了新的字段，但是它并没有将它们保存到数据库中。

我该如何修复这个问题？

- Joaquin McCoy

3个回答

7

您需要重写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

- manji

4

在django 1.10中，以下是我在admin.py中编写的内容，用于将first_name、email和last_name添加到默认的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)

- Lucas B

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- chandsie · Accepted Answer

在用户模型中没有名为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