将字段保存为哈希并在 django 中进行比较

Question

我创建了自己的用户模型：

class CustomUser(AbstractBaseUser,PermissionsMixin):
    phone_number = models.CharField(verbose_name="Mobile Number", max_length=11, unique=True, blank=False, null=False)
    security_code = models.IntegerField(verbose_name="Security Code",default=create_security_code)
    security_code_creation = models.DateTimeField(auto_now=True)

用户可以使用密码或安全码登录。

我希望它是这样的：

如果用户想使用安全代码登录，我将在我的视图中创建一个随机安全代码。将其发短信给他们，并将安全代码的散列版本保存在我的数据库中。用户从他们收到的文本中输入数字，我将用户输入的哈希版本与我在数据库中的内容进行比较。很像密码。

但我不想将安全代码作为纯文本存储在我的数据库中。我想像密码（散列）一样存储它。

我找到了

from django.contrib.auth.hashers import make_password

我可以使用它来创建安全代码的散列版本。

我的问题是我应该编写一个单独的身份验证后端还是简单地编写/使用像 check_password 这样植入 AbstractBaseUser 的函数？

Answer 1

在您的自定义用户类中覆盖 check_password 可以像这样工作。

您在登录时提供密码或代码，该方法首先检查该值是否与密码匹配，如果不匹配则检查该值是否与安全代码匹配

from django.contrib.auth.hashers import check_password, make_password

class CustomUser(AbstractBaseUser, PermissionsMixin):
    ...
    security_code = models.CharField(verbose_name="Security Code",max_length=128)
    ...
    def set_security_code(self, code):
        self.security_code = make_password(code)
        self.security_code_creation = now()

    def check_password(self, raw_password):
        password_matched = super().check_password(raw_password)
        if not password_matched and self.security_code_creation > now() - timedelta(days=1):
            return check_password(raw_password, self.security_code)
        return False