Django中如何使用自定义PetOwner模型完成用户注册

问题原因 #

你当前使用Django默认的UserCreationForm仅会处理内置User模型的字段，表单保存后也没有额外逻辑创建关联的PetOwner实例，因此注册完成后只会生成User数据，不会同步生成宠物主人记录。

解决方案 #

1. 自定义注册表单（新增PetOwner所需字段） #

首先在你的应用目录下新建forms.py文件，扩展默认注册表单，增加PetOwner模型需要的字段：

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

class PetOwnerSignUpForm(UserCreationForm):
    first_name = forms.CharField(max_length=50, required=True)
    last_name = forms.CharField(max_length=50, required=True)
    email = forms.EmailField(max_length=50, required=True)
    phone_number = forms.CharField(max_length=15, required=False)

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

2. 修改signup视图，新增PetOwner创建逻辑 #

替换原视图中的默认表单为你自定义的表单，在User实例保存完成后，同步创建关联的PetOwner实例：

from django.shortcuts import render, redirect
from django.contrib.auth import authenticate, login
from .forms import PetOwnerSignUpForm
from .models import PetOwner

def signup(request):
    if request.method == "POST":
        form = PetOwnerSignUpForm(request.POST)
        if form.is_valid():
            # 保存User实例
            user = form.save(commit=False)
            user.first_name = form.cleaned_data['first_name']
            user.last_name = form.cleaned_data['last_name']
            user.email = form.cleaned_data['email']
            user.save()
            # 同步创建PetOwner实例
            PetOwner.objects.create(
                user=user,
                first_name=form.cleaned_data['first_name'],
                last_name=form.cleaned_data['last_name'],
                email=form.cleaned_data['email'],
                phone_number=form.cleaned_data['phone_number']
            )
            # 原有登录逻辑保持不变
            username = form.cleaned_data.get("username")
            password = form.cleaned_data.get("password1")
            user = authenticate(username=username, password=password)
            login(request, user)
            return redirect("/")
        else:
            return render(request, "signup.html", {"form": form})
    else:
        form = PetOwnerSignUpForm()
        return render(request, "signup.html", {"form": form})

可选：用信号实现自动创建（更解耦） #

如果你希望所有新建的普通User都自动关联PetOwner实例，可以用Django的信号机制，不用在每个创建User的地方重复写逻辑：

1. 在应用目录下新建signals.py：

from django.db.models.signals import post_save
from django.dispatch import receiver
from django.contrib.auth.models import User
from .models import PetOwner

@receiver(post_save, sender=User)
def create_pet_owner_profile(sender, instance, created, **kwargs):
    if created:
        PetOwner.objects.create(
            user=instance,
            first_name=instance.first_name,
            last_name=instance.last_name,
            email=instance.email
        )

@receiver(post_save, sender=User)
def save_pet_owner_profile(sender, instance, **kwargs):
    instance.petowner.save()

2. 在应用的apps.py中注册信号：

from django.apps import AppConfig

class 你的应用名Config(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = '你的应用名'

    def ready(self):
        import 你的应用名.signals

注意事项 #

如果你的PetOwner模型中email、phone_number设置了unique=True约束，需要在自定义表单中增加字段唯一性校验，避免提交重复值触发数据库报错。
现有signup.html模板无需修改，form.as_p会自动渲染新增的表单字段。

内容的提问来源于stack exchange，提问作者The silent one