如何在Django中将用户输入传入自定义User模型

你已经搭建好了基础框架，只需要补上几个关键步骤，就能把注册表单里的team选择值关联到自定义User模型里，我来一步步帮你搞定：

1. 给自定义User模型添加Team字段 #

首先得在models.py里给User模型加上对应的team字段，为了避免重复定义选项，建议把选项集合统一放在模型里管理：

# models.py
from django.contrib.auth.models import AbstractUser, BaseUserManager
from django.utils.translation import gettext_lazy as _

class UserManager(BaseUserManager):
    # 保留你原有的UserManager实现代码
    pass

# 统一定义团队选项
TEAM_CHOICES = (
    ('red','red'),
    ('green','green'),
    ('blue','blue'),
    ('orange','orange'),
)

class User(AbstractUser):
    username = None
    email = models.EmailField(_('email address'), unique=True)
    # 添加team字段，和表单选项对应
    team = models.CharField(max_length=20, choices=TEAM_CHOICES, default='red')  # 可设置默认值
    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = []
    objects = UserManager()

修改完模型后，必须执行数据库迁移才能让字段生效：

python manage.py makemigrations
python manage.py migrate

2. 调整RegisterForm（复用模型选项） #

把表单里重复定义的选项删掉，直接引用模型里的TEAM_CHOICES，这样后续修改选项只需要改一处：

# forms.py
from django.contrib.auth.forms import UserCreationForm
from .models import User, TEAM_CHOICES

class RegisterForm(UserCreationForm):
    first_name = forms.CharField(max_length=30,widget=forms.TextInput(attrs={'placeholder':'First Name', 'class': 'form-input'}))
    last_name = forms.CharField(max_length=30,widget=forms.TextInput(attrs={'placeholder':'Last Name', 'class': 'form-input'}))
    email = forms.EmailField(max_length=254, widget=forms.TextInput(attrs={'placeholder':'Email', 'class': 'form-input'}))
    # 直接复用模型里的选项集合
    team = forms.ChoiceField(choices=TEAM_CHOICES)

    class Meta:
        model = User
        fields = ('first_name','last_name','email','team','password1','password2')

3. 在视图中处理表单数据，保存到User模型 #

最后要确保表单提交时，team字段的值能正确赋值给用户对象并保存。这里分两种视图类型给出示例：

函数视图示例： #

# views.py
from django.shortcuts import render, redirect
from .forms import RegisterForm

def register(request):
    if request.method == 'POST':
        form = RegisterForm(request.POST)
        if form.is_valid():
            # 创建用户对象但暂不保存到数据库
            user = form.save(commit=False)
            # 从表单清理后的数据中取出team值，赋值给用户对象
            user.team = form.cleaned_data['team']
            # 正式保存用户
            user.save()
            # 这里可以添加登录或跳转逻辑
            return redirect('home')
    else:
        form = RegisterForm()
    return render(request, 'register.html', {'form': form})

类视图（CreateView）示例： #

# views.py
from django.views.generic.edit import CreateView
from django.urls import reverse_lazy
from .forms import RegisterForm
from .models import User

class RegisterView(CreateView):
    form_class = RegisterForm
    model = User
    success_url = reverse_lazy('home')
    template_name = 'register.html'

    def form_valid(self, form):
        # 重写form_valid方法，先获取未保存的用户对象
        user = form.save(commit=False)
        user.team = form.cleaned_data['team']
        user.save()
        return super().form_valid(form)

做完这些操作后，用户在注册页面选择的团队就会被正确保存到自定义User模型的team字段里了。

内容的提问来源于stack exchange，提问作者Talha Ahmed