如何在Django中用OneToOneField扩展AbstractBaseUser并实现分类型用户注册

我来帮你搞定这个分类型用户注册的需求，结合你已经写好的自定义User模型，咱们一步步来实现：

1. 创建自定义注册表单 #

首先在forms.py里写一个继承自UserCreationForm的表单，它会包含Account模型的必填字段，加上acctype选择，以及学生/教职工各自的专属字段，同时添加验证逻辑确保字段的正确性：

from django import forms
from django.contrib.auth.forms import UserCreationForm
from .models import Account, StudentAccnt, StaffAccnt

class UserRegistrationForm(UserCreationForm):
    # 添加入户类型选择，以及学生、教职工的专属字段（默认非必填，后续验证时根据类型强制要求）
    acctype = forms.ChoiceField(choices=Account.ACCTYPE, required=True)
    reg_number = forms.CharField(max_length=10, required=False, label="Registration Number")
    id_number = forms.CharField(max_length=10, required=False, label="ID Number")

    class Meta:
        model = Account
        # 包含Account模型的必填字段，加上密码字段
        fields = ['username', 'email', 'phone', 'gender', 'acctype', 'password1', 'password2']

    # 表单验证：确保选择对应类型时填写专属字段，且字段唯一
    def clean(self):
        cleaned_data = super().clean()
        selected_type = cleaned_data.get('acctype')
        reg_num = cleaned_data.get('reg_number')
        id_num = cleaned_data.get('id_number')

        # 学生必须填学号，教职工必须填工号/身份证号
        if selected_type == 'Student' and not reg_num:
            raise forms.ValidationError("Students must provide a Registration Number.")
        if selected_type == 'Staff' and not id_num:
            raise forms.ValidationError("Staff must provide an ID Number.")
        
        # 验证学号和工号的唯一性
        if selected_type == 'Student' and StudentAccnt.objects.filter(reg_number=reg_num).exists():
            raise forms.ValidationError("This Registration Number is already in use.")
        if selected_type == 'Staff' and StaffAccnt.objects.filter(id_number=id_num).exists():
            raise forms.ValidationError("This ID Number is already in use.")
        
        return cleaned_data

2. 编写注册视图处理逻辑 #

接下来在views.py里写视图函数，负责处理表单提交：先创建Account用户，再根据选择的acctype创建对应的子模型记录：

from django.shortcuts import render, redirect
from .forms import UserRegistrationForm
from django.contrib.auth import login

def user_register(request):
    if request.method == 'POST':
        form = UserRegistrationForm(request.POST)
        if form.is_valid():
            # 先创建Account实例，但暂时不保存到数据库（commit=False）
            user = form.save(commit=False)
            # 从表单获取用户类型并赋值
            user.acctype = form.cleaned_data['acctype']
            user.save()

            # 根据用户类型创建对应的子模型
            if user.acctype == 'Student':
                StudentAccnt.objects.create(
                    user=user,
                    reg_number=form.cleaned_data['reg_number']
                )
            elif user.acctype == 'Staff':
                StaffAccnt.objects.create(
                    user=user,
                    id_number=form.cleaned_data['id_number']
                )
            
            # 可选：注册完成后自动登录用户
            login(request, user)
            # 跳转到首页或其他你指定的页面
            return redirect('home')
    else:
        # GET请求时初始化空表单
        form = UserRegistrationForm()
    
    return render(request, 'registration/register.html', {'form': form})

3. 优化注册模板（可选但推荐） #

为了提升用户体验，我们可以在模板里用JavaScript实现动态显示字段：选择“Student”时只显示学号输入框，选择“Staff”时只显示工号输入框。创建templates/registration/register.html：

<!DOCTYPE html>
<html>
<head>
    <title>User Registration</title>
</head>
<body>
    <h2>Register</h2>
    <form method="post">
        {% csrf_token %}
        
        <!-- 通用字段 -->
        <div class="form-group">
            {{ form.username.label_tag }}
            {{ form.username }}
            {% if form.username.errors %}
                <div class="error">{{ form.username.errors }}</div>
            {% endif %}
        </div>

        <div class="form-group">
            {{ form.email.label_tag }}
            {{ form.email }}
            {% if form.email.errors %}
                <div class="error">{{ form.email.errors }}</div>
            {% endif %}
        </div>

        <div class="form-group">
            {{ form.phone.label_tag }}
            {{ form.phone }}
            {% if form.phone.errors %}
                <div class="error">{{ form.phone.errors }}</div>
            {% endif %}
        </div>

        <div class="form-group">
            {{ form.gender.label_tag }}
            {{ form.gender }}
            {% if form.gender.errors %}
                <div class="error">{{ form.gender.errors }}</div>
            {% endif %}
        </div>

        <div class="form-group">
            {{ form.acctype.label_tag }}
            {{ form.acctype }}
            {% if form.acctype.errors %}
                <div class="error">{{ form.acctype.errors }}</div>
            {% endif %}
        </div>

        <!-- 动态显示的专属字段 -->
        <div id="student-fields" class="form-group" style="display: none;">
            {{ form.reg_number.label_tag }}
            {{ form.reg_number }}
            {% if form.reg_number.errors %}
                <div class="error">{{ form.reg_number.errors }}</div>
            {% endif %}
        </div>

        <div id="staff-fields" class="form-group" style="display: none;">
            {{ form.id_number.label_tag }}
            {{ form.id_number }}
            {% if form.id_number.errors %}
                <div class="error">{{ form.id_number.errors }}</div>
            {% endif %}
        </div>

        <!-- 密码字段 -->
        <div class="form-group">
            {{ form.password1.label_tag }}
            {{ form.password1 }}
            {% if form.password1.errors %}
                <div class="error">{{ form.password1.errors }}</div>
            {% endif %}
        </div>

        <div class="form-group">
            {{ form.password2.label_tag }}
            {{ form.password2 }}
            {% if form.password2.errors %}
                <div class="error">{{ form.password2.errors }}</div>
            {% endif %}
        </div>

        <button type="submit">Register</button>
    </form>

    <script>
        // 获取DOM元素
        const acctypeSelect = document.getElementById('id_acctype');
        const studentFields = document.getElementById('student-fields');
        const staffFields = document.getElementById('staff-fields');

        // 切换字段显示的函数
        function toggleExclusiveFields() {
            const selectedType = acctypeSelect.value;
            if (selectedType === 'Student') {
                studentFields.style.display = 'block';
                staffFields.style.display = 'none';
            } else if (selectedType === 'Staff') {
                staffFields.style.display = 'block';
                studentFields.style.display = 'none';
            } else {
                studentFields.style.display = 'none';
                staffFields.style.display = 'none';
            }
        }

        // 页面加载时初始化显示状态
        toggleExclusiveFields();
        // 监听选择变化事件
        acctypeSelect.addEventListener('change', toggleExclusiveFields);
    </script>
</body>
</html>

4. 配置URL和验证设置 #

最后，别忘了在urls.py里添加注册路由：

from django.urls import path
from . import views

urlpatterns = [
    path('register/', views.user_register, name='register'),
    # 其他路由...
]

同时确保你的settings.py已经配置了自定义用户模型：

AUTH_USER_MODEL = 'your_app_name.Account'  # 替换成你的实际APP名称

这样就完成了整个流程：用户注册时选择类型，填写对应字段，表单验证通过后，会自动创建Account记录和对应的StudentAccnt/StaffAccnt记录。

内容的提问来源于stack exchange，提问作者Abhiram P Jayan