Django 自定义 Django 中 UserCreationForm 的外观

在本文中，我们将介绍如何自定义 Django 中 UserCreationForm 的外观。UserCreationForm 是 Django 默认提供的用于创建用户的表单，它包含用户名、密码和确认密码等字段。然而，在实际开发中，我们可能需要根据项目需求对 UserCreationForm 进行一些自定义，例如添加额外的字段或修改字段的展示方式。

阅读更多：Django 教程

自定义 UserCreationForm

我们可以通过继承 UserCreationForm 类来自定义表单。首先，我们需要导入所需的类和函数：

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

然后，我们创建一个新的表单类，并继承自 UserCreationForm：

class CustomUserCreationForm(UserCreationForm):
    # 添加自定义字段
    email = forms.EmailField(required=True)

    class Meta(UserCreationForm.Meta):
        model = User
        fields = ('username', 'email', 'password1', 'password2')

在上述示例中，我们添加了一个邮箱字段，并在表单的 Meta 类中指定了要显示的字段顺序。

修改字段展示方式

UserCreationForm 默认使用 Django 的内置小部件（widgets）来展示字段。如果我们想要修改字段的展示方式，可以通过覆盖表单类中相关字段的 widget 属性来实现。

以下示例展示了如何将用户名的展示方式修改为 EmailInput 小部件：

from django.forms.widgets import EmailInput

class CustomUserCreationForm(UserCreationForm):
    email = forms.EmailField(required=True)

    class Meta(UserCreationForm.Meta):
        model = User
        fields = ('username', 'email', 'password1', 'password2')

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        # 修改用户名的展示方式为 EmailInput
        self.fields['username'].widget = EmailInput()

在上述示例中，我们在表单的初始化方法中修改了 username 字段的展示方式为 EmailInput。

添加表单验证规则

除了修改字段的展示方式，我们还可以添加表单验证规则来确保用户输入的数据符合要求。以下示例展示了如何添加对密码强度的验证规则：

from django.core.exceptions import ValidationError
from django.utils.translation import gettext as _

class CustomUserCreationForm(UserCreationForm):
    email = forms.EmailField(required=True)

    class Meta(UserCreationForm.Meta):
        model = User
        fields = ('username', 'email', 'password1', 'password2')

    def clean_password1(self):
        password1 = self.cleaned_data.get("password1")
        if len(password1) < 8:
            raise ValidationError(
                _("Password must be at least 8 characters long.")
            )
        return password1

在上述示例中，我们定义了一个名为 clean_password1 的方法，并在该方法中对密码长度进行了验证。如果密码长度小于8个字符，将抛出一个验证错误。

使用自定义 UserCreationForm

在完成自定义 UserCreationForm 后，我们可以在视图中使用它来创建用户。以下是一个简单的示例：

from django.shortcuts import render, redirect
from .forms import CustomUserCreationForm

def register(request):
    if request.method == 'POST':
        form = CustomUserCreationForm(request.POST)
        if form.is_valid():
            form.save()
            return redirect('home')  # 注册成功后跳转到首页
    else:
        form = CustomUserCreationForm()
    return render(request, 'register.html', {'form': form})

在上述示例中，我们通过 POST 请求提交表单数据，并在视图中判断表单是否有效，然后进行保存操作。如果注册成功，我们将用户重定向到首页。如果请求方式是 GET，则直接渲染带有自定义表单的注册页面。