Django扩展User模型用信号，如何避免用户注册同时生成两类实例？

问题核心是你给User模型绑定了两套独立的post_save信号（Customer和Employee各一套），用户创建时两个信号会同时触发，导致两个关联实例被自动生成。可以用以下几种方案解决：

方案一：给User增加类型标识，按需创建关联实例 #

1. 先给User模型添加用户类型字段，用来区分用户身份：

# models.py
from django.contrib.auth.models import AbstractUser
from django.db import models

class CustomUser(AbstractUser):
    USER_TYPE_CHOICES = [
        ('employee', '员工'),
        ('customer', '客户'),
    ]
    user_type = models.CharField(
        max_length=20,
        choices=USER_TYPE_CHOICES,
        default='customer'  # 可根据业务调整默认值
    )

2. 修改两个模型的信号函数，根据user_type判断是否创建对应实例：

# Customer模型的信号
@receiver(post_save, sender=CustomUser)
def create_customer(sender, instance, created, **kwargs):
    if created and instance.user_type == 'customer':
        Customer.objects.create(user=instance)

@receiver(post_save, sender=CustomUser)
def update_customer(sender, instance, created, **kwargs):
    if not created and instance.user_type == 'customer':
        instance.customer.save()

# Employee模型的信号
@receiver(post_save, sender=CustomUser)
def create_employee(sender, instance, created, **kwargs):
    if created and instance.user_type == 'employee':
        Employee.objects.create(user=instance)

@receiver(post_save, sender=CustomUser)
def update_employee(sender, instance, created, **kwargs):
    if not created and instance.user_type == 'employee':
        instance.employee.save()

注册用户时指定对应的user_type，就只会生成匹配的关联实例。

方案二：移除自动创建信号，在业务逻辑中手动创建 #

删掉两个模型里自动创建关联实例的create_xxx信号，在注册视图里根据业务场景手动生成对应实例：

# views.py
from .models import CustomUser, Customer, Employee

def register_customer(request):
    # 处理用户注册逻辑
    user = CustomUser.objects.create_user(username=..., password=..., user_type='customer')
    # 手动创建Customer实例
    Customer.objects.create(user=user)
    # 其他响应逻辑...

def register_employee(request):
    # 处理员工注册逻辑
    user = CustomUser.objects.create_user(username=..., password=..., user_type='employee')
    # 手动创建Employee实例
    Employee.objects.create(user=user)
    # 其他响应逻辑...

这种方式完全由业务代码控制实例创建时机，避免信号自动触发带来的冗余创建。

方案三：通过业务规则限制信号触发（适用于有固定判断逻辑的场景） #

如果系统中有明确的用户身份判断规则（比如邮箱域名、注册渠道），可以直接在信号中加入判断条件：

# 示例：通过邮箱域名区分员工和客户
@receiver(post_save, sender=CustomUser)
def create_customer(sender, instance, created, **kwargs):
    if created and not instance.email.endswith('@yourcompany.com'):
        Customer.objects.create(user=instance)

@receiver(post_save, sender=CustomUser)
def create_employee(sender, instance, created, **kwargs):
    if created and instance.email.endswith('@yourcompany.com'):
        Employee.objects.create(user=instance)

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