Django 多个 AUTH_USER_MODEL 在 Django 中的应用

在本文中，我们将介绍如何在 Django 中使用多个 AUTH_USER_MODEL，以及相关的配置和注意事项。在开发过程中，有时我们需要对用户模型进行扩展或创建不同类型的用户。Django 提供了 AUTH_USER_MODEL 设置来满足这些需求。

阅读更多：Django 教程

AUTH_USER_MODEL

在 Django 中，AUTH_USER_MODEL 是一个非常重要的设置，它定义了用于身份验证和授权的用户模型。默认情况下，Django 使用的是内置的 auth 应用中的 User 模型。

在开始使用多个 AUTH_USER_MODEL 之前，我们首先需要了解一些重要的概念。首先，我们需要明确的是 AUTH_USER_MODEL 只能设置一次，这意味着我们只能有一个自定义的用户模型。

扩展 User Model

如果我们需要在默认的 User 模型基础上进行扩展，我们可以使用 AbstractUser 或 AbstractBaseUser 来创建自定义的用户模型。AbstractUser 提供了一些常用的字段和方法，而 AbstractBaseUser 是一个灵活的基类，我们可以完全自定义用户模型。

下面是一个使用 AbstractUser 扩展 User 模型的示例：

from django.contrib.auth.models import AbstractUser
from django.db import models

class CustomUser(AbstractUser):
    # 添加我们自定义的字段
    bio = models.TextField(max_length=500, blank=True)
    birth_date = models.DateField(null=True, blank=True)

    # 新增方法和属性
    def get_full_name(self):
        return self.first_name + " " + self.last_name

    @property
    def is_teacher(self):
        return self.groups.filter(name='Teacher').exists()

    class Meta(AbstractUser.Meta):
        swappable = 'AUTH_USER_MODEL'

上述示例中，我们创建了一个新的 CustomUser 模型，并添加了 bio 和 birth_date 两个自定义字段。还添加了一个 get_full_name 方法和一个用于判断用户是否是教师的属性 is_teacher。Meta 类中的 swappable 设置表示该模型可以替代默认的 User 模型。

创建多个用户模型

有时候，我们需要创建不同类型的用户模型，并将它们与不同的应用关联。在 Django 中，我们可以使用 AUTH_USER_MODEL 设置来指定不同的用户模型。

假设我们有一个 e-learning 网站，我们需要创建学生和教师两种类型的用户。我们可以创建两个不同的用户模型，并将它们与不同的应用关联。

首先，我们需要在 settings.py 文件中定义两个 AUTH_USER_MODEL 常量，分别对应学生和教师的用户模型。示例如下：

AUTH_USER_MODEL = 'students.Student'

AUTH_USER_MODEL = 'teachers.Teacher'

然后，我们需要在对应的应用中创建两个用户模型，分别是 Student 和 Teacher。示例如下：

from django.contrib.auth.models import AbstractUser
from django.db import models

class Student(AbstractUser):
    # 添加学生特有的字段和方法
    grade = models.IntegerField()
    ...

class Teacher(AbstractUser):
    # 添加教师特有的字段和方法
    title = models.CharField(max_length=100)
    ...

在上述示例中，我们分别在 students 和 teachers 应用中创建了 Student 和 Teacher 模型，并分别添加了学生和教师特有的字段。

配置相关应用

一旦我们创建了多个用户模型，我们需要将它们与对应的应用关联起来。在 Django 中，我们可以使用 AUTH_USER_MODEL 的字符串形式的路径来指定用户模型。

接下来，我们需要配置相关应用。假设我们的应用包含以下两个应用：

students：学生应用
teachers：教师应用

下面是一个示例配置，将默认的 User 模型指定为学生应用的用户模型：

# settings.py

AUTH_USER_MODEL = 'students.Student'

INSTALLED_APPS = [
    ...
    'students',
    'teachers',
    ...
]

在上述示例中，我们将 AUTH_USER_MODEL 设置为 ‘students.Student’，并将 students 和 teachers 应用添加到 INSTALLED_APPS 中。

使用多个用户模型

在多个用户模型的情况下，我们可以使用相应的用户模型类来创建、验证和管理用户。

下面是一些使用多个用户模型的示例：

from django.contrib.auth import get_user_model

# 创建学生用户
student = get_user_model().objects.create_user(username='student1', password='password', grade=10)

# 创建教师用户
teacher = get_user_model().objects.create_user(username='teacher1', password='password', title='Professor')

# 验证学生用户
is_valid_student = get_user_model().objects.get(username='student1').check_password('password')