在django中设置自定义用户注册表单的样式

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

class CustomUserCreationForm(UserCreationForm):
    username = forms.CharField(widget=forms.TextInput(attrs={'class': 'custom-input'}))
    password1 = forms.CharField(widget=forms.PasswordInput(attrs={'class': 'custom-input'}))
    password2 = forms.CharField(widget=forms.PasswordInput(attrs={'class': 'custom-input'}))

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

from .forms import CustomUserCreationForm

def register(request):
    if request.method == 'POST':
        form = CustomUserCreationForm(request.POST)
        if form.is_valid():
            form.save()
            # 处理注册成功的逻辑
            return redirect('success')
    else:
        form = CustomUserCreationForm()
    return render(request, 'register.html', {'form': form})