更新從 django 中的用戶 model 獲取數據的 model 字段

Question

我有一個使用 AbstractBaseUser 的用戶 model。 我有一個 model 員工使用 Model class。 我希望通過注冊創建用戶時，用戶繼承員工 model 的所有字段。 我在用戶 model 和員工之間建立了 OnetoOne 關系。 但是創建用戶后，當我查詢員工 model 時，我找不到與創建的用戶相關的任何數據。 用戶管理員：

class UserManager(BaseUserManager):
    """
    Creates and saves a User with the given email and password.
    """
    def create_user(self, email, username, password=None):
        if not email:
            raise ValueError('Users must have an email address')

        user = self.model(
            email=self.normalize_email(email),
            username=username
        )

        user.set_password(password)
        user.save(using=self._db)
        return user

    def create_staffuser(self, email,username, password):
        user = self.create_user(
            email,
            password=password,
            username=username
        )
        user.is_staff = True
        user.save(using=self._db)
        return user

    def create_superuser(self, email, username, password):
        user = self.create_user(
            email,
            password=password,
            username=username
        )
        user.is_staff = True
        user.is_admin = True
        user.save(using=self._db)
        return user

用戶：

class User(AbstractBaseUser):
    """
    Creates a customized database table for user
    """
    email = models.EmailField(
        verbose_name='email address',
        max_length=255,
        unique=True,
    )
    username = models.CharField(max_length=255, unique=True)
    is_active = models.BooleanField(default=True)
    is_staff = models.BooleanField(default=False)
    is_admin = models.BooleanField(default=False)

    USERNAME_FIELD = 'username'
    REQUIRED_FIELDS = ['email',]

    objects = UserManager()

    def get_full_name(self):
        # The user is identified by their email address
        return self.email

    def get_short_name(self):
        # The user is identified by their email address
        return self.email

    def __str__(self):
        return 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

我使用OneToOneField(User, on_delete=models.CASCADE, null=True)擴展了用戶 model 與 OnetoOne 關系

Employee model:

    class Employee(models.Model):
        """
        Create employee attributes
        """
        employee_user = models.OneToOneField(User, on_delete=models.CASCADE, null=True)
        e_id = models.IntegerField(unique=True, null=False)
        first_name = models.CharField(max_length=128)
        last_name = models.CharField(max_length=128)
        address = models.CharField(max_length=256, blank=True)
    phone_number = models.CharField(max_length=128, blank=True)
    email = models.EmailField(max_length=128, unique=True)
        def get_absolute_url(self):
            return reverse("employee:employee-list")
        @receiver(post_save, sender=User)
    def create_or_update_user_profile(sender, instance, created, **kwargs):
        if created:
            Employee.objects.create(employee_user=instance)
        instance.Employee.save()

在這個 model 中還有一個名為 email 的字段。 我必須保留此字段，因為我正在使用此 model 創建員工創建表單。 但問題是，用戶模型的 email 字段與員工 email 字段沖突。 我必須保留兩個 email 字段。

我可以用用戶 email 字段更新員工 email 字段嗎？ 如果是，那么應該查詢什么？？？

Answer 1

您可以嘗試，但所有數據都將在同一個表中：

class Employee(User, models.Model):
    ...

否則，如果您想使用 OneToOneField 我相信您必須將 User 擴展到models.Model

Answer 2

您已經創建了一個自定義用戶 model 所以您需要從您的設置文件中導入他的 Auth_User_model，而不是使用 django 內置身份驗證用戶。 你可以這樣做

from django.conf import settings

employee_user = models.OneToOneField(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)