在 django 中用抽象用户覆盖用户模型的保存方法

Question

具有多种两种用户类型的 Django 项目。我按照tutorial 为内置的用户模型提供了两种类型的标志。当我尝试登录到 django rest_framework_simple_jwt 的登录端点时，响应。覆盖保存方法是否有问题。

{ “non_field_errors”：[ “没有找到具有给定凭据的活动帐户” ] }

class User(AbstractUser):
    email = models.EmailField(unique=False,)
    first_name = models.CharField(max_length=30, blank=True)
    last_name = models.CharField(max_length=30, blank=True)
    date_joined = models.DateTimeField(auto_now_add=True)
    is_active = models.BooleanField(default=True)
    is_employee=models.BooleanField(default=False)
    is_manager=models.BooleanField(default=False)

    def save(self,*args,**kwargs):
        if self.is_employee:
                Employee.objects.create(manager=self)
        super(User,self).save(*args,**kwargs)

Answer 1

我认为问题在于，当您尝试创建 Employee 对象时，没有创建您的 User。一种方法是像这样使用post_save 信号：

from django.db import models
from django.dispatch import receiver

class User(AbstractUser):
    email = models.EmailField(unique=False,)        
    # ...other fields
    # and don't override save()

@receiver(models.signals.post_save, sender=User)
def user_created(sender, instance, created, **kwargs):
    if created and instance.is_employee:
        Employee.objects.create(manager=instance)

基本上，我们只想在数据库中保存用户（并且有一个pk）之后创建一个Employee。

更多参考：https://docs.djangoproject.com/en/2.0/ref/signals/#post-save