在 Django 中覆盖默认用户模型后无法登录

Posted 2023-02-23

【中文标题】在 Django 中覆盖默认用户模型后无法登录

【英文标题】：Can't login after overriding default user model in Django

【发布时间】：2022-01-11 04:46:22

【问题描述】：

我已经自定义了默认的 Django 模型，现在在创建新的超级用户后我无法登录。 它说登录名或密码无效。

models.py：

class UserProfileManager(BaseUserManager):

    def create_user(self, email, first_name, last_name, password=None):
        if not email:
            raise ValueError("Email is required")

        user = self.model(
            email = self.normalize_email(email),
            first_name = first_name,
            last_name = last_name,
            password = password
        )
        user.save(using=self._db)
        return user

    def create_superuser(self, email, first_name, last_name,password=None):
        user = self.create_user(
            email = self.normalize_email(email),
            password = password,
            first_name = first_name,
            last_name = last_name
        )
        user.is_admin = True
        user.is_staff = True
        user.is_superuser = True
        user.save(using=self._db)
        return user

class UserProfile(AbstractBaseUser):
    GENDERS = (
        ("Male", "Male"),
        ("Female", "Female"),
        ("Other", "Other")
    )
    email = models.EmailField(max_length=150, unique=True)
    first_name = models.CharField(max_length=50, blank=True, null=True)
    last_name = models.CharField(max_length=50, blank=True, null=True)
    birth_date = models.DateField(blank=True, null=True)
    gender = models.CharField(choices=GENDERS, max_length=50, blank=True, null=True)
    location = models.CharField(max_length=150, blank=True, null=True)
    
    date_joined = models.DateTimeField(auto_now_add=True)
    last_login = models.DateTimeField(auto_now=True)
    is_admin = models.BooleanField(default=False)
    is_active = models.BooleanField(default=True)
    is_staff = models.BooleanField(default=False)
    is_superuser = models.BooleanField(default=False)

    objects = UserProfileManager()

    USERNAME_FIELD = "email"
    REQUIRED_FIELDS = ["first_name", "last_name"]

    def __str__(self):
        return self.email

    def has_perm(self, perm, obj=None):
        return self.is_admin

    def has_module_perms(self, app_label):
        return True

settings.py：

AUTH_USER_MODEL = "UserProfile.UserProfile"
AUTHENTICATION_BACKENDS = (
        'django.contrib.auth.backends.ModelBackend',
)

所以当我创建一个新的超级用户并尝试登录管理面板时，我收到一个错误“请输入正确的员工帐户的电子邮件和密码。” 我已经尝试过删除数据库和迁移。

【参考方案1】：

您可以尝试使用基本用户管理器

请根据需要自定义字段，我只是写基础知识

def _create_user(self, email, password=None, **extra_fields):
    if not email:
        raise ValueError('The given email must be set')
    email = self.normalize_email(email)
    user = self.model(email=email, **extra_fields)
    user.set_password(password)
    user.save(using=self._db)
    return user

def create_user(self, email, password=None, **extra_fields):
    extra_fields.setdefault('is_staff', False)
    extra_fields.setdefault('is_superuser', False)
    return self._create_user(email, password, **extra_fields)
    

def create_superuser(self, email, password=None, **extra_fields):
    """Create and save a SuperUser with the given email and password."""
    extra_fields.setdefault('is_staff', True)
    extra_fields.setdefault('is_superuser', True)

    if extra_fields.get('is_staff') is not True:
        raise ValueError('Superuser must have is_staff=True.')
    if extra_fields.get('is_superuser') is not True:
        raise ValueError('Superuser must have is_superuser=True.')

    return self._create_user(email, password, **extra_fields)

Password = Password 通常不保存散列密码，这就是为什么没有用户保存方法不能直接保存密码的原因。

【讨论】：

谢谢，它有效！但是你能告诉我如何用我展示的代码实现同样的事情吗？我尝试将 Password = Password 更改为 user.set_password(password)，但它仍然不起作用。

欢迎，如果您同意这个答案，这对其他人也有帮助，并且在您的代码中您没有 _create_user 功能，这可能是原因，请尝试使用_create_user 函数在创建用户之前，user.set_password(password) 写在 _create_user 函数中而不是在创建用户函数中。