📜  django 中的抽象用户 - Python (1)

📅  最后修改于: 2023-12-03 14:40:46.806000             🧑  作者: Mango

Django中的抽象用户

在Django中,用户验证是应用程序中最常见和最基本的部分之一。然而,有时候默认的用户模型并不能涵盖我们的需求,这时就需要使用Django中的抽象用户模型。

抽象用户模型指的是我们可以定制的基础用户模型,这时我们可以继承该模型,并在其上进行更改和定制。

在Django中,抽象用户模型由AbstractBaseUserPermissionsMixin两个类组成。我们通过继承这两个类创建自己的用户模型,如下所示:

from django.contrib.auth.models import AbstractBaseUser, PermissionsMixin
from django.db import models
from django.utils import timezone

class MyUser(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(unique=True)
    first_name = models.CharField(max_length=30)
    last_name = models.CharField(max_length=30)
    is_staff = models.BooleanField(default=False)
    is_active = models.BooleanField(default=True)
    date_joined = models.DateTimeField(default=timezone.now)

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

    objects = MyUserManager()

    def get_full_name(self):
        return f"{self.first_name} {self.last_name}"

    def get_short_name(self):
        return self.first_name

在上述代码中,我们使用了Django内置的EmailField作为用户名字段,同时根据需求添加了其他一些字段。同时我们还重写了get_full_nameget_short_name方法。

需要注意的是,我们必须自定义UserManager来管理该用户模型。通常情况下,我们会在管理器中添加一些额外的方法来管理和查询用户对象。

from django.contrib.auth.models import BaseUserManager

class MyUserManager(BaseUserManager):
    def create_user(self, email, first_name, last_name, password=None):
        if not email:
            raise ValueError('Users must have an email address')

        user = self.model(
            email=self.normalize_email(email),
            first_name=first_name,
            last_name=last_name,
        )

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

    def create_superuser(self, email, first_name, last_name, password):
        user = self.create_user(
            email=email,
            first_name=first_name,
            last_name=last_name,
            password=password,
        )

        user.is_staff = True
        user.is_superuser = True
        user.save(using=self._db)
        return user

上述代码中,我们自定义了UserManager,重写了create_usercreate_superuser方法。通常情况下,create_user方法用来创建普通用户,而create_superuser方法用来创建超级用户。

在自定义用户模型完成后,我们需要在settings.py文件中将其配置:

AUTH_USER_MODEL = 'myapp.MyUser'

在完成以上步骤后,我们就可以像使用内置的用户模型一样使用我们自定义的用户模型了。

结语

抽象用户模型让我们可以完全掌控用户模型的定义,从而更好地适应我们的应用需求。但是我们也需要注意,自定义用户模型会带来更多的维护工作,包括数据库迁移和用户认证。所以在实际应用中需要权衡利弊。