Django中如何通过信号读取CustomUser字段值创建对应角色档案

核心逻辑说明 #

读取CustomUser的字段值操作非常简单：post_save/pre_save信号的处理函数会自带instance参数，这个参数就是当前正在被保存的CustomUser实例，直接通过instance.字段名即可读取对应字段的取值。

完整实现步骤 #

1. 完善档案模型（必做） #

首先需要给两个档案模型增加一对一外键关联到CustomUser，才能建立用户和档案的对应关系：

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

class CustomUser(AbstractUser):
    is_employee = models.BooleanField(default=False)
    is_contractor = models.BooleanField(default=False)

class Employee(models.Model):
    # 关联对应用户，删除用户时同步删除档案
    user = models.OneToOneField(CustomUser, on_delete=models.CASCADE, related_name="employee_profile")
    # 其他员工拓展字段可在此处添加

class Contractor(models.Model):
    user = models.OneToOneField(CustomUser, on_delete=models.CASCADE, related_name="contractor_profile")
    # 其他承包商拓展字段可在此处添加

2. 编写信号处理函数 #

在你的app目录下新建signals.py，编写信号逻辑：

from django.db.models.signals import post_save
from django.dispatch import receiver
from .models import CustomUser, Employee, Contractor

@receiver(post_save, sender=CustomUser)
def sync_user_profile(sender, instance, created, **kwargs):
    # created参数为True代表当前是新建用户操作，如果你需要修改用户身份时同步更新档案，可去掉这个判断
    if created:
        # 直接读取CustomUser实例的自定义字段值
        if instance.is_employee:
            # get_or_create避免重复创建档案
            Employee.objects.get_or_create(user=instance)
        if instance.is_contractor:
            Contractor.objects.get_or_create(user=instance)

3. 注册信号 #

在app的apps.py中导入信号，完成注册：

from django.apps import AppConfig

class 你的App名称Config(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = '你的app名称'

    def ready(self):
        # 导入写好的信号文件
        import 你的app名称.signals

注意事项 #

• 如果业务要求一个用户只能是员工/承包商其中一种身份，可以在CustomUser的clean方法中添加校验逻辑，避免两个布尔字段同时为True
• 不需要信号触发时删除旧档案的场景下，不需要额外处理字段修改为False的逻辑，保留历史档案即可

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