用戶配置文件（客戶和員工）的 Django 信號

Question

我正在嘗試創建一個信號來為新創建的用戶創建配置文件。 但是，我需要有兩種不同類型的配置文件：客戶配置文件和員工配置文件。 將創建的配置文件類型在我的用戶模型中通過“user_type”字段決定：

用戶/models.py：

class User(AbstractBaseUser):
    USER_TYPES = (
        ('Employee', 'employee'),
        ('Customer', 'customer'),
        ('Vendor', 'vendor')
    )
    first_name = models.CharField(max_length=100)
    last_name = models.CharField(max_length=100)
    email = models.EmailField(
        verbose_name='email address',
        max_length=255,
        unique=True,
    )
    phone_number = models.CharField(max_length=20)
    user_type = models.CharField(max_length=8, choices=USER_TYPES)

    is_active = models.BooleanField(default=True)
    is_admin = models.BooleanField(default=False)

    objects = MyUserManager()

    USERNAME_FIELD = 'email'

    def __str__(self):
        return f'{self.first_name} {self.last_name} : {self.email}'

    def has_perm(self, perm, obj=None):
        "Does the user have a specific permission?"
        # Simplest possible answer: Yes, always
        return True

    def has_module_perms(self, app_label):
        "Does the user have permissions to view the app `app_label`?"
        # Simplest possible answer: Yes, always
        return True

    @property
    def is_staff(self):
        "Is the user a member of staff?"
        # Simplest possible answer: All admins are staff
        return self.is_admin


# if user.is_employee == True
class EmployeeProfile(models.Model):
    EMPLOYEE_ROLES = (
        ('Driver', 'driver'),
        ('Production', 'production'),
        ('Manager', 'manger')
    )
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    role = models.CharField(max_length=12, choices=EMPLOYEE_ROLES)

    def __str__(self):
        return str(self.user)


# if user.is_customer == True
class CustomerProfile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    company = models.CharField(max_length=100, null=True)
    address = models.CharField(max_length=100, null=True)
    address_2 = models.CharField(max_length=100, null=True)
    city = models.CharField(max_length=50, null=True)
    state = models.CharField(max_length=2, help_text="State Abbreviation (ex: OH)", null=True)
    zipcode = models.CharField(max_length=5, null=True)

    def __str__(self):
        return str(self.user)

然后在我的 user/signals.py 文件中：

@receiver(post_save, sender=User)
def create_customer_profile(sender, instance, created, **kwargs):
    if created:
        if User.user_type == 'customer':
            CustomerProfile.objects.create(user=instance)


@receiver(post_save, sender=User)
def create_employee_profile(sender, instance, created, **kwargs):
    if created:
        if User.user_type == 'employee':
            EmployeeProfile.objects.create(user=instance)


@receiver(post_save, sender=User, dispatch_uid='save_new_user_profile')
def save_profile(sender, instance, created, **kwargs):
    user = instance
    if created:
        if User.user_type == 'customer':
            customer_profile = CustomerProfile(user=user)
            customer_profile.save()
    if created:
        if User.user_type == 'employee':
            employee_profile = EmployeeProfile(user=user)
            employee_profile.save()

當我創建用戶時，不會創建任何配置文件。 我能夠獲得僅使用一個 Profile 模型和標准信號創建的配置文件：

@receiver(post_save, sender=User)
def create_user_profile(sender, instance, created, **kwargs):
    if created:
        Profile.objects.create(user=instance)


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

...但理想情況下，我希望根據創建的用戶類型擁有兩個單獨的配置文件。

Answer 1

您可以通過覆蓋User模型中的save方法來做到這一點。

def save(self, *args, **kwargs):
    super().save(*args, **kwargs)
    if self.user_type == 'customer':
        CustomerProfile.objects.create(user=self)
    elif self.user_type == 'employee':
        EmployeeProfile.objects.create(user=self)